Golang数据库操作单元测试技巧

使用内存数据库如SQLite配合事务回滚可实现高效隔离的单元测试，通过接口抽象与Mock提升逻辑独立性，集成测试则可用Docker启动真实数据库验证兼容性，确保测试可重复且无副作用。

在Go语言开发中，数据库操作的单元测试是保障数据层逻辑正确性的关键环节。直接使用生产数据库进行测试会带来副作用，比如数据污染、测试不可重复、运行速度慢等。因此，合理的做法是通过模拟或隔离手段让测试更可靠、可重复且高效。

使用内存数据库替代真实数据库

对于基于SQL的数据库（如MySQL、PostgreSQL），可以使用 SQLite 作为内存数据库来运行测试。SQLite 支持大部分标准 SQL 语法，并且可以通过 :memory: 模式在每次测试时创建一个全新的空数据库，确保测试之间互不干扰。

示例：使用 GORM 连接 SQLite 内存数据库

db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
if err != nil {
    t.Fatal("failed to connect database")
}
defer db.Close()

接着可以在测试前自动迁移表结构：

db.AutoMigrate(&User{})

这样每个测试都从干净的数据库开始，避免状态残留。

使用事务回滚保证测试隔离

即使使用了内存数据库，仍需防止多个测试用例之间相互影响。一个常见技巧是在每个测试中使用数据库事务，在测试结束时执行回滚，确保任何写入都不会真正提交。

实现方式如下：

tx := db.Begin()
defer tx.Rollback()

repo := NewUserRepository(tx)
// 执行插入、查询等操作

所有数据库操作都在事务内完成，测试结束后自动回滚，无需手动清理数据。

接口抽象 + Mock 测试逻辑

如果不想依赖任何数据库驱动，可以对数据库操作进行接口抽象，然后在测试中使用 mock 实现。

定义用户仓库接口：

type UserRepo interface {
    Create(user *User) error
    FindByID(id uint) (*User, error)
}

测试时提供一个模拟实现：

type MockUserRepo struct {
    users map[uint]*User
}

func (m *MockUserRepo) FindByID(id uint) (*User, error) {
    if user, ok := m.users[id]; ok {
        return user, nil
    }
    return nil, errors.New("not found")
}

这种方式适合业务逻辑复杂但数据库交互简单的场景，能显著提升测试速度并解耦外部依赖。

集成测试使用 Docker 启动真实数据库

单元测试推荐轻量快速的方式，但如果要做集成测试验证与真实数据库的兼容性，可借助 testcontainers-go 在测试时动态启动容器化数据库。

优点：

• 验证实际SQL语句在目标数据库中的行为
• 检测驱动配置问题（如时区、字符集）
• 适合CI/CD环境中运行端到端流程

注意这类测试不应归类为“单元测试”，而应单独组织在 integration_test.go 文件中。

基本上就这些。根据项目需求选择合适的方法：追求速度和隔离用内存数据库+事务回滚；强调逻辑独立可用接口mock；需要端到端验证再引入真实数据库环境。关键是保持测试可重复、无副作用、易于维护。