Go:使用数据库初始化从多个包运行测试

Question

我有一个具有此项目结构的 GO 项目(每个包中有多个此类文件)。

- api
    - userHandler.go
    - userHandler_test.go
- database
    - user.go
    - user_test.go

在 user.go 中，我有 User 结构和创建/获取/更新用户的函数(我使用的是 GORM，但这不是问题所在)。在 user_test.go 中。

我想为每个不同的文件清理数据库(删除所有数据或处于特定状态)，所以我尝试为每个文件创建 1 个套件(使用 Testify)，然后使用 SetupSuite功能，但行为似乎不是确定性的，可能我做错了什么。

所以我的问题是:

• 共享数据库连接的最佳方式是什么？使用全局变量是最佳选择？
• 在数据库中创建表一次，然后在每次运行 file_test.go 之前用自定义数据初始化数据库的最佳方法是什么？

现在我也有一个奇怪的错误:运行

go test path/package1
go test path/package2

一切正常，但如果我运行(测试所有包)

cd path && go test ./...

我有一些似乎不是确定性的错误，这就是为什么我猜测数据库连接处理不当

Answer 1

如果您的 api 包依赖于您的 database 包(看起来是这样)，那么您的 api 包应该有办法提供一个数据库连接池(例如 *sql.DB)到它。

在您对 api 包的测试中，您应该只传入一个您可以使用的初始化池(可能带有预填充的测试模式/夹具)。这可以是您在 init() 中为 api 包初始化的全局变量，也可以是 setup() 和 defer teardown() 每个测试函数中的模式。

这是前一种(最简单的)方法，您只需创建一个共享数据库和模式供您的测试使用。

package database

import testing

var testDB *sql.DB

// This gets run before your actual test functions do.
func init() {
    var err error
    db, err = sql.Open(...)
    if err != nil {
        log.Fatalf("test init failed: %s", err)
    }

    _, err := db.Exec(`CREATE TABLE ....`)
    if err != nil {
        log.Fatalf("test schema creation failed: %s", err)
    }
}

一些提示:

• 您还可以调用一个 setup() 函数来创建一个带有随机后缀的表并插入您的测试数据，这样您的测试就不会使用相同的测试表(因此存在冲突或冲突的风险)相互依赖)。捕获该表名并将其转储到您的 defer teardown() 函数中。
• https://medium.com/@benbjohnson/structuring-applications-in-go-3b04be4ff091值得一读以获得更多视角。