我試圖在單元測(cè)試中模擬 cmd.Start() 并且無(wú)法弄清楚我們需要模擬整個(gè)函數(shù)還是可以模擬 cmd.Start() 函數(shù)？有人可以幫助我嗎？ package main import ( "bytes" "fmt" "os/exec" ) var execCommand = exec.Command func main() { stdout := &bytes.Buffer{} cmd := execCommand("cmd") syscall.SysProcAttr := &syscall.SysProcAttr{CmdLine: "/S /c C:\\Temp\\test.exe /S C:\\Temp\\test.log"} cmd.Stdout = stdout errs := cmd.Start() if errs != nil { fmt.Println("command run fialed :", errs) fmt.Println("OUTPUT :", stdout.String()) } }單元測(cè)試：func fakeExecCommand(command string, args ...string) *exec.Cmd { cs := []string{"-test.run=TestExecCommandHelper", "--", command} cs = append(cs, args...) cmd := exec.Command(os.Args[0], cs...) es := strconv.Itoa(mockedExitStatus) cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1", "STDOUT=" + mockedStdout, "EXIT_STATUS=" + es} return cmd}
1 回答
倚天杖
TA貢獻(xiàn)1828條經(jīng)驗(yàn) 獲得超3個(gè)贊
我無(wú)法模擬 cmd.Start()，無(wú)論如何我們都不能這樣做，下面是對(duì)我有用的方式。
func fakeExecCommand(command string, args ...string) *exec.Cmd {
cs := []string{"-test.run=TestExecCommandHelper", "--", "ENTER YOUR COMMAND HERE"}
cs = append(cs, args...)
cmd := exec.Command(os.Args[0], cs...)
es := strconv.Itoa(mockedExitStatus)
cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1", "STDOUT=" + mockedStdout, "EXIT_STATUS=" + es}
return cmd
}
func TestPrintDate(t *testing.T) {
mockedExitStatus = 1
mockedStdout = "Sun Aug 201"
execCommand = fakeExecCommand
defer func() { execCommand = exec.Command }()
expDate := "Sun Aug 20"
out, _ := printDate()
if string(out) != expDate {
t.Errorf("Expected %q, got %q", expDate, string(out))
}
}
/////////CODE///////
func printDate() ([]byte, error) {
cmd := execCommand("date")
out, err := cmd.CombinedOutput()
return out, err
}
- 1 回答
- 0 關(guān)注
- 154 瀏覽
添加回答
舉報(bào)
0/150
提交
取消