Go中級 #6 テスティング — testing パッケージと table-driven

package math

func Add(a, b int) int {
	return a + b
}

package math

import "testing"

func TestAdd(t *testing.T) {
	got := Add(2, 3)
	want := 5
	if got != want {
		t.Errorf("Add(2,3) = %d, want %d", got, want)
	}
}

go test                  # 現在のパッケージ
go test ./...            # すべてのパッケージ
go test -v               # 詳細 (各テスト結果)
go test -run TestAdd     # 特定のテストのみ

func TestSomething(t *testing.T) {
	if !condition1 {
		t.Errorf("条件 1 失敗")    // 失敗を記録 + 次の行も実行
	}
	if !condition2 {
		t.Fatalf("条件 2 失敗")    // 失敗を記録 + 即時終了
	}
	// ...
}

func TestAdd(t *testing.T) {
	tests := []struct {
		name string
		a, b int
		want int
	}{
		{"正の数", 2, 3, 5},
		{"負の数", -1, -2, -3},
		{"混合", -5, 10, 5},
		{"ゼロ", 0, 0, 0},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			got := Add(tc.a, tc.b)
			if got != tc.want {
				t.Errorf("Add(%d, %d) = %d, want %d", tc.a, tc.b, got, tc.want)
			}
		})
	}
}

go test -run "TestAdd/正の数"

func assertEqual(t *testing.T, got, want int) {
	t.Helper()    // この関数の呼び出し元の位置を表示
	if got != want {
		t.Errorf("got %d, want %d", got, want)
	}
}

func TestAdd(t *testing.T) {
	assertEqual(t, Add(2, 3), 5)
}

func TestWithDB(t *testing.T) {
	db := setupDB(t)
	t.Cleanup(func() {
		db.Close()
	})

	// テスト本文
}

func TestFileWrite(t *testing.T) {
	dir := t.TempDir()    // 自動的に整理される一時ディレクトリ

	path := filepath.Join(dir, "test.txt")
	os.WriteFile(path, []byte("hi"), 0644)

	// テストが終わると dir 自動削除
}

// math/add_test.go
package math_test    // _test サフィックス

import (
	"testing"
	"path/to/math"   // 外部から import するように
)

func TestAdd(t *testing.T) {
	got := math.Add(2, 3)
	// ...
}

func BenchmarkAdd(b *testing.B) {
	for i := 0; i < b.N; i++ {
		Add(2, 3)
	}
}

go test -bench=.
go test -bench=. -benchmem    # メモリ割り当ても測定

BenchmarkAdd-8          1000000000      0.30 ns/op

func BenchmarkParse(b *testing.B) {
	data := generateBigData()    // セットアップ — 測定から除外
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		Parse(data)
	}
}

func ExampleAdd() {
	fmt.Println(Add(2, 3))
	// Output: 5
}

type UserStore interface {
	GetUser(id string) (*User, error)
}

type Service struct {
	store UserStore
}

// 実際の実装
type DBStore struct{ /* ... */ }
func (s *DBStore) GetUser(id string) (*User, error) { /* DB 検索 */ }

// テスト用 fake
type fakeStore struct {
	users map[string]*User
}
func (s *fakeStore) GetUser(id string) (*User, error) {
	if u, ok := s.users[id]; ok {
		return u, nil
	}
	return nil, ErrNotFound
}

// テスト
func TestService(t *testing.T) {
	store := &fakeStore{
		users: map[string]*User{"u1": {Name: "カーティス"}},
	}
	svc := &Service{store: store}

	user, err := svc.LookupUser("u1")
	// ...
}

import (
	"net/http"
	"net/http/httptest"
)

func TestHandler(t *testing.T) {
	handler := http.HandlerFunc(myHandler)
	server := httptest.NewServer(handler)
	defer server.Close()

	resp, err := http.Get(server.URL + "/api/users")
	// 検証
}

go test -cover
go test -coverprofile=cover.out
go tool cover -html=cover.out      # HTML で可視化

import "github.com/stretchr/testify/assert"

func TestSomething(t *testing.T) {
	got := Add(2, 3)
	assert.Equal(t, 5, got)
	assert.NoError(t, err)
	assert.NotNil(t, result)
}