如何測試從 Stdin 讀取的 Go 應用程式？

使用標準輸入測試 Go 應用程式

在 Go 中，測試從標準輸入讀取的應用程式可能具有挑戰性。考慮一個將 stdin 輸入回顯到 stdout 的簡單應用程式。雖然看起來很簡單，但編寫驗證輸出的測試案例可能會帶來困難。

嘗試失敗

最初的方法是使用管道模擬 stdin 和 stdout並手動寫入標準輸入管道。但是，這可能會導致競爭條件和意外失敗。

解決方案：提取邏輯並測試獨立函數

而不是使用stdin 和在main 函數中執行所有操作stdout，創建一個單獨的函數，接受io. Reader 和io.Writer 作為參數。這種方法允許主函數呼叫該函數，而測試函數直接測試它。

重構程式碼

<code class="go">package main

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

// Echo takes an io.Reader and an io.Writer and echoes input to output.
func Echo(r io.Reader, w io.Writer) {
    reader := bufio.NewReader(r)
    for {
        fmt.Print("> ")
        bytes, _, _ := reader.ReadLine()
        if bytes == nil {
            break
        }
        fmt.Fprintln(w, string(bytes))
    }
}

func main() {
    Echo(os.Stdin, os.Stdout)
}</code>

更新測試案例

<code class="go">package main

import (
    "bufio"
    "bytes"
    "io"
    "os"
    "testing"
)

func TestEcho(t *testing.T) {
    input := "abc\n"
    reader := bytes.NewBufferString(input)
    writer := &bytes.Buffer{}

    Echo(reader, writer)

    actual := writer.String()
    if actual != input {
        t.Errorf("Wanted: %v, Got: %v", input, actual)
    }
}</code>

這個測試案例透過直接呼叫來模擬main 函數，並使用一個用於stdin 輸入的緩衝區和一個用於捕獲輸出的緩衝區。然後將捕獲的輸出與預期輸入進行比較，確保函數正確回顯輸入。

以上是如何測試從 Stdin 讀取的 Go 應用程式？的詳細內容。更多資訊請關注PHP中文網其他相關文章！