Go测试超时后defer与t.Cleanup不执行的解决办法咨询

当Go测试超出-timeout指定的时间（或默认10分钟）时，测试进程会被直接终止，导致defer语句和t.Cleanup注册的清理逻辑无法执行。除了禁用超时（-timeout 0），还有以下几种可行方案：

1. 手动实现超时控制，主动触发清理 #

放弃依赖go test的全局超时机制，在测试函数内部通过context或定时器实现超时逻辑，这样能在超时触发时先执行清理操作，再退出测试。

示例代码：

package timeout_test

import (
    "fmt"
    "testing"
    "time"
    "context"
)

func TestTimeoutDeferCleanup(t *testing.T) {
    fmt.Println("This is printed.")
    defer fmt.Println("This is printed now.")
    t.Cleanup(func() {
        fmt.Println("This is also printed now.")
    })

    // 手动设置1秒超时
    ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
    defer cancel() // 确保上下文被正确取消

    select {
    case <-time.After(2 * time.Second):
        fmt.Println("Test completed normally (won't trigger here)")
    case <-ctx.Done():
        t.Fatalf("Test timed out: %v", ctx.Err())
    }
}

运行时无需指定-timeout（或设置一个比内部超时更长的值），超时后defer和t.Cleanup都会正常执行。

2. 子goroutine执行耗时逻辑，主goroutine监控超时 #

如果测试中的耗时操作无法直接通过上下文控制，可以将其放到子goroutine中，主goroutine用定时器监控超时，超时后先执行清理再终止测试。

示例代码：

package timeout_test

import (
    "fmt"
    "testing"
    "time"
)

func TestTimeoutDeferCleanup(t *testing.T) {
    fmt.Println("This is printed.")
    defer fmt.Println("This is printed now.")
    cleanupDone := make(chan struct{})
    t.Cleanup(func() {
        fmt.Println("This is also printed now.")
        close(cleanupDone)
    })

    // 子goroutine执行耗时操作
    taskDone := make(chan struct{})
    go func() {
        time.Sleep(2 * time.Second)
        close(taskDone)
    }()

    // 监控超时
    select {
    case <-taskDone:
        fmt.Println("Test finished normally")
    case <-time.After(1 * time.Second):
        t.Fatal("Test timed out")
    }
    <-cleanupDone // 可选：确保清理逻辑执行完成
}

3. 全局测试套件统一管理超时与清理 #

如果多个测试用例共享需要清理的资源，可以通过testing.M编写测试套件的入口逻辑，将超时控制和全局清理逻辑放在一起，确保即使单个测试超时，全局资源也能被正确清理。这种方式更适合全局资源的管理，单个测试的场景还是前两种方案更直接。

内容的提问来源于stack exchange，提问作者dwvldg