stdinに書き込む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に書き込んだ出力を入力と比較するテストケースを作成したいとします。例えば：私は明らかにここに間違った何かをやっている

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

：

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を実行すると、次の私に与えます。このタイプのコードをテストするにはどうすればよいですか？代わりにstdinとstdoutとmainですべてを行うことの

Answer 1

ここでは、stdinに書き込み、stdoutから読み込む例を示します。出力には最初に ">"が含まれているため、動作しないことに注意してください。それでも、ニーズに合わせて変更することができます。

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

、あなたはio.Readerやパラメータなどio.Writerを取り、あなたはそれがやりたいものは何でもない関数を定義することができます。 mainはその関数を呼び出すことができ、テスト関数はその関数を直接テストできます。