Goのテストでsetup/teardownする #golang

go1.4でtestingパッケージにTestMainが追加されました。

func TestMain(m *testing.M)を定義してあると、go testはこのメソッドを呼び出します。試しに書いてみます:

package example_test

import (
    "os"
    "testing"
)

func TestA(t *testing.T) {
}

func TestB(t *testing.T) {
}

func setup() {
    println("setup")
}

func teardown() {
    println("teardown")
}

func TestMain(m *testing.M) {
    setup()
    ret := m.Run()
    if ret == 0 {
        teardown()
    }
    os.Exit(ret)
}

gist: go1.4 TestMain example

TestAとTestBはなにかのテストコードだと思ってください。 TestMainで、setupとteardownを呼び出すようにしています。この例ではテストが成功した時だけteardornするようにしています。 go test -vしてみると:

$ go test -v
setup
=== RUN TestA
--- PASS: TestA (0.00s)
=== RUN TestB
--- PASS: TestB (0.00s)
PASS
teardown
ok      testmain    0.004s

となり、パッケージテストの最初と最後でそれぞれsetupとteardownが呼ばれるのがわかると思います。

これまでinitメソッドでsetupしてz_test.goなどでteardownさせるなどの黒魔術的なことが必要でしたがこれで解決ですね。

基礎からわかる Go言語

WEB+DB PRESS Vol.82