如何在 Golang 中使用表格驱动的测试?
表格驱动的测试(Table-Driven Testing)是一种软件测试技术,它使用数据表来为测试用例提供输入。这种方法可以帮助简化测试代码,并且更容易维护。
在本教程中,我们将向您展示如何在 Golang 中使用表格驱动的测试。
1. 创建数据表
首先,您需要创建一个数据表包含要测试的输入和预期输出。例如,对于一个计算平方根的函数,您的数据表可能如下所示:
| 输入 | 预期输出 |
| :---------- | :------ |
| 4 | 2 |
| 9 | 3 |
| 16 | 4 |
2. 定义测试函数
接下来,您需要定义一个测试函数并使用表格驱动进行参数化。在 Golang 中,您可以使用 testing.T 和 subtests 库来编写表格驱动的测试。
立即学习“go语言免费学习笔记(深入)”;
以下是使用 testing.T 的示例:
func TestSquareRoot(t *testing.T) {
tests := []struct {
input float64
expected float64
}{
{4, 2},
{9, 3},
{16, 4},
}
for _, tt := range tests {
t.Run(fmt.Sprintf("input_%g", tt.input), func(t *testing.T) {
result := math.Sqrt(tt.input)
if result != tt.expected {
t.Errorf("Expected %f, got %f", tt.expected, result)
}
})
}
}
3. 运行测试
最后,您可以使用 go test 命令运行测试。确保在当前目录中包含一个名为 main.go 的文件,该文件中包含您的测试函数。
go test
实战案例
以下是使用表格驱动的测试进行单元测试的一步一步实战示例:
- 创建一个 main.go 文件并编写一个 Sum 函数,该函数将两个数字相加。
package main
import "fmt"
func Sum(a, b int) int {
return a + b
}
func main() {
fmt.Println(Sum(1, 2)) // 输出:3
}
- 在同一个文件中添加一个测试函数 TestSum。
import (
"testing"
)
func TestSum(t *testing.T) {
tests := []struct {
a, b int
expected int
}{
{1, 2, 3},
{3, 4, 7},
{5, 6, 11},
}
for _, tt := range tests {
t.Run(fmt.Sprintf("input_%d+%d", tt.a, tt.b), func(t *testing.T) {
result := Sum(tt.a, tt.b)
if result != tt.expected {
t.Errorf("Expected %d, got %d", tt.expected, result)
}
})
}
}
- 运行测试:
go test
如果所有测试通过,您将看到以下输出:
PASS
ok _ 0.008s