Golang 如何编写单元测试_Golang 单元测试函数编写与执行示例

Go语言单元测试需遵循命名规范,测试文件以_test.go结尾,测试函数以Test开头并接收*testing.T参数;通过go test命令运行测试,推荐使用表驱动测试覆盖多场景,结合t.Run为子测试命名,提升可读性与维护性。

在 Go 语言中,编写单元测试非常简单且原生支持。Go 提供了 testing 包和 go test 命令,让你无需引入第三方框架即可完成测试的编写与执行。下面通过一个具体示例,说明如何编写和运行 Golang 单元测试。

单元测试文件命名规范

Go 要求测试文件以 _test.go 结尾,并与被测试的源文件放在同一目录下。例如,如果你有一个文件 calc.go,那么对应的测试文件应命名为 calc_test.go

编写测试函数

测试函数必须以 Test 开头,参数类型为 *testing.T。下面是一个简单的加法函数及其测试示例:

1. 源码文件:calc.go

package main

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

2. 测试文件:calc_test.go

package main

import "testing"

func TestAdd(t *testing.T) { result := Add(2, 3) expected := 5

if result != expected {
    t.Errorf("Add(2, 3) = %d; expected %d", result, expected)
}

}

在这个测试中,我们调用 Add(2, 3) 并验证结果是否等于 5。如果不符合预期,使用 t.Errorf 输出错误信息。

运行单元测试

在项目根目录下执行以下命令运行测试:

go test

输出结果:

ok      example.com/calc    0.001s

如果想查看更详细的输出,加上 -v 参数:

go test -v

输出:

=== RUN   TestAdd
--- PASS: TestAdd (0.00s)
PASS
ok      example.com/calc    0.001s

表驱动测试(Table-Driven Tests)

Go 推荐使用表驱动方式编写测试,便于覆盖多种输入场景。修改上面的测试如下:

func TestAdd(t *testing.T) {
    tests := []struct {
        name     string
        a, b     int
        expected int
    }{
        {"2+3=5", 2, 3, 5},
        {"0+0=0", 0, 0, 0},
        {"-1+1=0", -1, 1, 0},
        {"10+(-5)=5", 10, -5, 5},
    }
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 可以为每个子测试命名,便于定位失败用例。运行后输出会显示每个子测试的名称。

基本上就这些。Golang 的测试机制简洁高效,配合表驱动测试能写出清晰、可维护的测试代码。不复杂但容易忽略细节,比如命名规范和测试结构设计。掌握这些基础后,可以进一步学习性能测试(Benchmark)和代码覆盖率分析。