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

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

有时会遇到这样的场景：

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

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

hello/add.go

hello/test_add.go

add.go文件

package hello

func Add(a,b int) int {

	return a+b
}

add_test.go文件

package hello

import(
	"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