在 Go 中测试 os.Exit 场景
在编写涉及使用 os.Exit() 调用退出程序的函数的测试时，它有必要隔离它们对测试套件其余部分的影响。为了应对这一挑战，我们可以利用以下受 Go 团队核心成员 Andrew Gerrand 演讲启发的方法。
给定一个通过 os.Exit() 终止程序的函数：
package main
import (
"fmt"
"os"
)
func Crasher() {
fmt.Println("Going down in flames!")
os.Exit(1)
}创建相应的测试用例：
package main
import (
"os"
"os/exec"
"testing"
)
func TestCrasher(t *testing.T) {
// Check if the BE_CRASHER environment variable is set to 1.
if os.Getenv("BE_CRASHER") == "1" {
Crasher()
return
}
// Construct a command to re-run the test binary, limiting its execution to TestCrasher.
cmd := exec.Command(os.Args[0], "-test.run=TestCrasher")
// Set additional environment variables.
cmd.Env = append(os.Environ(), "BE_CRASHER=1")
// Execute the command.
err := cmd.Run()
// Verify the exit status of the command.
if e, ok := err.(*exec.ExitError); ok && !e.Success() {
return
}
// Report failure if the command ran with an unexpected exit code.
t.Fatalf("process ran with err %v, want exit status 1", err)
}此测试用例在单独的进程中重新调用 go test，将 TestCrasher 的执行与套件的其余部分隔离。它还设置一个环境变量 (BE_CRASHER=1)，调用的进程会检查该环境变量，如果存在，则调用被测试的函数。因此，我们避免无限循环并确保验证正确的退出代码。
以上是如何在 Go 中测试调用 os.Exit() 的函数？的详细内容。更多信息请关注PHP中文网其他相关文章！
