01介绍
我们使用 Golang 语言开发的项目,怎么保证逻辑正确和性能要求呢?也就是说我们如何测试我们的 Golang 代码呢?在 Golang 语言中,可以使用标准库 testing 包编写单元测试和基准测试,使用 go test 命令执行单元测试和基准测试的代码。本文我们介绍在 Golang 语言中怎么编写测试代码。
02命名规范
在 Golang 语言中编写测试代码,需要遵循一些命名规范,包含文件名、包名、函数(方法)名和变量名。
文件名和包名
测试文件名以 _test.go 结尾,go test 工具可以遍历以 _test.go 结尾的文件,执行测试函数。而 go build 和 go run 会忽略以 _test.go 结尾的文件,文件名开头一般是被测试函数所在的文件名。
包名一般和被测试文件的包名相同,这样即可以测试被测试文件的可导出函数和不可导出函数。
函数名和方法名
测试函数(方法)名必须以 Test、Benchmark 和 Example 开头,并且必须是可导出函数。函数名一般是被测试函数名,首字母大写。如果我们需要给同一个函数编写多个测试函数,可以在函数名后接上测试函数的场景,例如:TestXxxxXxxx。
变量名
测试函数(方法)的变量名,Golang 语言和 go test 工具没有明确的约束,但是,社区针对输出结果有一些规范供大家参考。在编写单元测试代码时,一般会得到一个实际输出结果,和一个我们预期的输出结果做对比。针对这两个变量,社区的变量名规范是 got/want 或 expected/actual。
03编写测试代码
单元测试
所谓单元测试,顾名思义就是对单元进行测试,一般进行测试的单元是一个最小的单元,在 Golang 语言中,最小的单元就是指一个函数或方法。
单元测试的函数,函数名以 Test 开头,例如:TestXxx。参数必须是 *testing.T 类型,可以使用该类型的方法记录测试信息和测试状态。例如,一般使用 Log 和 Logf 记录测试信息,使用 Error、Errorf、Fatal 和 Fatalf 方法记录测试状态,该类型的更多方法可以阅读官方文档。
被测试函数:
- func Sum(a, b int) int {
- return a+b
- }
测试函数:
- func TestSum(t *testing.T) {
- a, b := 1,2
- rst := Sum(a, b)
- if rst == 3 {
- t.Logf("expected=%d, actual=%d", 3, rst)
- } else {
- // t.Errorf("expected=%d, actual=%d", 3, rst)
- t.Fatalf("expected=%d, actual=%d", 3, rst)
- }
- t.Log("done")
- }
阅读上面这段代码,是我们编写的 Sum 函数的单元测试,给定 a, b 两个变量作为 Sum 函数的输入参数,此外,我们还可以使用表格测试发,给定一组被测试函数的输入参数,限于篇幅,本文不准备花费篇幅介绍。
使用 go test 命令执行以上单元测试的代码:
- go test
- PASS
- ok learn_go/lesson27 0.555s
go test 命令遍历所有 _test.go 结尾的文件,执行文件中所有的测试函数。此外,go test 支持一些参数,例如,-v 输出测试函数的运行详情;-run 指定执行的测试函数;-count 指定执行次数。
此外,使用参数 --coverprofile 统计单元测试的覆盖率。
- go test --coverprofile=func.cover
- PASS
- coverage: 100.0% of statements
- ok learn_go/lesson27 0.499s
阅读上面的执行结果,可以发现我们编写的单元测试覆盖率为 100%。
如果我们想要查看详细的覆盖率统计结果,我们可以执行以下命令生成 html 文件,使用浏览器打开生成的 html 文件,可以查看详细的单元测试覆盖率统计结果。
- go tool cover -html=func.cover -o func_cover.html
运行以上命令,会生成一个名为 func_cover.html 的文件,我们可以使用浏览器打开它,查看详细的单元测试覆盖率统计结果。
基准测试
在 Golang 语言中,可以使用基准测试查看代码的性能。基准测试的函数名以 Benchmark 开头,例如:BenchmarkXxx。参数必须是 *testing.B 类型,函数体中 for 循环的条件,以 b.N 作为循环次数,它是基准测试框架提供的,它在 Golang 运行时动态调整,通过多次测试,得到性能评估结果。
示例代码:
- func BenchmarkSum(b *testing.B) {
- for i := 0; i < b.N; i++ {
- Sum(1, 2)
- }
- }
我们可以使用 go test 工具执行以上基准测试的代码,基准测试函数不会自动执行,必须使用参数 -bench。
- go test -bench=".*"
- goos: darwin
- goarch: amd64
- pkg: learn_go/lesson27
- cpu: Intel(R) Core(TM) i9-9880H CPU @ 2.30GHz
- BenchmarkSum-16 1000000000 0.2325 ns/op
- PASS
- ok learn_go/lesson27 0.748s
阅读上面的执行结果,我们主要介绍一下 BenchmarkXxx-n 这一行的意思。这一行共有三列,第一列 BenchmarkSum-16 分别代表基准测试的函数名和参与基准测试的 CPU 线程数,默认是 GOMAXPROCS 的值。第二列 1000000000 表示基准测试循环执行的次数。第三列 0.2325 ns/op 表示每次循环的平均执行耗时是 0.2325 纳秒,该值越小,说明代码性能越高。
除了 b.N 之外,还有几个关于性能测试时间计数的方法,例如:b.ResetTimer()、b.StopTimer() 和 b.StartTimer(),我们可以根据我们的测试场景,灵活使用。
此外,go test 工具关于基准测试的参数,除了参数 -bench 之外,还有 -benchmem 统计内存分配;-cpu 指定参与执行基准测试的 CPU 线程数;-benchtime 指定测试时间和循环次数,其中值的单位为 s 表示指定执行多少秒,单位为 x 表示指定循环执行次数;-timeout 指定基准测试函数执行的超时时间。
04总结
本文我们介绍怎么编写测试代码,包含单元测试和基准测试。特别需要注意的是一些命名规范。
养成编写测试代码的习惯,不仅可以降低代码逻辑的错误率,而且在多人开发中,还可以提升联调效率和提测通过率。
本文转载自微信公众号「Golang语言开发栈」,可以通过以下二维码关注。转载本文请联系Golang语言开发栈公众号。