Go 多个测试怎么共享一次初始化并统一清理

多个 Go 测试都要连接同一个测试服务、创建一次客户端，或在结束时统一关闭资源时，推荐在测试包中写 TestMain(m *testing.M)。把一次性初始化放在 m.Run() 前，把清理放在它后面，并把返回码交给 os.Exit；这样所有 TestXxx 可以共享基础设施，但测试结果不会被清理逻辑覆盖。

用 TestMain 把测试包的生命周期收拢起来

普通的 init 函数适合初始化包变量，却不适合表达“测试开始前准备、测试全部结束后关闭”的成对生命周期。TestMain 由测试框架调用，运行在主 goroutine 中；m.Run() 返回前会执行这个包里的测试和基准测试，返回后才进入收尾阶段。

最小写法如下。示例用一个共享的测试客户端表示数据库连接、模拟服务或消息客户端，实际项目可以替换成自己的类型。

package service_test

import (
    "os"
    "testing"
)

var testClient *Client

func TestMain(m *testing.M) {
    // 测试包只创建一次共享依赖。
    testClient = newTestClient()

    // 保存测试结果，清理失败时也不能覆盖它。
    code := m.Run()

    // 所有 TestXxx 结束后再关闭共享依赖。
    testClient.Close()
    os.Exit(code)
}

func TestCreate(t *testing.T) {
    // 测试只复用客户端，数据仍应使用独立的测试夹具。
    if err := testClient.Create("demo"); err != nil {
        t.Fatal(err)
    }
}

有一个容易忽略的点：TestMain 依赖测试命令行参数时，应在读取这些参数前显式调用 flag.Parse()。普通测试函数运行时参数已经解析完成，但 TestMain 被调用时并不是这个状态。

共享基础设施，别共享会互相污染的测试数据

“只初始化一次”不等于“所有测试共用一份可变数据”。客户端、连接池、只读配置通常适合包级共享；用户记录、临时文件、请求体和会被修改的内存对象，应该在测试或子测试内部创建。

局部资源优先使用 t.Cleanup。它会在当前测试及其子测试返回后执行，注册位置离资源创建点很近，失败路径也不容易遗漏：

func newFixture(t *testing.T) *Fixture {
    t.Helper()

    // 每个测试创建独立数据，避免测试顺序影响结果。
    fixture := createFixture(testClient)
    t.Cleanup(func() {
        // 无论断言成功或失败，都删除本测试的数据。
        if err := fixture.Delete(); err != nil {
            t.Logf("清理测试夹具失败: %v", err)
        }
    })
    return fixture
}

func TestUpdate(t *testing.T) {
    fixture := newFixture(t)
    if err := fixture.Update("new-value"); err != nil {
        t.Fatal(err)
    }
}

如果几个子测试要共享一组准备动作，可以让父测试持有资源，再用 t.Run 创建子测试。父测试的清理会等子测试结束后执行；如果子测试调用了 t.Parallel()，父测试仍是这组并行子测试的收束点。

不要让清理逻辑改写 m.Run 的结果

m.Run() 的返回值是整个测试包的退出码。正确顺序是先保存，再清理，最后退出。不要直接写成 os.Exit(m.Run()) 后又在后面清理，因为 os.Exit 不会执行后续语句；也不要在清理失败时把成功码重新赋回去。

如果共享客户端本身有可变状态，初始化一次可能反而制造竞争。此时可以共享不可变配置或连接工厂，实际会话和请求上下文在测试内创建；需要并行运行时，再检查客户端文档是否明确支持并发使用。

落地前检查这几个边界

• 初始化失败要尽早结束：如果共享服务没有创建成功，不要让后续测试拿着空指针继续跑。
• 清理动作要可重复：测试失败、短测试模式和筛选单个测试时，都不应依赖完整套件才可释放资源。
• 并行测试不要写共享可变字段：必要时用每个测试独立的 fixture、锁或事务回滚。
• 需要参数时在 TestMain 中先解析 flag，并把测试包退出码原样传给 os.Exit。

官方 testing 包文档把 TestMain 定义为测试程序的低层入口；如果只是给一个普通测试准备资源，优先使用 t.Cleanup，代码会更容易理解。

常见问题

TestMain 会为每个 Test 函数执行一次吗？

不会。一个测试包通常只有一个生成的测试程序，TestMain 围绕这次程序运行执行一遍；每个测试自己的准备动作仍需放在测试函数或辅助函数中。

TestMain 里可以不调用 m.Run 吗？

可以编译，但会跳过测试和基准测试，通常不符合预期。除非你明确要接管整个测试程序，否则应调用 m.Run() 并传递其退出码。

共享数据库连接后测试还是偶发失败怎么办？

先区分“连接共享”和“数据共享”：保留连接池可以，但每个测试要隔离记录、事务或 schema，并检查并行测试是否修改了同一行数据。