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