Goのテストを並列で実行する

はしがき

テストもコードベースが巨大になってくると、
時間が掛かるので並列処理したいヨネーって要望とか。
並列処理のベンチマークどないしよか的な事を
まとめてみました。

CIで並列実行してもらえばよくね?

とか言われると、そうですね感溢れるかもですが :P
最近、testの事ばかり調べてたので、メモがてら。

テストの並列実行

testing.T.Parallel() を呼び出したテストは並列に実行されます。
go test -parallel n(並列数)で指定すれば並列で実行してくれます。

試してみましょう。

package main

import (
    "testing"
    "time"
)

// 10秒かかるテスト
func Test10sSleep(t *testing.T) {
    t.Parallel()
    time.Sleep(10 * time.Second)
}

// 20秒かかるテスト
func Test20sSleep(t *testing.T) {
    t.Parallel()
    time.Sleep(20 * time.Second)
}

// 30秒かかるテスト
func Test30sSleep(t *testing.T) {
    t.Parallel()
    time.Sleep(30 * time.Second)
}

• 通常実行

$ go test ./...
ok      playground_/paralleltest    60.855s

• 3並列

$ go test ./... -parallel 3
ok      playground_/paralleltest    30.025s

並列テストの恩恵を受けてますね :)

注意点

並列処理なので、testAが走っている最中にtestBが走って相互に影響してコケたりします。
折角並列実行しても、成功するときもあれば失敗しちゃうよお ふぇぇぇぇ...
ってなってしまいます。
他のテストに影響を出すような物(例えばDB周りとか）はmock化するか、
並列実行を避けたほうが無難かなと。

テストしたい関数の並列実行時のベンチマークについて

testing.B.RunParallel が使えます。
SetParallelism(n int) で 設定値*GOMAXPROCS を反映する事できます。

• p.go

package main

import "time"

// 3秒止める処理
func DoAnyThing() {
    time.Sleep(3 * time.Second)
}

• p_test.go

package main

import "testing"

func BenchmarkSingle(b *testing.B) {
    for i := 0; i < b.N; i++ {
        DoAnyThing()
    }
}

func BenchmarkParallel(b *testing.B) {
    b.SetParallelism(5) // 私の実行環境の場合、GOMAXPROCS=4なので20並列
    b.ResetTimer()
    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() { 
            DoAnyThing()
        }
    })
}

• 結果

$ go test -bench . -benchmem -benchtime 30s -v
testing: warning: no tests to run
PASS
BenchmarkSingle-4         10    3001148727 ns/op          64 B/op          1 allocs/op
BenchmarkParallel-4      200     150105216 ns/op         108 B/op          1 allocs/op

20倍実行できてますね :)
GOMAXPROCは runtime.GOMAXPROCS(0) で取得できます。

参考

• testing
• runtime