如何在 Golang 中使用表格驱动的测试？

如何在 Golang 中使用表格驱动的测试？

表格驱动的测试（Table-Driven Testing）是一种软件测试技术，它使用数据表来为测试用例提供输入。这种方法可以帮助简化测试代码，并且更容易维护。

在本教程中，我们将向您展示如何在 Golang 中使用表格驱动的测试。

1. 创建数据表

首先，您需要创建一个数据表包含要测试的输入和预期输出。例如，对于一个计算平方根的函数，您的数据表可能如下所示：

| 输入 | 预期输出 |
| :---------- | :------ |
| 4 | 2 |
| 9 | 3 |
| 16 | 4 |

2. 定义测试函数

接下来，您需要定义一个测试函数并使用表格驱动进行参数化。在 Golang 中，您可以使用 testing.T 和 subtests 库来编写表格驱动的测试。

以下是使用 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

实战案例

以下是使用表格驱动的测试进行单元测试的一步一步实战示例：

1. 创建一个 main.go 文件并编写一个 Sum 函数，该函数将两个数字相加。

package main

import "fmt"

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

func main() {
    fmt.Println(Sum(1, 2)) // 输出：3
}

2. 在同一个文件中添加一个测试函数 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)
            }
        })
    }
}

3. 运行测试：

go test

如果所有测试通过，您将看到以下输出：

PASS
ok      _     0.008s