如何使用Golang对数据库操作进行测试_结合事务和回滚保证隔离_技术教程

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 方法无需修改，就能自然运行在事务上下文中

注意事务边界与驱动兼容性

不是所有操作都能在事务中安全执行。例如：

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)
}