测试数据库交互

一分耕耘，一分收获！既然打开了这篇文章《测试数据库交互》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我有一个带有存储层的 api。它只进行数据库交互并执行 crud 操作。现在我想测试一下这些功能。 在我的路径 api/storage/ 中，我有不同的包，它们具有与同一数据库中的不同表交互的功能。表a、b、c位于同一个数据库中。

我的文件层次结构如下：

--api
  --storage
    --a
      --a.go
      --a_test.go
    --b
    --c
  --server
    --a
  --testdata
    --a.sql
    --b.sql

这样我想使用命令测试整个存储层

go test ./...

我遵循的方法是我有一个函数refreshtables，它首先截断表，然后用我保存在 testdata 文件夹中的固定测试数据填充它。对于截断，我这样做：

db.exec("set foreign_key_checks = 0;")
db.exec("truncate " + table)
db.exec("set foreign_key_checks = 1;")

由于 go test 默认并行运行不同包的测试函数，因此会创建多个 sql 连接，并且 truncate 在其他连接上运行，而 setforeign key 在某些连接上运行其他连接从连接池中随机分配。

如果一起运行，我无法通过测试，但如果单独运行或按包运行，所有测试都会通过。

如果我这样做：

go test ./... -p 1

这使得测试函数一一运行，所有测试都通过。

我还尝试使用事务进行截断并在截断之前锁定表。

我检查了这篇文章（https://medium.com/kongkow-it-medan/parallel-database-integration-test-on-go-application-8706b150ee2e），他建议在每个测试函数中创建不同的数据库，函数结束后删除该数据库。我认为这会非常耗时。

如果有人建议在 golang 中测试数据库交互的最佳方法，那将会非常有帮助。

解决方案

我在集成测试方面没有太多经验，我不确定模拟数据库驱动程序是否适合您，但如果是这样，我一直在使用 go-sqlmock 包来模拟单元测试中的 sql 数据库结果，并且奇迹般有效。您可以使用它，并为每个测试提供一个单独的“数据库引擎”。这有点耗时，因为您必须手动告诉模拟期望什么查询以及返回什么，但相信我，这是一个很好的时间投资。

正如我之前所说，我不确定使用此策略是否适合您的情况，因为如果您有兴趣了解应用程序在“真实数据库场景”中的行为方式，例如验证注册表是否实际保存，然后进行模拟数据库结果有点无用。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《测试数据库交互》文章吧，也可关注golang学习网公众号了解相关技术文章。