17370845950

Go测试无内置setup/teardown，可用TestMain实现包级初始化与清理，子测试+闭包或辅助函数+defer实现函数级隔离，需注意并发安全与资源唯一性。

在 Go 的测试中，没有内置的 setup/teardown 机制（如 JUnit 的 @Before/@After），但可以通过标准测试框架和合理结构模拟出类似行为，确保每次测试前资源就绪、测试后干净释放，从而维持环境一致性。

用 TestMain 实现包级 setup/teardown

适用于需要在整组测试开始前初始化一次、结束后清理一次的场景（例如启动本地数据库、创建临时目录、设置全局 mock）。

• 定义 func TestMain(m *testing.M)，调用 setup() → m.Run() → teardown()
• 必须显式调用 os.Exit(m.Run())，否则测试会提前退出
• 注意：TestMain 是包级的，不适用于单个测试函数的隔离需求

示例：

func TestMain(m *testing.M) {
    setupDB()
    defer teardownDB()
    os.Exit(m.Run())
}

用子测试（subtest）+ 闭包模拟函数级 setup/teardown

对单个测试函数需要独立资源时最常用。把 setup/teardown 逻辑封装进闭包，配合 t.Run 使用，天然支持并行（需手动加锁或使用独立资源实例）。

• 每个 subtest 自己调用 setup（如创建临时文件、初始化内存 DB 实例）
• 用 defer 在函数末尾触发 cleanup（如 os.RemoveAll、关闭连接）
• 避免跨 subtest 共享可变状态，防止干扰

示例：

func TestUserService(t *testing.T) {
    t.Run("create user", func(t *testing.T) {
        db := setupInMemoryDB() // 每次新建
        defer db.Close()       // 自动清理
        svc := NewUserService(db)
        // ... 测试逻辑
    })
}

用测试辅助函数 + defer 组织可复用逻辑

将重复的初始化/清理抽象成函数，提高可读性和复用性。关键点是让 setup 返回 cleanup 函数，由调用方 defer 执行。

• setup 函数返回资源句柄 + cleanup 函数（类型为 func()）
• 调用方统一用 defer cleanup()，语义清晰且不易遗漏
• 适合数据库连接、HTTP server、临时目录等常见资源

示例：

func setupTempDir(t *testing.T) (string, func()) {
    dir, err := os.MkdirTemp("", "test-*")
    if err != nil {
        t.Fatal(err)
    }
    return dir, func() { os.RemoveAll(dir) }
}

func TestReadConfig(t *testing.T) {
    dir, cleanup := setupTempDir(t)
    defer cleanup()
    // ... 写配置、调用被测函数
}

注意并发与资源隔离

Go 测试默认并发执行（go test -race 可检测竞争），若多个测试共用同一资源（如端口、文件路径、全局变量），易导致失败或非确定行为。

• 避免硬编码端口、文件名；改用 net.Listen("tcp", ":0") 或 os.MkdirTemp 获取唯一值
• 不要在多个测试间共享可变全局状态（如 var db *sql.DB），应每个测试新建或重置
• 若必须共享（如慢启动的外部服务），加互斥锁 + once 控制，但要注明“非并行安全”