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

Go: Run test from multiple package with DB initialization

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

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

在 user.go 内部,我有用户结构和 Create/Get/Update 用户的功能(我正在使用 GORM,但这不是问题所在)。在user_test.go.

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

所以我的问题是:

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

go test path/package1
go test path/package2

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

cd path && go test ./...

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

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

在你对 api 包的测试中,你应该只传递一个你可以使用的初始化池(也许用测试 schema/fixtures 预填充)。这可以是您在 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() 函数可以创建一个带有随机后缀的 table 并插入您的测试数据,这样您的测试就不会使用相同的测试 table (因此存在冲突或相互依赖的风险)。捕获该 table 名称并将其转储到您的 defer teardown() 函数中。
  • https://medium.com/@benbjohnson/structuring-applications-in-go-3b04be4ff091 值得一读以获得更多视角。