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

在Go语言中写测试时,避免重复代码的核心思路是提取公共逻辑、使用辅助函数和合理组织测试结构。直接把测试用例堆在一起虽然能跑通,但维护成本高,一旦逻辑变更就得改多处。
提取测试辅助函数(Test Helpers)
如果多个测试需要相同的准备步骤,比如启动服务、构造数据、设置环境变量等,可以封装成小函数。
这些函数通常以 setup 或 newTestXxx 命名,并放在 _test.go 文件中。
- 例如,测试HTTP handler时,每次都要创建请求、记录响应,可以写一个 newRequestRecorder 函数返回 *http.Request 和 *httptest.ResponseRecorder
- 数据库测试中,可以用 setupTestDB 初始化内存数据库并预埋基础数据
注意:辅助函数不要自己调 t.Fatal,而是返回错误或让调用方判断。这样更灵活。
立即学习“go语言免费学习笔记(深入)”;
使用表驱动测试(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 容易让测试变慢或耦合。
基本上就这些。关键是把“准备数据”、“执行动作”、“验证结果”这三个阶段拆清楚,重复的部分拎出来。测试代码也是代码,该重构就重构。










