如何有效测试 Golang 函数中的参数传递？

测试 Golang 中的参数传递

在 Golang 中，将参数传递给函数是提供输入的常见方式。测试参数传递的正确性对于确保代码的可靠性至关重要。

挑战

考虑以下代码片段：

package main

import (
    "flag"
    "fmt"
)

func main() {
    passArguments()
}

func passArguments() string {
    username := flag.String("user", "root", "Username for this server")
    flag.Parse()
    fmt.Printf("Your username is %q.", *username)

    usernameToString := *username
    return usernameToString
}

将参数 -user=bla 传递给该程序时，它会正确打印“Your username is 'bla'”。挑战在于创建一个测试来验证此行为，而无需每次手动构建和运行代码。

尝试

人们可能会尝试使用 os.Args在测试中提供参数：

package main

import (
    "os"
    "testing"
)

func TestArgs(t *testing.T) {
    expected := "bla"
    os.Args = []string{"-user=bla"}

    actual := passArguments()

    if actual != expected {
        t.Errorf("Test failed, expected: '%s', got:  '%s'", expected, actual)
    }
}

但是，运行此测试时，结果是：

Your username is "root".
Your username is "root".
--- FAIL: TestArgs (0.00s)
    args_test.go:15: Test failed, expected: 'bla', got:  'root'
FAIL
coverage: 87.5% of statements
FAIL    tool    0.008s

解决方案

问题是 os.Args 是一个全局变量，所以 ["args"] 的初始值被覆盖测试，影响程序的其他部分。为了解决这个问题，我们可以保留原始值并在测试后恢复它：

oldArgs := os.Args
defer func() { os.Args = oldArgs }()
os.Args = []string{"cmd", "-user=bla"}

// ...

以上是如何有效测试 Golang 函数中的参数传递？的详细内容。更多信息请关注PHP中文网其他相关文章！