是否所有测试都适用于TestMain？

对于大型项目中分散在不同包中的集成测试，在根目录中编写一个 TestMain 函数进行全局设置面临挑战。Go 测试命令为每个包编译单独的二进制文件，因此主包中的代码无法影响子包。

我有一个相当大的项目，许多集成测试分散在不同的包中。我使用构建标签来分隔单元、集成和 e2e 测试。

在运行集成和 e2e 测试之前，我需要进行一些设置，因此我将 testmain 函数放入根目录中的 main_test.go 文件中。这很简单：

//go:build integration || e2e
// +build integration e2e

package test

import (
  ...
)

func testmain(m *testing.m) {
  if err := setup(); err != nil {
    os.exit(1)
  }

  exitcode := m.run()

  if err := teardown(); err != nil {
    os.exit(1)
  }

  os.exit(exitcode)

}

func setup() error {
  // setup stuff here...
  return nil
}

func teardown() error {
  // tear down stuff here...
  return nil
}

但是，当我运行测试时：

$ go test -v --tags=integration ./...

testing: warning: no tests to run
PASS

# all of my subdirectory tests now run and fail...

我真的不想在每个需要它的包中编写一个 testmain ，并且希望我可以在根目录中编写一个 testmain 。您有什么可以建议的解决方案吗？谢谢。

我能想到的唯一选择是在代码之外设置所有内容，然后运行集成测试。也许有一些 shell 脚本会进行设置，然后调用 $ go test？

正确答案

go test ./... 命令在后台为每个包编译一个测试二进制文件并一一运行它们。如果您尝试指定输出，这也是您收到 cannot use -o flag with multiple packages 错误的原因。这就是主包中的代码不会影响子包的原因。

因此，使其发挥作用的唯一方法是将所有设置逻辑放入“设置”包中，并从所有子包中调用共享代码（我知道，仍然有很多工作）。

为了避免代码重复，我使用了一个函数来进行设置/拆卸并评估函数作为测试。

该函数应该如下所示

func withtestsetup(t *testing.t, testfunction func()) {
    // setup code

    testfunction()

    // teardown code
}

我使用 t *testing.t 参数来报告安装或拆卸中的错误，但可以省略。

然后在您的测试中您可以进行：

func TestFoo(t *testing.T) {
    WithTestSetup(
        t, func() {
            if err := Foo(); err != nil {
                t.Fatal(err)
            }
        },
    )
}

如果需要，只需调用 withtestsetup，对我来说看起来比在项目上添加一堆 testmain 更容易。

终于介绍完啦！小伙伴们，这篇关于《是否所有测试都适用于TestMain？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！