如何使用Golang testing进行单元测试_验证函数和方法正确性

Go单元测试需以Test开头、*testing.T为参数,用t.Error等报告失败;测试文件名以_test.go结尾且同包;支持表驱动、子测试、错误验证、接口mock等。

用 Go 的 testing 包写单元测试很简单,核心是写以 Test 开头、参数为 *testing.T 的函数,然后用 t.Errort.Errorft.Fatal 报告失败。

快速上手:写第一个测试函数

Go 测试文件必须以 _test.go 结尾,且与被测代码在同一包(通常同目录)。例如,你有一个 add.go

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

对应写 add_test.go

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

运行测试:go test(当前目录)或 go test -v 查看详细输出。

用表驱动测试覆盖多种输入

避免重复写多个 if 判断,把测试用例组织成结构体切片,遍历执行:

func TestAdd(t *testing.T) {
    tests := []struct {
        name     string
        a, b     int
        expected int
    }{
        {"positive", 2, 3, 5},
        {"negative", -1, -1, -2},
        {"zero", 0, 0, 0},
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            result := Add(tt.a, tt.b)
            if result != tt.expected {
                t.Errorf("got %d, want %d", result, tt.expected)
            }
        })
    }
}

t.Run 支持子测试,让输出更清晰,也方便单独运行某个用例(如 go test -run=TestAdd/positive)。

验证错误和边界情况

不仅要测“正常路径”,还要测 panic、错误返回、空值、越界等。例如函数返回 error

func Divide(a, b float64) (float64, error) {
    if b == 0 {
        return 0, errors.New("division by zero")
    }
    return a / b, nil
}

对应测试:

  • 成功场景:检查结果值是否符合预期
  • 错误场景:用 if err == nil 判断是否该出错却没出错,或用 if !errors.Is(err, expectedErr) 校验错误类型
  • 推荐用 require(需引入 github.com/stretchr/testify/require)简化断言,比如 require.NoError(t, err)require.EqualError(t, err, "division by zero")

模拟依赖与接口隔离

真实项目中函数常依赖外部(数据库、HTTP、时间等)。Go 鼓励通过接口抽象依赖,再在测试中传入 mock 实现。例如:

type Clock interface {
    Now() time.Time
}
func FormatTime(clock Clock) string {
    return clock.Now().Format("2006-01-02")
}

测试时可构造一个固定时间的 mock:

type mockClock struct{ t time.Time }
func (m mockClock) Now() time.Time { return m.t }

func TestFormatTime(t *testing.T) {
    c := mockClock{time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC)}
    got := FormatTime(c)
    if got != "2025-01-02" {
        t.Errorf("got %s", got)
    }
}

这样既不依赖系统时钟,又保持测试确定性和速度。