我有一个使用 MySQL 的 gRPC 服务,需要在每个测试用例后清除记录。我尝试用事务来包装每个测试用例。如果我的 rpc 代码中没有事务,它就可以工作,但如果有,它就会失败。并且会出现如下错误:
can't start transaction
...
sql: Transaction has already been committed or rolled back
然后我尝试使用 truncate 来清除记录,但一些测试用例随机失败。
我的代码是这样的(我使用 gorm):
func foo(db *gorm.DB) {
tx := db.Begin()
// query and insert
tx.Commit()
}
// Use transaction to do database cleanup
func TestFooVersion1() {
testDB := initDB()
tx = testDB.Begin() // setup
foo(testDB)
tx.Rollback() // teardown
}
// Use truncate to do database cleanup
func TestFooVersion2() {
testDB := initDB()
foo(testDB)
truncateTables(testDB) // teardown
}
func truncateTables(db *gorm.DB) {
// exec "TRUNCATE TABLE table;" for every table
}
使用 DB(MySQL) 测试代码的正确方法是什么? (我不喜欢像 go-sqlmock 这样的 mock)
最佳答案
如果没有看到代码很难说,但根据错误,您似乎在多个测试之间共享一个连接,并且每个测试都在尝试启动一个事务。确保每个测试都打开自己的连接,启动自己的事务,并在完成后提交或回滚并关闭连接。
关于mysql - 如何测试使用数据库(mysql)的 Go 代码?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45029545/