Go中如何巧用表驱动测试

2024-02-02 03:50:53

Go中的表驱动测试

表驱动测试是一种软件测试方法，它使用一个表格来定义测试用例，表格中的每一行都是一个测试用例。表驱动测试的优点在于它可以快速地创建和执行大量测试用例，并且可以很容易地维护和更新。

在Go中，可以使用标准库中的testing包来编写表驱动测试。testing包提供了许多用于编写和运行测试的函数，其中包括用于表驱动测试的T.Run()函数。

T.Run()函数接受两个参数：第一个参数是测试用例的名称，第二个参数是一个函数，该函数接受一个*testing.T类型的参数，用于编写测试用例。

例如，以下代码演示了如何使用T.Run()函数编写一个表驱动测试：

func TestFindAnagrams(t *testing.T) {
    tests := []struct {
        input    string
        expected []string
    }{
        {
            input:    "hello",
            expected: []string{"olleh"},
        },
        {
            input:    "world",
            expected: []string{"drow"},
        },
        {
            input:    "",
            expected: []string{},
        },
    }

    for _, test := range tests {
        t.Run(test.input, func(t *testing.T) {
            actual := findAnagrams(test.input)
            if !reflect.DeepEqual(actual, test.expected) {
                t.Errorf("findAnagrams(%s) = %v, want %v", test.input, actual, test.expected)
            }
        })
    }
}

上面的代码定义了一个名为TestFindAnagrams的测试函数，该函数接受一个*testing.T类型的参数t。测试函数首先定义了一个名为tests的切片，该切片包含了多个测试用例。每个测试用例都是一个struct，struct中包含了input和expected两个字段。input字段是测试用例的输入，expected字段是测试用例的预期输出。

接下来，测试函数使用T.Run()函数来遍历tests切片中的每个测试用例。对于每个测试用例，T.Run()函数都会创建一个新的子测试，子测试的名称是测试用例的input字段。在子测试中，测试函数调用findAnagrams()函数来计算测试用例的实际输出，并将其与测试用例的预期输出进行比较。如果实际输出与预期输出不一致，则测试函数会调用t.Errorf()函数来报告错误。