如何編寫寫入標準輸入的Go測試？

Question

假設我有一個簡單的應用程序，它從stdin中讀取行並簡單地將它回顯給stdout。例如：

package main 

import (
    "bufio" 
    "fmt" 
    "io" 
    "os" 
) 

func main() { 
    reader := bufio.NewReader(os.Stdin) 
    for { 
     fmt.Print("> ") 
     bytes, _, err := reader.ReadLine() 
     if err == io.EOF { 
      os.Exit(0) 
     } 
     fmt.Println(string(bytes)) 
    } 
} 

我想寫寫入stdin和然後輸出與該輸入測試案例。例如：

package main 

import (
    "bufio" 
    "io" 
    "os" 
    "os/exec" 
    "testing" 
) 

func TestInput(t *testing.T) { 
    subproc := exec.Command(os.Args[0]) 
    stdin, _ := subproc.StdinPipe() 
    stdout, _ := subproc.StdoutPipe() 
    defer stdin.Close() 

    input := "abc\n" 

    subproc.Start() 
    io.WriteString(stdin, input) 
    reader := bufio.NewReader(stdout) 
    bytes, _, _ := reader.ReadLine() 
    output := string(bytes) 
    if input != output { 
     t.Errorf("Wanted: %v, Got: %v", input, output) 
    } 
    subproc.Wait() 
} 

運行go test -v給了我下面的：

=== RUN TestInput 
--- FAIL: TestInput (3.32s) 
    echo_test.go:25: Wanted: abc 
     , Got: --- FAIL: TestInput (3.32s) 
FAIL 
exit status 1 

我明明做一些不正確這裏。我應該如何去測試這種類型的代碼？

Answer 1

這是一個寫入標準輸入並從標準輸出中讀取的例子。請注意，它不起作用，因爲輸出中首先包含「>」。不過，您可以修改它以適應您的需求。

func TestInput(t *testing.T) { 
    subproc := exec.Command("yourCmd") 
    input := "abc\n" 
    subproc.Stdin = strings.NewReader(input) 
    output, _ := subproc.Output() 

    if input != string(output) { 
     t.Errorf("Wanted: %v, Got: %v", input, string(output)) 
    } 
    subproc.Wait() 
} 

Answer 2

而不是做在main與stdin和stdout的一切，你可以定義一個函數，它接受一個io.Reader和io.Writer作爲參數和做任何你想要做的事。 main然後可以調用該函數，並且您的測試函數可以直接測試該函數。