Go言語のテストでタイムアウト・並行処理を扱うポイント

先生と生徒の会話形式で理解しよう

生徒

「Go言語で並行処理を使ったコードをテストしたいんですが、どうやってテストすればいいですか？」

先生

「Go言語では、goroutineという軽量スレッドを使って並行処理を実装します。テストする場合はタイムアウトを設定して安全に確認できます。」

生徒

「タイムアウトって何ですか？」

先生

「タイムアウトとは、一定時間内に処理が終わらなかった場合に強制的に終了させる仕組みです。無限ループや遅い処理をテストする時に便利です。」

生徒

「具体的にどのように書くんですか？」

先生

「それでは、基本的な使い方を見ていきましょう！」

1. 並行処理の基本

Go言語では、goroutineを使うと簡単に並行処理が書けます。goroutineとは、通常の関数呼び出しに「go」をつけるだけで実行される軽量スレッドのことです。


go func() {
    fmt.Println("並行処理の例")
}()

このコードでは、関数が別スレッドで実行されます。

2. タイムアウトを使ったテスト

テストで並行処理を扱うと、処理が終わらない場合があります。そこでcontextパッケージやtimeパッケージを使ってタイムアウトを設定します。


func TestWithTimeout(t *testing.T) {
    done := make(chan bool)
    
    go func() {
        // 長い処理の例
        time.Sleep(2 * time.Second)
        done <- true
    }()
    
    select {
    case <-done:
        t.Log("処理が正常に終了")
    case <-time.After(3 * time.Second):
        t.Error("タイムアウト")
    }
}

time.Afterで指定した時間を超えるとタイムアウトとしてテストを失敗させられます。

3. 複数のgoroutineをテストする場合

複数の並行処理を扱う場合は、sync.WaitGroupを使うと便利です。WaitGroupは全てのgoroutineが終わるまで待機する仕組みです。


func TestMultipleGoroutines(t *testing.T) {
    var wg sync.WaitGroup
    wg.Add(2)
    
    go func() {
        defer wg.Done()
        time.Sleep(1 * time.Second)
    }()
    
    go func() {
        defer wg.Done()
        time.Sleep(2 * time.Second)
    }()
    
    done := make(chan bool)
    go func() {
        wg.Wait()
        done <- true
    }()
    
    select {
    case <-done:
        t.Log("全てのgoroutineが終了")
    case <-time.After(3 * time.Second):
        t.Error("タイムアウト")
    }
}

これで複数の並行処理が正しく終了するかをテストできます。