使用事务回滚确保测试隔离，通过传入*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 实例。
使用 sqlmock 进行 SQL 模拟（可选）
若不想连接真实数据库，可以使用 sqlmock 库模拟 SQL 执行过程。
它能验证 SQL 语句、参数是否正确，并返回预设结果。
导入 github.com/DATA-DOG/go-sqlmock 用 sqlmock.New() 创建 mock 对象 设置期望的 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) }}
这个测试既验证了逻辑，也确认了数据写入正确。
基本上就这些。核心是控制数据状态，保证测试可重复、无副作用。
以上就是如何使用Golang测试数据库操作的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1411412.html
微信扫一扫 
支付宝扫一扫 
