Go测试如何测试数据库操作_Go集成测试思路

5次阅读

go集成测试数据库应通过Testcontainers动态启停容器实现环境隔离,结合事务回滚避免数据污染,并用接口抽象DB依赖便于mock单元测试,同时启用sql日志和t.Log辅助快速定位问题。

Go测试如何测试数据库操作_Go集成测试思路

Go集成测试中数据库连接怎么配才不污染本地环境

直接复用开发或测试数据库,轻则数据错乱,重则删库跑路。必须隔离——不是靠“我手动清空表”,而是让每次测试启动独立实例或临时数据库。

  • 本地开发用 docker run --rm -p 5432:5432 -e POSTGRES_PASSword=pass postgres:15 启一个临时 postgresql,测试完容器自动销毁
  • 更稳妥的是用 testcontainers-go:它在测试运行时动态拉起/销毁容器,支持 PostgreSQL、mysqlsqlite(通过内存模式),且能自动注入随机端口和连接字符串
  • 避免硬编码 localhost:5432 ——把数据库 DSN 放进 TestMainos.Setenv("TEST_DB_DSN", dsn),测试函数里统一用 os.Getenv("TEST_DB_DSN")

事务回滚比 delete FROM 更快更安全

很多人写完测试就 DELETE FROM users 清表,但表多、外键多、索引多时极慢;还可能漏删关联数据,导致下个测试失败。

  • PostgreSQL/MySQL 支持在测试前开启事务,执行完所有操作后 tx.Rollback(),完全无副作用
  • 示例:在 TestDBCreateUser 开头 tx, _ := db.Begin(),之后所有 tx.Query 操作都走该事务;结尾 tx.Rollback() 即可,不用管表结构或外键
  • 注意:不能用 db.Exec("CREATE table ...") 这类 DDL 语句——它们在 PostgreSQL 中会隐式提交事务,导致回滚失效

如何让 Go 测试跳过真实 DB 调用做单元测试

不是所有函数都需要连库——比如校验用户邮箱格式、拼接 SQL 字符串逻辑,完全可以不碰数据库。

  • 把数据库操作封装成接口,例如 type UserRepo Interface { CreateUser(ctx context.Context, u *User) Error },实现类叫 postgresUserRepo
  • 测试时传入 mock 实现(用 gomock 或手写),断言调用参数、返回错误类型,不启动任何连接
  • 关键点:不要在 Struct 里直接 embed *sql.DB,而要依赖接口;否则无法替换,mock 成本陡增

测试失败时怎么快速定位是 SQL 写错了还是数据没插进去

常见现象:expected 1, got 0,但不知道是 INSERT 没执行,还是 WHERE 条件写反了,还是时间字段没设默认值。

  • 在测试里加一句 t.Log("inserted user:", u)t.Log("fetched:", found),比看日志文件快得多
  • 启用 SQL 日志:PostgreSQL 可加 ?sslmode=disable&connect_timeout=5 并设置 log.Println 包裹 sql.Open 的 driver,或用 pgx/log/zap 输出每条实际执行的语句
  • 别信“我刚手动查过表里有数据”——测试用的是另一个连接、另一个事务、甚至另一个数据库实例

测试里最易被忽略的,是事务隔离级别和时区设置。PostgreSQL 默认 READ COMMITTED 没问题,但如果你代码里显式用了 REPEATABLE READ,而测试没同步设,行为就会不一致;还有 time.Now() 插入后查出来变成 UTC 时间,本地却是 CST——这些不会报错,但数据对不上。

text=ZqhQzanResources