如何测试使用可变参数列表的 Golang 函数

在 Golang 中，可变参数列表允许函数接受任意数量的参数。这在需要在函数中处理数量不确定的值时非常有用。然而，测试使用可变参数列表的函数可能会有些棘手。

编写可变参数列表函数

让我们以一个计算切片中所有元素之和的函数为例：

立即学习“go语言免费学习笔记（深入）”；

func Sum(nums ...int) int {
    sum := 0
    for _, num := range nums {
        sum += num
    }
    return sum
}

编写测试

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

为了测试 Sum 函数，我们可以编写一个单元测试：

import (
    "testing"
)

func TestSum(t *testing.T) {
    tests := []struct {
        input []int
        want  int
    }{
        {[]int{1, 2, 3}, 6},
        {[]int{4, 5, 6}, 15},
        {[]int{}, 0},
    }

    for _, test := range tests {
        got := Sum(test.input...)
        if got != test.want {
            t.Errorf("Sum(%v) = %d, want %d", test.input, got, test.want)
        }
    }
}

在 TestSum 函数中：

• 我们定义了一个 tests 切片，其中包含了不同的输入值和期望输出。
• 我们使用 ... 展开运算符将 input 切片传递给 Sum 函数，作为可变参数列表。
• 我们断言 got（实际输出）等于 want（期望输出）。

实战案例

下面是一个使用可变参数列表函数的实际示例：

import (
    "fmt"
)

func PrintValues(values ...interface{}) {
    for _, value := range values {
        fmt.Println(value)
    }
}

func main() {
    PrintValues(1, "hello", 3.14)
}

在 main 函数中，我们调用 PrintValues 函数并传递不同的值。该函数将打印每个值。

输出：

1
hello
3.14