如何测试从 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>

这个测试用例通过直接调用 Echo 函数来模拟 main 函数，并使用一个用于 stdin 输入的缓冲区和一个用于捕获输出的缓冲区。然后将捕获的输出与预期输入进行比较，确保函数正确回显输入。

以上是如何测试从 Stdin 读取的 Go 应用程序？的详细内容。更多信息请关注PHP中文网其他相关文章！