本文介绍如何通过依赖注入将 stdin/stdout 逻辑抽象为可测试函数，避免直接操作 os.stdin/stdout 导致的测试难题，并提供完整、可靠、可复现的测试示例。
本文介绍如何通过依赖注入将 stdin/stdout 逻辑抽象为可测试函数，避免直接操作 os.stdin/stdout 导致的测试难题，并提供完整、可靠、可复现的测试示例。
Go 程序中直接读写 os.Stdin 和 os.Stdout 会使单元测试变得困难——因为标准流是全局、不可重入、且难以模拟的资源。正确的做法是将核心逻辑解耦为接受 io.Reader 和 io.Writer 参数的纯函数，让 main 函数仅负责“胶水”调用，从而实现高内聚、低耦合与可测试性。
✅ 推荐方案：函数化 + 依赖注入
首先重构原始程序，将交互逻辑提取为独立函数：
package main
import (
"bufio"
"fmt"
"io"
"os"
)
// EchoLoop 是可测试的核心逻辑：从 r 读取行，写入 w，支持任意 io.Reader/io.Writer
func EchoLoop(r io.Reader, w io.Writer) error {
reader := bufio.NewReader(r)
for {
fmt.Fprint(w, "> ")
bytes, isPrefix, err := reader.ReadLine()
if err != nil {
if err == io.EOF {
return nil // 正常结束
}
return err // 其他错误
}
if isPrefix {
return fmt.Errorf("line too long")
}
_, _ = fmt.Fprintln(w, string(bytes))
}
}
func main() {
if err := EchoLoop(os.Stdin, os.Stdout); err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
}
✅ 编写可靠单元测试（无需 exec.Command）
现在可以使用 strings.Reader 模拟输入，bytes.Buffer 捕获输出，完全在内存中完成端到端验证：
package main
import (
"bytes"
"io"
"strings"
"testing"
)
func TestEchoLoop(t *testing.T) {
// 构造输入：含换行符的多行字符串
input := "hello\nworld\n"
reader := strings.NewReader(input)
var buf bytes.Buffer
// 执行被测函数
err := EchoLoop(reader, &buf)
if err != nil {
t.Fatalf("EchoLoop failed: %v", err)
}
// 预期输出格式："> hello\n> world\n"
expected := "> hello\n> world\n"
actual := buf.String()
if actual != expected {
t.Errorf("output mismatch\nWanted:\n%q\nGot:\n%q", expected, actual)
}
}
? 注意事项：
- strings.NewReader 提供确定性、无副作用的输入源；
- bytes.Buffer 完美实现 io.Writer 接口，便于断言输出；
- 避免使用 exec.Command 启动子进程：它引入竞态（如未及时 flush）、超时风险和平台差异，且无法覆盖错误路径（如 io.ErrUnexpectedEOF）；
- 若需测试真实终端行为（如颜色、光标控制），应单独做集成测试，而非单元测试。
✅ 进阶：支持上下文取消与超时（可选）
若业务要求响应式退出（例如用户按 Ctrl+C），可进一步增强函数签名：
func EchoLoopWithContext(ctx context.Context, r io.Reader, w io.Writer) error {
reader := bufio.NewReader(r)
for {
select {
case ")
bytes, isPrefix, err := reader.ReadLine()
if err != nil {
if err == io.EOF {
return nil
}
return err
}
if isPrefix {
return fmt.Errorf("line too long")
}
_, _ = fmt.Fprintln(w, string(bytes))
}
}
此时测试中可传入 context.WithTimeout，精准验证超时行为。
通过这种设计，你的代码既保持了简洁性，又具备生产级可测试性与可维护性——这才是 Go “少即是多”哲学的真正实践。
