Kaynağa Gözat

Merge pull request #7112 from heyitsanthony/expect-debug

expect: EXPECT_DEBUG environment variable
Anthony Romano 9 yıl önce
ebeveyn
işleme
998f8bf291
1 değiştirilmiş dosya ile 6 ekleme ve 0 silme
  1. 6 0
      pkg/expect/expect.go

+ 6 - 0
pkg/expect/expect.go

@@ -17,6 +17,7 @@ package expect
 
 import (
 	"bufio"
+	"fmt"
 	"io"
 	"os"
 	"os/exec"
@@ -39,6 +40,8 @@ type ExpectProcess struct {
 	err   error
 }
 
+var printDebugLines = os.Getenv("EXPECT_DEBUG") != ""
+
 // NewExpect creates a new process for expect testing.
 func NewExpect(name string, arg ...string) (ep *ExpectProcess, err error) {
 	ep = &ExpectProcess{cmd: exec.Command(name, arg...)}
@@ -65,6 +68,9 @@ func (ep *ExpectProcess) read() {
 		ep.mu.Lock()
 		ep.err = rerr
 		if l != "" {
+			if printDebugLines {
+				fmt.Printf("%s-%d: %s", ep.cmd.Path, ep.cmd.Process.Pid, l)
+			}
 			ep.lines = append(ep.lines, l)
 			ep.count++
 			if len(ep.lines) == 1 {