Golang表格驱动测试写法及示例详解

表格驱动测试通过将用例组织成表格提升代码可读性和扩展性。其核心在于集中管理测试用例，使用循环执行相同逻辑。1. 定义包含name、input、expected等字段的测试表格；2. 编写测试函数遍历表格，调用t.Run()执行子测试；3. 对复杂场景可增加setup、teardown或结构体输入；4. 注意其局限性，如处理复杂逻辑时灵活性不足及大表格影响可读性。

表格驱动测试，简单来说，就是把测试用例组织成表格（实际上是数据结构，通常是切片或映射），然后用循环遍历这些用例，针对每个用例执行相同的测试逻辑。 这样做的好处是简洁、可读性强，而且方便扩展新的测试用例。

表格驱动测试示例

表格驱动测试是一种在 Go 语言中编写单元测试的常用方法，它通过将测试用例组织成表格的形式，使得测试代码更加清晰、易于维护和扩展。

为什么选择表格驱动测试？

表格驱动测试能让你的测试代码更易读。 想象一下，如果每个测试用例都散落在不同的测试函数里，代码会显得非常冗长。表格驱动测试将它们集中起来，一目了然。它也让添加新测试用例变得非常简单，只需在表格中添加一行即可。代码复用率也更高，避免了重复编写相似的测试逻辑。

package mypackage

import "testing"

func TestMyFunction(t *testing.T) {
    // 定义测试用例表格
    var tests = []struct {
        name     string
        input    int
        expected int
    }{
        {"Positive", 5, 25},
        {"Negative", -5, 25},
        {"Zero", 0, 0},
    }

    // 循环遍历测试用例
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            actual := MyFunction(tt.input)
            if actual != tt.expected {
                t.Errorf("Test Name: %s, Input: %d, Expected: %d, Actual: %d", tt.name, tt.input, tt.expected, actual)
            }
        })
    }
}

func MyFunction(n int) int {
    return n * n
}

如何编写一个有效的表格驱动测试？

首先，明确你的测试目标。 确定要测试的函数或方法，并分析其输入和输出。 接下来，设计测试用例表格。 表格通常包含以下几列：

• name: 测试用例的名称，方便识别和调试。
• input: 测试用例的输入数据。
• expected: 测试用例的预期输出结果。

然后，编写测试函数，循环遍历测试用例表格，并针对每个用例执行测试逻辑。 使用 t.Run() 创建子测试，可以更清晰地组织测试结果。

如何处理复杂的测试场景？

对于复杂的测试场景，表格驱动测试同样适用。 你可以在测试用例表格中添加更多的列，例如：

• setup: 在测试用例执行前需要执行的准备工作。
• teardown: 在测试用例执行后需要执行的清理工作。
• error: 预期返回的错误信息。

或者，如果输入数据非常复杂，可以考虑使用结构体来表示输入数据，并在表格中使用结构体类型的字段。 甚至，如果需要模拟一些外部依赖，可以使用 mock 对象，并在测试用例表格中配置 mock 对象的行为。

表格驱动测试的局限性

虽然表格驱动测试有很多优点，但它也有一些局限性。 例如，对于非常复杂的测试逻辑，表格驱动测试可能会显得不够灵活。 此外，如果测试用例表格非常大，可能会影响测试代码的可读性。 在这种情况下，可以考虑将测试用例表格拆分成多个小的表格，或者使用其他测试方法。