如何在 Go 中测试参数解析
考虑以下 Go 代码，它使用 flag 包解析命令行参数：
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
}为了测试这个函数，我们可以将一个参数传递给编译后的代码，如图所示下面：
./args -user=bla
这应该会产生输出：
Your username is "bla"
但是，如果我们尝试使用以下测试来测试此行为：
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)
}
}我们收到一个错误：
FAIL coverage: 87.5% of statements FAIL tool 0.008s
此失败的原因是简单地将 os.Args 设置为新的slice 不将参数传递给函数。为了解决这个问题，我们需要确保 os.Args 中的第一个值是可执行文件本身（的路径），如本修订后的测试所示：
func TestArgs(t *testing.T) {
expected := "bla"
oldArgs := os.Args
defer func() { os.Args = oldArgs }()
os.Args = []string{"cmd", "-user=bla"}
actual := passArguments()
if actual != expected {
t.Errorf("Test failed, expected: '%s', got: '%s'", expected, actual)
}
}通过推迟 os.Args 的恢复恢复到其原始状态，我们可以确保测试不会干扰其他测试，这些测试可能正在检查调用 go test 时传递的真实参数。
以上是如何在单元测试中正确测试Go的'flag”包参数解析？的详细内容。更多信息请关注PHP中文网其他相关文章！
