Gaminik


testing - 其他功能

TestMain

在写测试时,有时需要在测试之前或之后进行额外的设置(setup)或拆卸(teardown);有时,测试还需要控制在主线程上运行的代码。为了支持这些需求,testing 提供了 TestMain 函数:

func TestMain(m *testing.M)

如果测试文件中包含该函数,那么生成的测试将调用 TestMain(m),而不是直接运行测试。TestMain 运行在主 goroutine 中, 可以在调用 m.Run 前后做任何设置和拆卸。注意,在 TestMain 函数的最后,应该使用 m.Run 的返回值作为参数调用 os.Exit。

另外,在调用 TestMain 时, flag.Parse 并没有被调用。所以,如果 TestMain 依赖于 command-line 标志 (包括 testing 包的标记), 则应该显示的调用 flag.Parse。注意,这里说的依赖,说的是如果 TestMain 函数内依赖 flag,则必须显示调用 flag.Parse,否则不需要,因为 m.Run 中调用 flag.Parse。

一个包含 TestMain 的例子如下:

package mytestmain

import (  
	"flag"
	"fmt"
	"os"
	"testing"
)

var db struct {  
	Dns string
}

func TestMain(m *testing.M) {
	db.Dns = os.Getenv("DATABASE_DNS")
	if db.Dns == "" {
		db.Dns = "root:123456@tcp(localhost:3306)/?charset=utf8&parseTime=True&loc=Local"
	}

	flag.Parse()
	exitCode := m.Run()

	db.Dns = ""

	// 退出
	os.Exit(exitCode)
}

func TestDatabase(t *testing.T) {
	fmt.Println(db.Dns)
}

Run 感兴趣的可以阅读源码,了解其原理。

Test Coverage

测试覆盖率,这里讨论的是基于代码的测试覆盖率。

Go 从 1.2 开始,引入了测试覆盖率的支持,使用的是 cover 相关的工具(go test -covergo tool cover)。虽然 testing 包提供了 cover 相关函数,不过它们是给 cover 的工具使用的。

关于测试覆盖率的更多信息,可以参考官方的博文:The cover story

导航


书籍推荐