go test 初始化--- TestMain的使用

go test 功能，提高了开发和测试的效率。

有时会遇到这样的场景：

进行测试之前需要初始化操作(例如打开连接)，测试结束后，需要做清理工作(例如关闭连接)等等。这个时候就可以使用TestMain()。

下面例子的文件结构如下：

hello/add.go

hello/test_add.go

add.go文件

package hellofunc Add(a,b int) int {return a+b
}


add_test.go文件

package helloimport(
"fmt"
"testing"
)func TestAdd(t *testing.T) { r := Add(1, 2)
if r !=3 { t.Errorf("Add(1, 2) failed. Got %d, expected 3.", r)
}}func TestMain(m *testing.M) {
fmt.Println("begin")
m.Run()
fmt.Println("end")
}


测试从TestMain进入，依次执行测试用例，最后从TestMain退出。

$ go test –v


output:

begin

=== RUN TestAdd

--- PASS: TestAdd (0.00s)

PASS

end

ok hello 0.432s