Golang数据库测试核心是事务回滚隔离:每个测试开启事务、执行后强制回滚,配合DBTX接口抽象与testdb等工具实现干净、可并行的测试;需注意DDL限制、引擎兼容性及外部依赖mock。

用 Golang 对数据库操作做测试时,核心难点是避免测试间相互干扰、不污染真实数据,同时保证每次测试的独立性和可重复性。最稳妥的方式是:在每个测试中开启事务,执行业务逻辑,然后回滚(rollback),而不是提交(commit)。这样既模拟了真实数据库交互,又确保测试干净、隔离、可并行。
使用 testdb + 事务包装器简化测试
手动管理事务容易出错(比如忘记回滚、panic 后未恢复)。推荐用 testdb(或类似轻量工具)自动包裹事务。它本质是在测试开始前开启事务,测试结束时强制回滚,无论成功或失败。
- 安装:
go get github.com/ory/dockertest/v3(用于启动临时 PostgreSQL/MySQL 容器)或直接用内存 SQLite(适合简单场景) - 更轻量的选择是自己封装一个
TestTxDB:接收 *sql.DB,调用Begin(),返回带Rollback()的 wrapper 结构体 - 关键点:wrapper 必须实现
driver.Connector或直接透传*sql.Tx给你的 DAO 层,让业务代码“以为”在操作普通 DB
在测试中注入事务连接
你的数据访问层(如 UserRepository)不应硬依赖 *sql.DB,而应接受可替换的 executor 接口(例如 Queryer 或自定义 DBTX)。
- 定义接口:
type DBTX interface { QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error); ExecContext(context.Context, string, ...interface{}) (sql.Result, error) } - 测试时传入
*sql.Tx实例(它实现了DBTX),生产环境传*sql.DB - 这样 DAO 方法无需修改,就能自然运行在事务上下文中
注意事务边界与驱动兼容性
不是所有操作都能在事务中安全执行。例如:
立即学习“go语言免费学习笔记(深入)”;
- DDL 语句(
CREATE TABLE、ALTER COLUMN)在 PostgreSQL 中不能在事务块内执行(除非用BEGIN; SET LOCAL ...等变通) - 某些 MySQL 存储引擎(如 MyISAM)不支持事务,务必用 InnoDB
- SQLite 默认支持事务,但需确认
PRAGMA journal_mode = WAL等设置不影响测试稳定性 - 若测试涉及
time.Sleep或外部 HTTP 调用,事务可能超时;建议用 mock 替代外部依赖
完整测试示例(PostgreSQL + dockertest)
以下是一个最小可行测试片段:
func TestUserRepository_Create(t *testing.T) {
pool, err := dockertest.NewPool("")
require.NoError(t, err)
resource, err := pool.Run("postgres", "15-alpine", []string{"POSTGRES_PASSWORD=secret"})
require.NoError(t, err)
defer func() { _ = pool.Purge(resource) }()
db, err := sql.Open("postgres", fmt.Sprintf("host=localhost port=%s user=postgres password=secret dbname=postgres sslmode=disable", resource.GetPort("5432/tcp")))
require.NoError(t, err)
defer db.Close()
// 每个测试用独立事务
tx, err := db.Begin()
require.NoError(t, err)
defer tx.Rollback() // 关键:无论如何都回滚
repo := NewUserRepository(tx) // 注入事务连接
user := User{Name: "alice", Email: "a@example.com"}
err = repo.Create(context.Background(), &user)
require.NoError(t, err)
var count int
err = tx.QueryRow("SELECT COUNT(*) FROM users WHERE email = $1", user.Email).Scan(&count)
require.NoError(t, err)
assert.Equal(t, 1, count)
}










