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 包的测试中，您应该只传入一个您可以使用的初始化池（可能预先填充了测试架构/固定装置）。这可以是您在 api 包中为 init() 初始化的全局变量，也可以是每个测试函数中的 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 值得一读，以获得更多视角。