我可以从 TestMain 输出日志消息吗？

Question

我想编写一个集成测试来测试向数据库写入数据和从数据库读取数据。为此，我需要第一个测试函数来连接并准备测试数据库；如果成功，则运行其他测试，然后对其进行查询。

因此，我实现了一个 TestMain(m *testing.M)，它创建一个数据库并在其上运行迁移，如果没有错误，我会运行 m.Run()。

但它没有像 testing.T 那样的任何 m.Print 或 m.Fatal 方法。如果我尝试使用 log.Print() 记录某些内容，它不会获得输出（可能是因为没有测试可以被正确标记为失败）。

我不喜欢使用常规测试功能创建数据库的原因是，常规测试当然不能阻止其他测试的运行，因此该文件中的所有其他测试都将运行并困扰输出消息，因为他们都会失败。

那么我如何从 TestMain 中注销一些内容呢？我的 TestMain 仍然可能以多种方式失败，我想在输出中看到它是什么。

显示 TestMain 的输出未显示的示例：

$ ls
go.mod      main_test.go
$ cat go.mod
module App

go 1.16
$ cat main_test.go 
package main

import (
    "testing"
    "log"
)

func TestMain(m *testing.M) {
    log.Print("hello")
}
$ go test ./.
ok      App 0.147s
$ 

Answer 1

如果测试以非零代码退出，则使用 log.Print() 打印的消息似乎会显示，因此这种情况最好通过在日志行之后以非零代码退出来解决，例如所以：

log.Print("setup of database has failed")
os.Exit(1)

Answer 2

您需要 -v 标志：

go test -v .

注意：. 相当于 ./.

如果您想在所有子目录上运行测试，请使用 < code>./... 就像这样：

go test -v ./...

这是为什么？由于您正在所谓的包列表模式（即 ./.）中运行测试 - 来自 go help test 中的文档：

在此（包列表）模式下，go test 编译并测试列出的每个包
在命令行上。如果包测试通过，则 go test 仅打印
最后的“ok”总结行。如果包测试失败，则 go test 打印
完整的测试输出。

这是您注意到的行为。进一步阅读：

...如果使用 -bench 或 -v 标志调用，则转到
即使通过了包测试，测试也会打印完整的输出，以便
显示请求的基准测试结果或详细日志记录。

---

注意： go test 单独（即没有包列表）将允许从 TestMain 输出日志记录，如 log.Print。

---

编辑
以下是我通常为测试进行飞行前设置的方式：

var testDB     *pgxpool.Pool // all tests should use this

func createDBconn() (*pgxpool.Pool, error) {
    // read ENV VARs - establish connection
    // ...
    return dbconn, nil
}

func TestMain(m *testing.M) {
    var err error
    
    testDB, err = createDBconn()
    if err != nil {
        log.Fatal(err)
    }

    // call flag.Parse() here if TestMain uses flags
    os.Exit(m.Run())
}

如果任何准备失败，则 log.Fatal 确保记录原因并记录原因。程序终止。