如何在 Go 中测试 os.exit 场景

Question

鉴于此代码func doomed() {  os.Exit(1)}我如何正确测试调用此函数将导致使用退出go test？这需要在一组测试中发生，换句话说，os.Exit()调用不会影响其他测试，应该被捕获。

Answer 1

我通过使用bouk/monkey来做到这一点：

func TestDoomed(t *testing.T) {

  fakeExit := func(int) {

    panic("os.Exit called")      

  }

  patch := monkey.Patch(os.Exit, fakeExit)

  defer patch.Unpatch()

  assert.PanicsWithValue(t, "os.Exit called", doomed, "os.Exit was not called")

}

在处理这类工作、故障注入和其他困难任务时，monkey 是超级强大的。它确实带有一些警告。

Answer 2

我认为如果os.Exit不从外部（使用exec.Command）过程模拟测试，就无法测试实际情况。

也就是说，您可以通过创建接口或函数类型，然后在测试中使用 noop 实现来实现您的目标：

package main

import "os"

import "fmt"

type exiter func (code int)

func main() {

    doExit(func(code int){})

    fmt.Println("got here")

    doExit(func(code int){ os.Exit(code)})

}

func doExit(exit exiter) {

    exit(1)

}