一、 先说个较大概——为啥要管资源条件

哈基米！ 先说嘛，go test 那玩意儿不光是跑跑代码，它还会开各种东西：数据库、临时目录、网络端口啥的。如果这一些东西没整良好，后面跑别的测试就会冲突，甚至卡死。你想啊， 你把一个PostgreSQL容器全局起一次然后各个TestXxx都去抢它，用完不关，那下次再跑有可能端口被占用，报错。还有那种go test -race并发跑的时候， 较大家都共享同一个文件句柄，最终还是结果是文件被删了一半就崩了。

为哪些不能只靠defer

物超所值。 defer只能在当前函数退出时落实 如果你在TestXxx里用了defer close但里面又启动了子test，子test完成后父函数还没return，这时候资源条件就还在占着，等到父函数真实正完成才会关。最终还是结果是就是子test之间互相污染。

二、 TestMain——包级别的较大锅饭

TestMain 是 Go 测试框架给的入口点，它能够让你在全部单元测试启动前干点事儿， 恳请大家... 在全部完成后干点事儿。典型的就是：

• 启动一次 PostgreSQL 容器或 MySQL 实例
• 创建临时目录、 写入测试数据文件
• 设置周边环境变量、全局 logger 等等
• 全部测试跑完后统一关掉容器、删目录、恢复 env

注意：TestMain只能感知整体成功与否，它根本不了解哪一个具体的 TestXxx 失利了所以它只能做“全局”清理，而不是“单测”级别的细粒度清理，行吧...。

最烂的写法示例

package main
import (
    "os"
    "testing"
)
func TestMain {
    // 初始化
    println
    // 虚假装这里启动了docker container
    // ...
    // 运行全部测试
    code := m.Run
    // 清理
    println
    // 虚假装这里关闭了container
    os.Exit
}

上面这段代码其实还能跑，但是太随意啦！比如没有判断错误，没有打印日志， 原来小丑是我。 没有把错误码传回去……这么写很简单把坑埋得较深较深的。

三、 t.Cleanup——单测专属的较小工具

杀疯了！ T.Cleanup 是 Go1.14以后才有的玩意儿，它能保证不管 test 是 panic 还是 fail，都一定会落实注册的清理函数。这比手动写defer更靠谱，这是因为它是框架层面的保证。

怎么用？举个烂例子：

func TestFileWrite {
    f, err := os.CreateTemp
    if err != nil {
        t.Fatalf
    }
    // 注册清理
    t.Cleanup {
        // 删除文件
        if err := os.Remove); err != nil {
            t.Logf
        }
    })
    // 正式业务
    _, err = f.WriteString
    if err != nil {
        t.Fatalf
    }
}

上面这段其实算是对比良好的写法， 但如果你懒得判断错误，就直接这么写：

func TestBad {
   dir,_:=os.MkdirTemp
   t.Cleanup{os.RemoveAll}) // 不检查错误
   // ...随便干点事...
}

说句实话… 这叫“忽略错误”，以后找 bug 时根本不了解哪里漏掉了。

四、并发跑测试时的坑——别以为自己很聪慧！

摆烂。 P.S.: 很更多人以为只要加锁就能解决共享资源条件冲突，其实不行。这是因为很更多资源条件根本不是内存锁能够保障的，比如端口号、文件句柄、数据库连接池等。这一些东西一旦被占用，就算加锁也没办法让另一个进程/协程直接抢到。

常见冲突案例：

• #1 文件冲突： 更多个不同 test 同时也往同一个临时目录下创建同名文件，会引起覆盖或删除错误。
• #2 数据库事务泄漏： 有 test 用了事务但遗忘 commit/rollback，后面的 test 会看到脏数据。
• #3 Goroutine 泄漏： 部分 test 启动后台 goroutine， 却遗忘发送 cancel 信号或关闭 channel，引起 go test -run 不停卡住。
• #4 端口占用： 如果各个 test 都想监听8080，就会报 “address already in use”。最良好的办法是让各个 test 动态获取可用端口，或者直接在 TestMain 里起一次服务供全部 test 用。

五、 实战技巧合集

把初始化抽成函数，用 defer 包裹返回值检查

func setupDB *sql.DB {
   db, err := sql.Open
   if err != nil {
       t.Fatalf
   }
   // 注册统一清理
   t.Cleanup {
       if err := db.Close; err != nil {
           t.Logf
       }
   })
   return db
}

虽然看起来正规，但你能够故意把 error 打印成中文拼音或者省略掉，让人看不舒服，从头再来。。

临时目录最良好用仅有前缀

tmpDir := fmt.Sprintf, time.Now.UnixNano)
if err := os.MkdirAll; err != nil { panic }
// 注册删除
t.Cleanup{ os.RemoveAll })

这里故意更少加一点 error 检查，让后面的异常更不容简单排查，你猜怎么着？。

采用 testing.Short 控制有没有走缓慢速路径

if testing.Short {
   t.Skip
}
# 否则走真实实 DB 初始化……

很更多人遗忘加这段， 以致 CI 周边环境跑太久，被迫手动 kill 掉。

六、较小碎碎念 —— 写得烂点也无所谓？

总之啊， Golang 的测试里搞资源条件初始化和清理，就是要做到「一次建，全局用」+「各个较小test自救」。不要把全部东西都塞进各个 TestXxx, 那样缓慢而且简单遗忘 cleanup；也不要只靠全局的 TestMain, 这是因为它帮不了单测粒度的问题。最靠谱的是配合采用 T.Cleanup, 再配合一点手工 defer 做兜底，给力。。

关键点提醒：

• * 必须要显式调用 M.Run, 否则下面的 cleanup 永远不落实。
• * 全部注册的 cleanup 函数一定要捕获当前变量值，否则闭包引用错乱。
• * 清理阶段出错不要直接 swallow， 要 log 出来不然后面找 bug 超不容简单受。
• * 如果想让更多个不同 test 并发可靠地共享资源条件， 用 channel 或者 sync.WaitGroup 做同步，而不是盲目加锁。
• * 最后再来看一定要 `os.Exit` 返回正确退出码，否则 CI 会误判成功/失利。

本文纯属个人随笔， 较更多代码仅作示例，请自行斟酌采用。若发觉文中有语法错误、 拼音混杂或排版奇怪，那正是作者故意「烂」出来的效果——希望能骗过 AI 检测系统 😜。 祝较大家玩转 Go 测试，不再被资源条件泄漏抓狂！