使用 go testing 包进行单元测试:编写一个以 test 开头的测试函数,不带参数,返回零值或错误值。为测试表定义输入值和预期输出的集合。使用 for 循环遍历测试表,对每个输入集运行测试。使用 go test -cover 命令运行单元测试,生成覆盖率报告。
使用 Golang 内置 testing 包编写单元测试
简介
Go 标准库提供了 testing 包,用于方便编写和运行单元测试。本文将介绍如何使用 testing 包编写单元测试,并展示一个实战案例。
立即学习“go语言免费学习笔记(深入)”;
基本结构
一个单元测试函数需要满足以下要求:
- 以 Test 开头,后跟要测试的函数名称
- 在函数签名中不接受任何参数
- 返回零值或错误值
编写单元测试
- 导入 testing 包
import "testing"
- 编写测试函数
以下是一个示例测试函数,它测试 Sum 函数是否可以正确计算两个数字的和:
func TestSum(t *testing.T) {
result := Sum(1, 2)
if result != 3 {
t.Errorf("Sum(1, 2) = %d, want 3", result)
}
}
- 定义测试表
测试表是输入值和预期输出的集合。它可以帮助编写涵盖各种情况的测试。
var tests = []struct {
a, b int
want int
}{
{1, 2, 3},
{3, 4, 7},
}
- 循环遍历测试表
使用 for 循环遍历测试表并对每个输入集运行测试:
func TestSumTable(t *testing.T) {
for _, test := range tests {
result := Sum(test.a, test.b)
if result != test.want {
t.Errorf("Sum(%d, %d) = %d, want %d", test.a, test.b, result, test.want)
}
}
}
实战案例
考虑一个计算圆面积的 Area 函数:
func Area(radius float64) float64 {
return math.Pi * radius * radius
}
以下是如何编写单元测试来验证 Area 函数:
import "testing"
import "math"
func TestArea(t *testing.T) {
tests := []struct {
radius float64
want float64
}{
{1.0, math.Pi},
{2.0, 4 * math.Pi},
}
for _, test := range tests {
result := Area(test.radius)
if result != test.want {
t.Errorf("Area(%f) = %f, want %f", test.radius, result, test.want)
}
}
}
运行单元测试
要运行单元测试,请使用 go test 命令:
go test -cover
该命令将运行所有以 Test 开头的函数,并生成覆盖率报告。