Add SetEnv() to Cmd mocking interface

Author: Alfonso Acosta

exec/exec.go

@@ -14,6 +14,7 @@ type Cmd interface {
 	Kill() error
 	Output() ([]byte, error)
 	Run() error
+	SetEnv([]string)
 }
 
 // Command is a hook for mocking
@@ -28,3 +29,7 @@ type realCmd struct {
 func (c *realCmd) Kill() error {
 	return c.Cmd.Process.Kill()
 }
+
+func (c *realCmd) SetEnv(env []string) {
+	c.Cmd.Env = env
+}

test/exec/exec.go

@@ -68,6 +68,8 @@ func (c *mockCmd) Run() error {
 	return nil
 }
 
+func (c *mockCmd) SetEnv([]string) {}
+
 func (b *blockingReader) Read(p []byte) (n int, err error) {
 	<-b.quit
 	return 0, nil