减少Golang测试代码重复的技巧

提取公共逻辑、使用辅助函数和表驱动测试是Go中避免重复的核心方法。通过封装setup函数处理共通准备步骤，用表驱动整合多用例，结合TestMain管理全局初始化，提升测试可维护性。

在Go语言中写测试时，避免重复代码的核心思路是提取公共逻辑、使用辅助函数和合理组织测试结构。直接把测试用例堆在一起虽然能跑通，但维护成本高，一旦逻辑变更就得改多处。

提取测试辅助函数（Test Helpers）

如果多个测试需要相同的准备步骤，比如启动服务、构造数据、设置环境变量等，可以封装成小函数。

这些函数通常以 setup 或 newTestXxx 命名，并放在 _test.go 文件中。

• 例如，测试HTTP handler时，每次都要创建请求、记录响应，可以写一个 newRequestRecorder 函数返回 *http.Request 和 *httptest.ResponseRecorder
• 数据库测试中，可以用 setupTestDB 初始化内存数据库并预埋基础数据

注意：辅助函数不要自己调 t.Fatal，而是返回错误或让调用方判断。这样更灵活。

使用表驱动测试（Table-Driven Tests）

这是Go中最常见的去重方式。把多个相似测试用例组织成切片，在循环中执行。

每个用例包含输入、期望输出和名称（可选），结构清晰又容易扩展。

• 比如测试一个解析函数，不同输入对应不同结果，全塞进一个 for range 里
• 用 t.Run(name, func) 为每个子测试命名，失败时能快速定位是哪个用例出问题

表驱动不仅减少重复代码，还鼓励你思考更多边界情况。

共用测试数据和配置

如果多个包或测试文件需要相同的数据结构（如用户JSON、配置对象），可以单独建一个 testdata 包或目录。

里面放共享的模型实例、文件路径、默认配置等。

• 例如定义 var ValidUser = User{Name: "alice", Age: 30}
• 或者提供 LoadFixture(filename) 从 testdata 目录读取测试文件

这样修改一份数据就能影响所有相关测试，一致性更好。

利用 TestMain 控制全局流程

如果测试前需要初始化资源（如连接数据库、加载配置），可以在 TestMain 中统一处理。

它像 main 函数一样运行，能控制所有测试的执行时机。

• 在 m.Run() 前做 setup，之后做 teardown
• 适合一次性启动 mock 服务或清理临时目录

但别滥用，大多数场景用 setup 函数就够了，TestMain 容易让测试变慢或耦合。

基本上就这些。关键是把“准备数据”、“执行动作”、“验证结果”这三个阶段拆清楚，重复的部分拎出来。测试代码也是代码，该重构就重构。