17370845950

go 1.4 引入 `testmain` 函数，允许在所有测试执行前后统一进行初始化和清理操作；但需确保 go 版本 ≥1.4，且必须显式调用 `m.run()` 并通过 `os.exit()` 退出，否则测试框架无法正确接管流程。

在 Go 测试中，若希望在每个测试函数执行后自动清理资源（如重置数据库、关闭连接、清空缓存等），需明确区分两种场景：

• ✅ 单个测试后的清理：应使用 t.Cleanup()（Go 1.14+）或 defer，这是最推荐、最安全的方式；
• ✅ 整个测试包运行前/后的全局操作（如启动/关闭测试服务器、迁移数据库 schema）：才使用 func TestMain(m *testing.M) —— 它不是“每次测试后运行”，而是整个 go test 过程仅执行一次。

⚠️ 重要前提：testing.M 类型自 Go 1.4 起引入。若遇到 undefined: testing.M 错误，请先运行 go version 确认版本。低于 1.4 的环境不支持该特性。

正确使用 T

estMain

TestMain 不会“在每个测试后运行”，而是在所有测试开始前执行初始化，在 m.Run() 返回后执行收尾。示例：

package main

import (
    "os"
    "testing"
)

// 模拟测试前准备（如创建临时目录、启动 mock server）
func setup() {
    // e.g., os.MkdirAll("testdata", 0755)
}

// 模拟测试后清理（如删除临时文件、关闭服务）
func teardown() {
    // e.g., os.RemoveAll("testdata")
}

func TestSomeTest(t *testing.T) {
    t.Cleanup(func() {
        // ✅ 推荐：每个测试专属的后置清理（Go 1.14+）
        // 例如：重置数据库记录、恢复全局状态
        t.Log("cleaned up after TestSomeTest")
    })
    t.Log("running TestSomeTest")
}

func TestMain(m *testing.M) {
    setup()
    // 必须调用 m.Run() 启动测试流程
    code := m.Run()
    teardown()
    // 必须用 os.Exit(code)，否则 defer 不生效且测试可能提前退出
    os.Exit(code)
}

注意事项

• TestMain 是可选的；若未定义，测试框架将默认逐个运行 Test* 函数；
• m.Run() 会阻塞并返回整数退出码（0 表示全部通过），不可省略；
• os.Exit() 是必需的——defer 在 os.Exit() 后不会执行，因此清理逻辑必须写在 m.Run() 之后、os.Exit() 之前；
• 若需“每个测试后清理”，优先使用 t.Cleanup()（线程安全、自动按注册逆序执行），它比手动 defer 更可靠，尤其在并发测试（t.Parallel()）中；
• TestMain 无法访问具体 *testing.T 实例，因此不能用于测试粒度的日志或失败标记。

总结

升级 Go 版本并合理组合 TestMain 与 t.Cleanup，可构建健壮、高效、可维护的测试套件。