单元测试的必要性与基础
单元测试不仅是保障代码质量的手段，也是优秀的设计工具和文档形式，对软件开发具有重要意义。
另一种形式的文档：好的单元测试是一种活文档，能清晰展示代码单元的预期用途和行为，有时比注释更有用。
提升API设计品味：编写测试会促使开发者从使用者角度思考，推动设计出更简洁、易用的API。难以测试的代码往往是设计问题的早期信号，比如职责过多、耦合过紧或依赖关系混乱。
发现被忽略的角落：在编写单元测试时，可能会发现之前未考虑到的边界情况或特殊使用场景，从而完善代码。
Go 语言单元测试基础回顾
Go语言通过内置的 go test 命令和标准库中的 testing 包提供了原生的测试支持，我们无需像其他语言一样依赖复杂的第三方测试框架。
Go测试的"三板斧"：_test.go文件、TestXxx函数和*testing.T
Go语言的测试遵循一套简单而严格的约定，这使得 go test 工具能够自动发现并执行测试：
测试文件命名：测试代码必须放在以 _test.go 结尾的文件中。例如，如果你的业务代码在 calculator.go 里，那么测试代码就应该放在 calculator_test.go 中。go build 命令在编译时会自动忽略这些测试文件。
测试文件位置：按照惯例，测试文件通常与被测试的代码文件位于同一个包（同一个文件夹）内。
测试函数签名：测试函数必须以 Test 开头，并且后面跟一个首字母大写的名称（例如，TestMyFunction）。它只接受一个参数，即 t *testing.T。
*testing.T：这个 testing.T 类型的参数 t 是测试的“状态表示器”，它提供了一系列方法用于报告测试失败、记录日志、控制测试流程等。
*testing.T 的常用方法：
t.Logf() / t.Log(): 记录日志。当测试通过时，只有在执行 go test 时加上 -v 参数，这些日志才会显示。
t.Errorf() / t.Error(): 报告测试失败，但测试会继续执行。这对于希望一次性看到所有失败情况的场景很有用。
t.Fatalf() / t.Fatal(): 报告测试失败，并 立即停止 当前测试函数的执行。
t.Skipf() / t.Skip(): 跳过当前测试。
t.Run(): 运行子测试。这对于组织相关的测试用例非常方便，是实现表驱动测试的核心。
t.Helper(): 将一个函数标记为测试辅助函数。这样，当测试失败时，日志会显示调用该辅助函数的测试代码行号，而不是辅助函数内部的行号，极大地提升了调试效率。
t.Parallel(): 将一个测试标记为可并行执行。这对于加速大型测试套件的执行时间很有帮助，但使用者需要注意并发安全问题。
表驱动测试：高效组织测试用例
表驱动测试是 Go 社区推崇的一种清晰且可维护的测试方式。它并非工具或库，而是一种代码风格，通过将测试用例组织到一个“表”（通常是结构体切片）中，再用统一逻辑遍历执行这些用例。
为什么推荐表驱动测试？
减少重复代码：核心测试逻辑只需编写一次，即可通用于所有用例。
提高可读性：每个用例的输入、期望输出和描述都清晰地列在表中，一目了然。
易于维护：添加、修改或删除测试用例，只需在表中增删改一行即可。
清晰的失败信息：结合 t.Run() 为每个子测试命名，一旦有测试失败，能立刻知道是哪个用例挂了，以及具体的输入和期望输出是什么。
基本结构
定义一个结构体来描述你的测试用例。
type addTestCase struct {\ name string // 测试用例名称\ a, b int // 输入参数\ want int // 期望输出\ wantErr bool // 是否期望错误 (如果函数可能返回error)\}创建一个该结构体的切片，并填充各种测试用例。
在测试函数中，遍历这个切片，并为每个测试用例使用 t.Run 来创建一个子测试。
实战演练:表驱动测试的应用示例
假设我们需要测试项目中的 processNumbers 函数，它接收一个整数切片，并返回其中所有正数的和。
// 文件: calculator.gopackage main
// processNumbers 接收一个整数切片，并返回其中所有正数的和。// 如果切片为空，返回0。func processNumbers(numbers []int) int { sum := 0 for _, num := range numbers { if num > 0 { sum += num } } return sum}下面是为其编写的表驱动单元测试：
// 文件: calculator_test.gopackage main
import "testing"
func TestProcessNumbers_TableDriven(t *testing.T) {// 1. 定义测试用例结构体 tests := []struct { name string numbers []int want int // 期望正数和 }{ { name: "空切片应返回0", numbers: []int{}, want: 0, }, { name: "所有数字均为正数", numbers: []int{1, 2, 3, 4, 5}, want: 15, }, { name: "包含负数和零", numbers: []int{-1, 0, 10, -5, 20}, want: 30, }, { name: "所有数字均为负数或零", numbers: []int{-1, -2, 0}, want: 0, }, { name: "单个正数", numbers: []int{42}, want: 42, }, }
// 2. 遍历所有测试用例for _, tt := range tests {// 3. 使用 t.Run 创建子测试 t.Run(tt.name, func(t *testing.T) { got := processNumbers(tt.numbers)
// 4. 验证结果if got != tt.want { t.Errorf("processNumbers() = %v, want %v", got, tt.want) } }) }}
模拟 (Mocking)：
隔离"外部势力"，提升测试效率
单元测试的核心是“单元”，即隔离被测代码。但现实中，许多代码会依赖外部服务、数据库、文件系统或其他复杂组件。直接在测试中使用这些真实依赖会让测试变得缓慢、不稳定，甚至无法进行（比如待测试函数依赖于一个还没上线的服务接口）。
此时，“模拟（Mocking）”就派上了用场。Mocking 就是用一个行为可控的“替身”来取代那些真实依赖，让我们能专注于测试自己的代码逻辑。
为什么要 Mock？
隔离性 (Isolation)：确保测试只关注当前单元的行为，不受外部依赖变化的影响。
可控性 (Control)：可以精确控制 Mock 对象的行为，比如让它返回特定数据、模拟错误情况等。
速度 (Speed)：Mock 对象通常比真实依赖快得多，能显著提升测试执行效率。
确定性 (Determinism)：避免真实依赖可能带来的不确定性（如网络波动、数据变化等）。
如何 Mock?
Go社区有多种流行的 Mock 框架，例如：GoMock、Testify/mock、monkey、gomonkey。
在团队实践中，gomonkey是常用的 Mock 框架之一，下面主要介绍一下gomonkey 的主要用法：
ApplyFunc - 替换函数：用于替换一个普通函数的实现。
patches := gomonkey.ApplyFunc(time.Now, func() time.Time { return time.Date(2023, 10, 1, 12, 0, 0, 0, time.UTC)})defer patches.Reset() // 测试结束后恢复原函数ApplyMethod - 替换方法：用于替换一个结构体方法的实现。
patches := gomonkey.ApplyMethod(reflect.TypeOf(&someStruct{}), "MethodName", func(*someStruct, param1Type) returnType { // 模拟实现 return mockValue })defer patches.Reset()
最低0.47元/天 解锁文章
扫一扫
377
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
