使用真实数据库或Mock进行Go语言数据库测试，关键在于连接管理、事务隔离与数据清理。建议用Docker启动PostgreSQL或SQLite做集成测试，通过TestMain初始化db，在事务中运行每个测试并回滚，避免数据污染；结合seed函数准备测试数据，使用testify/assert断言结果。对于单元测试，可借助go-sqlmock库模拟SQL执行过程，验证语句与参数。接口抽象有助于解耦逻辑与数据层，提升可测性。最终根据场景选择集成测试保真或mock测试提效，确保稳定性和速度兼顾。
在Go语言开发中，数据库操作是常见需求。为了保证数据层逻辑的正确性，编写可靠的测试用例至关重要。本文介绍如何用Golang对数据库操作进行测试，涵盖连接管理、测试结构设计、事务控制和常用工具实践。
使用真实数据库进行集成测试
虽然单元测试可以模拟数据库行为，但集成测试更能反映真实场景。建议使用轻量级数据库如SQLite或临时PostgreSQL实例进行测试。
以PostgreSQL为例，可以在测试前启动一个Docker容器：
docker run -d --name testdb -e POSTGRES_PASSWORD=test -p 5432:5432 postgres测试代码中使用sqlx或标准database/sql包建立连接：
立即学习“go语言免费学习笔记（深入）”；
- 定义全局
*sql.DB变量用于测试 - 在
TestMain中初始化连接和清理资源 - 每个测试运行后清空相关表数据
利用事务隔离测试数据
避免测试间的数据干扰，最佳做法是在事务中运行每个测试，并在结束时回滚。
示例：
func TestUserRepository_Create(t *testing.T) {
tx, _ := db.Begin()
defer tx.Rollback()
repo := NewUserRepository(tx)
user := &User{Name: "alice"}
err := repo.Create(user)
if err != nil {
t.Fatal(err)
}
}
这样测试不会留下脏数据，也不影响其他测试用例。
准备测试数据与断言结果
测试前插入预设数据能提高可重复性。可以写辅助函数快速构建测试环境。
例如：
- 编写
seedUsers(t *testing.T, db *sql.DB)函数插入基础用户 - 使用
testify/assert库做结构化断言 - 查询数据库验证记录是否存在或字段是否匹配
断言示例：
assert.Equal(t, "alice", fetchedUser.Name) assert.NotZero(t, fetchedUser.ID)
使用Mock替代数据库（适用于单元测试）
若想脱离数据库运行测试，可用接口抽象DB操作，然后实现mock版本。
定义仓库接口：
type UserRepository interface {
Create(*User) error
FindByID(int) (*User, error)
}
测试时传入mock实现，验证方法调用次数和参数即可。
推荐使用go-sqlmock库，它能模拟*sql.DB行为：
- 预期SQL语句执行
- 返回伪造的结果集
- 验证参数绑定是否正确
基本上就这些。关键在于根据场景选择合适方式：集成测试重真实，mock测试重速度。结合事务回滚和数据初始化，能让数据库测试既稳定又高效。不复杂但容易忽略的是清理机制和隔离设计，务必重视。
