请求示例：Golang中模拟Stdin和Stdout编写控制台应用测试

没问题，我来给你展示怎么测试这个依赖标准输入输出的函数！核心思路是在测试时临时替换os.Stdin和os.Stdout，模拟输入并捕获输出，测试完成后再恢复原来的标准流，避免干扰其他测试。

假设你的SimpleFunc在main包中，下面是完整的测试代码：

package main

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

// 你的原函数
func SimpleFunc() {
	reader := bufio.NewReader(os.Stdin)
	output, _ := reader.ReadString('\n')
	fmt.Println(output)
}

func TestSimpleFunc(t *testing.T) {
	// 先保存原来的标准输入输出，测试结束后恢复，避免影响其他测试
	originalStdin := os.Stdin
	originalStdout := os.Stdout
	defer func() {
		os.Stdin = originalStdin
		os.Stdout = originalStdout
	}()

	// 模拟用户输入的内容：这里输入"Hello, Test!\n"
	inputBuffer := bytes.NewBufferString("Hello, Test!\n")
	os.Stdin = inputBuffer

	// 创建缓冲区来捕获函数的输出
	var outputBuffer bytes.Buffer
	os.Stdout = &outputBuffer

	// 调用待测试的函数
	SimpleFunc()

	// 验证输出是否符合预期
	// 注意：ReadString会读取到换行符，而fmt.Println会额外追加一个换行，所以预期是两个换行
	expectedOutput := "Hello, Test!\n\n"
	if outputBuffer.String() != expectedOutput {
		t.Errorf("输出不符合预期：实际得到 %q，期望得到 %q", outputBuffer.String(), expectedOutput)
	}
}

关键细节说明： #

• 恢复标准流：用defer语句确保测试结束后把os.Stdin和os.Stdout恢复成原来的状态，不然其他测试用例会出问题。
• 模拟输入：通过bytes.Buffer构造包含预设输入的缓冲区，替换os.Stdin后，函数读取的就是我们指定的内容。
• 捕获输出：同样用bytes.Buffer替换os.Stdout，函数执行时的fmt.Println会把内容写入这个缓冲区，之后就能直接读取验证。
• 换行符注意：ReadString('\n')会把输入里的换行符一起读进来，而fmt.Println会自动加一个换行，所以预期输出是输入内容加上两个换行符。

如果你的函数在其他包中，只需要调整测试文件的包名即可，核心逻辑完全通用。

内容的提问来源于stack exchange，提问作者Hifnie Bilfash