如何为 bufio.NewScanner 模拟 stdout 流进行单元测试

本文介绍如何通过自定义 io.reader 实现对 bufio.newscanner 的轻量级测试，避免依赖真实标准输出，提升测试的可重复性与隔离性。

本文介绍如何通过自定义 io.reader 实现对 bufio.newscanner 的轻量级测试，避免依赖真实标准输出，提升测试的可重复性与隔离性。

在 Go 中，bufio.NewScanner 接收任意实现了 io.Reader 接口的类型，因此无需 mock 复杂对象——只需提供一个能按需返回字节并适时返回 io.EOF 的简易 reader 即可完成测试。核心在于理解：Scanner 本身负责按行（或按分隔符）切分数据，而 Reader 只需提供原始字节流。

以下是一个简洁、生产就绪的测试用 io.Reader 实现：

type stringReader struct {
    data []byte
    done bool
}

func (r *stringReader) Read(p []byte) (n int, err error) {
    if r.done {
        return 0, io.EOF
    }
    n = copy(p, r.data)
    r.done = true
    return n, nil
}

// NewStringReader 创建一个一次性读取指定字符串的 Reader
func NewStringReader(s string) io.Reader {
    return &stringReader{data: []byte(s)}
}

该实现具备以下优点：

• ✅ 线程安全（仅用于单次测试，无并发场景）；
• ✅ 零依赖（不引入额外包）；
• ✅ 行为明确：首次 Read 返回全部内容，后续立即返回 io.EOF；
• ✅ 兼容换行符处理：scanner.Scan() 自动按 \n（或 \r\n）分割，无需 reader 关心逻辑。

使用示例（完整测试片段）：

func TestScanLines(t *testing.T) {
    input := "hello\nworld\nGo\n"
    scanner := bufio.NewScanner(NewStringReader(input))

    var lines []string
    for scanner.Scan() {
        lines = append(lines, scanner.Text())
    }

    if err := scanner.Err(); err != nil {
        t.Fatal(err)
    }

    expected := []string{"hello", "world", "Go"}
    if !reflect.DeepEqual(lines, expected) {
        t.Errorf("expected %v, got %v", expected, lines)
    }
}

⚠️ 注意事项：

• 不要复用同一 stringReader 实例多次调用 Scan() —— 它设计为“一次读完即 EOF”；
• 若需模拟流式分块读取（如网络延迟场景），可扩展 Read 方法，按需填充 p 并控制 done 时机；
• 避免在 Read 中直接 copy(p, []byte(s))（触发重复切片分配），应提前转为 []byte 缓存；
• scanner.Text() 返回不含换行符的纯文本；若需保留原始行尾，改用 scanner.Bytes()。

总结：测试 bufio.Scanner 的关键不是模拟 stdout，而是提供符合 io.Reader 合约的可控输入源。上述 stringReader 简洁、高效、语义清晰，是 Go 单元测试中推荐的标准实践。