使用事务回滚确保测试隔离,通过传入*sql.Tx实现数据操作函数的可测试性,结合sqlmock模拟SQL验证逻辑,保证测试可重复且无副作用。

测试数据库操作在 Golang 中是确保数据层逻辑正确的重要环节。关键在于隔离真实环境、使用事务控制以及合理模拟数据。以下是具体实践方法。
使用测试数据库或事务回滚
避免影响生产或开发数据库,测试时应使用独立的数据库实例,或在事务中运行测试并在结束后回滚。
常见做法是在测试开始时创建一个事务,所有操作在此事务中进行,测试结束调用 Rollback() 清理数据。
- 打开数据库连接后,使用 db.Begin() 启动事务
- 将事务对象(*sql.Tx)传给数据操作函数
- 测试完成后调用 tx.Rollback(),无论成功失败都清理数据
这样每个测试都是干净且独立的,不会相互干扰。
立即学习“go语言免费学习笔记(深入)”;
编写可测试的数据访问函数
为了让函数更容易被测试,应避免直接依赖全局 *sql.DB,而是通过参数传入 *sql.DB 或 *sql.Tx。
例如:
func CreateUser(db *sql.DB, name string) error {
_, err := db.Exec("INSERT INTO users (name) VALUES (?)", name)
return err
}
测试时就可以传入事务对象代替 DB 实例。
使用模板与程序分离的方式构建,依靠专门设计的数据库操作类实现数据库存取,具有专有错误处理模块,通过 Email 实时报告数据库错误,除具有满足购物需要的全部功能外,成新商城购物系统还对购物系统体系做了丰富的扩展,全新设计的搜索功能,自定义成新商城购物系统代码功能代码已经全面优化,杜绝SQL注入漏洞前台测试用户名:admin密码:admin888后台管理员名:admin密码:admin888
使用 sqlmock 进行 SQL 模拟(可选)
若不想连接真实数据库,可以使用 sqlmock 库模拟 SQL 执行过程。
它能验证 SQL 语句、参数是否正确,并返回预设结果。
适合单元测试,但不能替代集成测试。
完整测试示例(使用事务回滚)
假设有一个插入用户的方法:
func TestCreateUser(t *testing.T) {
db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/test_db")
if err != nil {
t.Fatal(err)
}
defer db.Close()
tx, err := db.Begin()
if err != nil {
t.Fatal(err)
}
defer tx.Rollback() // 确保测试后数据不残留
err = CreateUser(tx, "alice")
if err != nil {
t.Errorf("expected no error, got %v", err)
}
var count int
err = tx.QueryRow("SELECT COUNT(*) FROM users WHERE name = ?", "alice").Scan(&count)
if err != nil {
t.Fatal(err)
}
if count != 1 {
t.Errorf("expected 1 user, got %d", count)
}
}
这个测试既验证了逻辑,也确认了数据写入正确。
基本上就这些。核心是控制数据状态,保证测试可重复、无副作用。









