如何使用Golang表格驱动测试多输入

在Golang的单元测试场景中，当我们需要验证同一个函数在多组不同输入下的返回结果是否符合预期时，表格驱动测试是一种非常高效的实现方式，它通过结构化的用例定义和循环执行逻辑，避免重复的测试代码编写。

Golang表格驱动测试的核心思路

表格驱动测试的核心是把所有测试用例整理成一个结构体切片，每个结构体元素包含一组测试的输入参数和对应的预期结果，之后通过一个循环遍历这个切片，依次执行每一组用例的验证逻辑。这种方式不需要为每一个用例单独写一个测试函数或者重复的测试逻辑，当需要新增用例时，只需要在切片中新增一个结构体元素即可。

基础示例：实现加法函数的表格驱动测试

首先我们定义一个简单的加法函数，后续为这个函数编写表格驱动测试：

package main

// Add 两个整数相加的函数
func Add(a, b int) int {
    return a + b
}

接下来编写对应的测试文件，文件名通常为原文件名称加_test后缀，比如main_test.go：

package main

import "testing"

func TestAdd(t *testing.T) {
    // 定义测试用例表格，每个元素包含用例名称、输入参数a、输入参数b、预期结果
    testCases := []struct {
        name     string
        a        int
        b        int
        expected int
    }{
        {name: "两个正数相加", a: 1, b: 2, expected: 3},
        {name: "正数加负数", a: 5, b: -3, expected: 2},
        {name: "两个负数相加", a: -1, b: -2, expected: -3},
        {name: "加零", a: 10, b: 0, expected: 10},
    }

    // 遍历所有测试用例
    for _, tc := range testCases {
        t.Run(tc.name, func(t *testing.T) {
            result := Add(tc.a, tc.b)
            if result != tc.expected {
                t.Errorf("用例 %s 执行失败，输入 a=%d, b=%d，预期结果 %d，实际结果 %d", tc.name, tc.a, tc.b, tc.expected, result)
            }
        })
    }
}

在上面的代码中，我们使用了t.Run方法为每个用例设置独立的子测试名称，这样在测试执行结果中可以清晰看到每个用例的执行情况，方便定位问题。

复杂场景的表格驱动测试

当测试的函数逻辑更复杂，或者需要验证错误返回时，表格驱动测试同样适用。比如我们定义一个除法函数，需要验证正常除法和除数为0的异常情况：

package main

import "errors"

// Divide 除法运算，除数为0时返回错误
func Divide(a, b int) (int, error) {
    if b == 0 {
        return 0, errors.New("除数不能为0")
    }
    return a / b, nil
}

对应的表格驱动测试代码如下：

package main

import "testing"

func TestDivide(t *testing.T) {
    testCases := []struct {
        name        string
        a           int
        b           int
        expectedRes int
        expectedErr error
    }{
        {name: "正常除法", a: 6, b: 2, expectedRes: 3, expectedErr: nil},
        {name: "除数为0", a: 5, b: 0, expectedRes: 0, expectedErr: errors.New("除数不能为0")},
        {name: "结果为负数", a: 5, b: -2, expectedRes: -2, expectedErr: nil},
    }

    for _, tc := range testCases {
        t.Run(tc.name, func(t *testing.T) {
            res, err := Divide(tc.a, tc.b)
            // 验证错误是否符合预期
            if (err != nil && tc.expectedErr == nil) || (err == nil && tc.expectedErr != nil) {
                t.Errorf("用例 %s 错误返回不符合预期，预期错误 %v，实际错误 %v", tc.name, tc.expectedErr, err)
                return
            }
            if err != nil && err.Error() != tc.expectedErr.Error() {
                t.Errorf("用例 %s 错误信息不符合预期，预期错误 %v，实际错误 %v", tc.name, tc.expectedErr, err)
                return
            }
            // 验证结果是否符合预期
            if res != tc.expectedRes {
                t.Errorf("用例 %s 结果不符合预期，预期结果 %d，实际结果 %d", tc.name, tc.expectedRes, res)
            }
        })
    }
}

表格驱动测试的优势

• 减少重复代码：所有用例共享同一套执行和验证逻辑，不需要为每个用例编写重复的代码
• 用例扩展方便：新增测试用例只需要在表格切片中新增元素，不需要修改测试逻辑
• 用例结构清晰：所有用例的输入和预期结果集中展示，可读性和维护性更强
• 支持子测试：配合t.Run可以单独运行某一个用例，方便调试

注意事项

在编写表格驱动测试时需要注意，如果测试用例中包含指针类型或者需要修改的变量，要避免循环变量的闭包问题，不过在Golang 1.22及之后的版本中，循环变量的作用域已经调整为每次迭代独立，之前的版本如果需要兼容，可以在循环内把循环变量赋值给一个新的局部变量。另外用例的名称尽量清晰，能够直观体现用例的测试场景，方便后续排查问题。

Golangtable_driven_testtestcaseunit_test修改时间：2026-06-20 16:51:20