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

在编写集成测试时，开发人员需要在 testmain 中执行测试前的准备工作，例如连接和设置数据库。但是，testmain 缺乏类似 testing.t 的输出方法，这使得从 testmain 输出日志消息变得困难。这篇文章将介绍一种解决方案，通过在日志行后以非零代码退出，从而显示 testmain 中输出的消息。

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

因此，我实现了 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
$ 

正确答案

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

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

好了，本文到此结束，带大家了解了《我可以从 TestMain 输出日志消息吗？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！