17370845950

Go中测试命令行程序应解耦主逻辑与I/O依赖，通过接口抽象、参数注入和缓冲I/O实现高效单元测试；避免os/exec.Command，改用flag.NewFlagSet、bytes.Buffer模拟stdin/stdout/stderr，并替换os.Exit为可拦截的panic机制。

在 Go 中测试命令行程序，关键在于解耦主逻辑与 I/O 依赖，让核心功能可被直接调用和断言。不建议在测试中真正执行 os/exec.Command 启动新进程（难控制、慢、跨平台行为不一致），而应通过接口抽象、参数注入和缓冲 I/O 来实现高效、可靠的单元测试。

将业务逻辑从 main 分离出来

避免把所有代码写在 func main() 里。提取一个可测试的入口函数，接收 *flag.FlagSet、io.Reader 和 io.Writer 等依赖：

• 用 flag.NewFlagSet 替代全局 flag 包，避免测试间标志冲突
• 把标准输入/输出作为参数传入，便于替换为 bytes.Buffer
• 返回错误而非直接调用 os.Exit，让测试能捕获失败路径

模拟命令行参数和标准输入

使用 bytes.Buffer 模拟 stdin，并用 strings.Fields 或切片构造参数列表：

• 对输入：创建 buf := bytes.NewBufferString("hello\nworld\n")，传给函数作为 io.Reader
• 对参数：调用时传入 []string{"-v", "--input=file.txt"}，再用 fs.Parse(args)
• 注意：第一个参数（通常是程序名）需显式包含，如 args := []string{"mytool", "-h"}

捕获并断言标准输出和错误输出

用两个 bytes.Buffer 分别接管 stdout 和 stderr，然后检查内容：

• outBuf, errBuf := &bytes.Buffer{}, &bytes.Buffer{}
• 调用目标函数时传入 outBuf 和 errBuf
• 用 outBuf.String() 获取输出字符串，配合 assert.Equal 或 require.Contains 验证
• 特别注意换行符（\n）和空格，Go 的 fmt.Println 会自动加换行

处理 os.Exit 的测试（可选但实用）

如果原逻辑调用了 os.Exit，测试会终止整个进程。可用 panic 拦截模拟：

• 定义一个可替换的 exit 函数变量：var exit = os.Exit
• 在测试中临时设为 exit = func(int) { panic("exit called") }
• 用 defer func() { ... }() 捕获 panic 并验证退出码
• 生产代码仍用真实 os.Exit，仅测试时替换

不复杂但容易忽略。核心就三点：抽离、注入、缓冲。测得越早，命令行工具越稳。