本文介绍一种纯 Go、无需外部框架的惯用方式：通过可变量（var out io.Writer 和 var exit func(int)）解耦标准输出与程序退出逻辑，使 CLI 功能可测试、可捕获、可断言。
本文介绍一种纯 go、无需外部框架的惯用方式：通过可变量（`var out io.writer` 和 `var exit func(int)`）解耦标准输出与程序退出逻辑，使 cli 功能可测试、可捕获、可断言。
在 Go 中对 CLI 程序进行单元测试时，直接调用 fmt.Println 或 os.Exit 会严重阻碍测试——前者将输出写入真实终端，后者直接终止进程，无法被常规测试流程捕获。常见的“绕过方案”如 os/exec 启动子进程（慢、不可靠、跨平台问题多）或完整封装 fmt（过度设计、维护成本高），均非最佳实践。
Go 社区公认的惯用解法是：将依赖项提升为包级可变量（exported or unexported var），利用 Go 的变量可重赋值特性，在测试中动态替换其行为。核心思想来自《The Go Programming Language》第 11 章（Kernighan & Donovan），强调“依赖注入 via variables”。
✅ 正确做法：依赖抽象化 + 可变绑定
首先，修改业务逻辑，将硬编码的 fmt.Println 替换为 fmt.Fprint(out, ...)，并将 os.Exit 抽象为函数变量：
package main
import (
"fmt"
"io"
"os"
)
// 可被测试替换的依赖项（包级变量）
var out io.Writer = os.Stdout
var exit func(int) = os.Exit
type sample struct {
value int64
}
func (s sample) useful() {
if s.value == 0 {
fmt.Fprint(out, "Error: something is wrong!\n")
exit(1)
} else {
fmt.Fprint(out, "May the force be with you!\n")
}
}
⚠️ 注意：out 类型为 io.Writer（而非 *bytes.Buffer），保证接口抽象性；exit 是函数类型，便于在测试中模拟退出逻辑而不真正终止进程。
✅ 测试实现：捕获输出 + 模拟退出
在测试中，我们分别：
- 使用 bytes.Buffer 替换 out，捕获所有 fmt.Fprint 输出；
- 用闭包临时覆盖 exit，记录期望的退出码；
- 利用 defer 确保测试后恢复原始状态（避免测试污染）。
package main
import (
"bytes"
"testing"
)
func TestUsefulPositive(t *testing.T) {
// 保存原始 out 并替换为 buffer
originalOut := out
out = new(bytes.Buffer)
defer func() { out = originalOut }()
s := sample{42}
s.useful()
output := out.(*bytes.Buffer).String()
expected := "May the force be with you!\n"
if output != expected {
t.Errorf("expected %q, got %q", expected, output)
}
}
func TestUsefulNegative(t *testing.T) {
// 保存并替换 out
originalOut := out
out = new(bytes.Buffer)
defer func() { out = originalOut }()
// 保存并替换 exit
originalExit := exit
var capturedCode int
exit = func(code int) { capturedCode = code }
defer func() { exit = originalExit }()
s := sample{0}
s.useful()
output := out.(*bytes.Buffer).String()
if output != "Error: something is wrong!\n" {
t.Errorf("unexpected output: %q", output)
}
if capturedCode != 1 {
t.Errorf("expected exit code 1, got %d", capturedCode)
}
}
✅ 为什么这是更优解？
- 零依赖：不引入任何第三方测试库或 mock 工具；
- 高性能：无进程 fork、无 I/O 等待，纯内存操作；
- 可读性强：逻辑清晰，符合 Go “explicit over implicit” 哲学；
- 可组合：该模式同样适用于 log, time.Now, rand 等标准库依赖；
- 生产安全：变量默认指向真实实现，不影响运行时行为。
? 总结
测试 CLI 的本质，不是“mock 标准库”，而是控制依赖的流向。通过将 io.Writer 和 func(int) 声明为包级变量，你赋予了测试对副作用（打印、退出）的完全掌控权。这既轻量又地道，是 Go 生态中经久验证的测试惯式。记住：让代码可测，从来不是测试的责任，而是设计的责任。
