unit-testing - 有没有办法动态设置测试超时值

Question

目前我正在使用以下命令运行我的测试，并在测试调用期间给定超时值。

go test myModule -run TestSanity -v --race -timeout 10h

Golang 测试模块中有没有办法在程序执行期间设置它。类似的东西，

func TestMain(m *testing.M) {
    // customTimeout = "10h"
    // m.Timeout(customTimeout)   <--- Something like this
    code := m.Run()
    os.Exit(code)
}

Answer 1

您可以编写自己的函数来执行此操作:

func panicOnTimeout(d time.Duration) {
    <-time.After(d)
    panic("Test timed out")
}

func TestMain(m *testing.M) {
    go panicOnTimeout(10 * time.Hour) // custom timeout

    code := m.Run()
    os.Exit(code)
}

这应该模拟 go test -timeout 的作用。请务必传递 -timeout 0 以防止触发默认测试超时。