Go testing 怎样为子测试设置独立环境变量

Go 为子测试设置临时环境变量，可以在 t.Run 的回调中调用该子测试的 t.Setenv(key, value)。它设置的是进程环境变量，并在当前测试清理阶段恢复原值；原来不存在的变量会恢复为未设置。所谓“独立”是串行子测试的设置与恢复范围独立，不是获得了一份并行私有环境。

官方文档：https://pkg.go.dev/testing#T.Setenv

Go 1.17 带来的是自动恢复，不是进程环境副本

T.Setenv 从 Go 1.17 开始提供，适用于测试直接读取环境配置的逻辑。它把设置和 Cleanup 绑定，让临时值的所有权落在具体测试上。这里介绍的是已有标准库能力，不是声称当前版本新增加了子测试环境隔离。

例如一个配置函数读取 ENVCASE_STAGE，blue、green 和空字符串三个用例应分别得到各自值，并且不能把值留给后续用例。用 Setenv 可以简化清理，但前提是子测试串行，且没有无约束的后台任务同时依赖这一环境变量。

能力引入说明：https://go.dev/doc/go1.17

手动设置的问题，常出在恢复时机与恢复对象

单独调用 os.Setenv，不会自动恢复旧值。若用 os.Getenv 保存旧值，还会丢失“原来不存在”与“原来设置为空”的区别。Setenv 的实现通过 LookupEnv 保存这两种状态，再注册相应清理动作。

另一个容易忽略的点是 t 的归属。将值设置在父测试 t 上，清理就属于父测试；即使调用发生在子测试回调里，也不会自动变成子测试级恢复。下面是错误归属的示意片段，缺少完整测试上下文，不是可独立运行程序：

// 此片段只展示错误的清理归属，不要作为推荐写法运行。
parentT := t
parentT.Run("blue", func(childT *testing.T) {
    parentT.Setenv(stageKey, "blue") // 恢复绑定父测试，不是当前子测试。
    _ = childT
})

正确写法是使用 childT.Setenv，或者像下一节那样让回调参数明确成为当前的 t。原有自定义环境 helper 迁移时，同样要检查传入的是哪一层 testing.TB，而不只做函数名替换。

完整示例：每个子测试结束后恢复父值

把下面内容保存为独立目录中的 env_test.go。代码包含被测读取函数与完整测试；无须创建生产配置文件，也不需要任何真实凭据。本文未运行示例，下面的断言描述的是预期行为。

package envcase

import (
    "os"
    "testing"
)

const stageKey = "ENVCASE_STAGE"

// currentStage 每次调用读取环境，便于测试不同配置值。
func currentStage() string {
    value, ok := os.LookupEnv(stageKey)
    if !ok {
        return "default"
    }
    return value // 空字符串与未设置是两种状态。
}

func TestStageSubtests(t *testing.T) {
    // 父测试为自己设置基线；结束时再由父测试恢复原始环境。
    t.Setenv(stageKey, "parent")
    cases := []struct {
        name  string
        value string
    }{
        {"blue", "blue"},
        {"green", "green"},
        {"empty", ""},
    }
    for _, tc := range cases {
        tc := tc // 兼容旧Go循环变量语义，示例不启动并行子测试。
        t.Run(tc.name, func(t *testing.T) {
            // 使用当前子测试的t，清理归属当前子测试。
            t.Setenv(stageKey, tc.value)
            if got := currentStage(); got != tc.value {
                t.Fatalf("子测试值=%q，期望=%q", got, tc.value)
            }
        })
        // 串行子测试完成清理后，父测试环境应已恢复。
        if got, ok := os.LookupEnv(stageKey); !ok || got != "parent" {
            t.Fatalf("父测试环境未恢复，值=%q，存在=%t", got, ok)
        }
    }
}

父测试先把环境设为 parent，子测试临时设为自己的值，子测试清理后父测试检查 parent 是否恢复。父测试自身结束时，它的 Setenv 清理才恢复测试前的原始状态。这里的层次是资源归属，不是值永远被固定为 parent。

empty 用例验证的是“已设置为空”：LookupEnv 返回存在，函数返回空字符串。它不测试“变量不存在”的分支，因为 Setenv(key, "") 不是 Unsetenv。测试未设置状态需要另外定义清理策略或采用可注入的配置读取设计，不能把两种输入混为一谈。

不能给这组测试直接加 Parallel

官方约束是：Setenv 影响整个进程，不能用于并行测试，也不能用于有并行祖先的测试。即使子测试本身没有调用 Parallel，只要父级测试已经进入并行模式，也不满足使用条件。

反过来，在同一个测试调用 Setenv 后再调用 Parallel，也不能靠调换顺序变成合法组合。用手写 os.Setenv 绕开 testing 的限制，不会让进程环境变成测试私有，只会失去这层明确的误用提示。

如果确实需要并行覆盖大量配置组合，可以将业务逻辑改为接收配置值，或在适合的集成测试中使用独立子进程环境；那是另一个隔离层次，不是 t.Setenv 提供的功能。当前示例不创建子进程、不运行网络服务。

实现参考：https://go.dev/src/testing/testing.go

设置时机还必须早于被测代码读取配置

如果配置在包级变量初始化或 init 中读取一次，测试函数内的 Setenv 已经晚了：它不会重新运行包初始化，也不会清空 sync.Once 或应用自建缓存。迁移前应区分“每次调用读取环境”与“启动时快照”两类设计。

示例 currentStage 明确每次读取，适合在每个子测试中设置后立即调用。若被测代码启动 goroutine，还要在用例结束前等待它完成并停止；子测试级 Cleanup 不会替你接管一个逃逸到后续用例的后台任务。

让读者复现值覆盖与恢复断言

在新的练习目录准备好 env_test.go 后，可以自行初始化模块并运行测试：

# 仅在新的练习目录初始化一次模块。
go mod init example.com/envcase
# 禁用测试结果缓存，多次执行这组串行用例。
go test -run '^TestStageSubtests$' -count=3 -v

预期三种用例都读到自己的值，每次子测试结束后都恢复 parent；若不符合，测试中的 Fatalf 会提供值与存在状态。本文没有执行这些命令，不提供伪造 PASS 日志。示例使用的环境变量名也不是任何真实服务配置。

子测试 Fatal 后，还会恢复环境吗？

在 testing 正常管理的测试结束与清理机制下，会执行已注册 Cleanup。它不是针对进程被强制终止、os.Exit 或崩溃后仍能继续运行的保证。

同一个子测试能设置多个变量吗？

可以分别调用 Setenv，每次都有对应恢复动作。但作用域仍是进程，变量数量增加并不会改变并行限制。

为什么不用 defer 恢复父环境？

defer 绑定函数返回时机，Cleanup 绑定测试及其子测试的清理生命周期。需要测试级资源管理时，用正确层级的 t.Setenv 更直接；不能靠父函数的 defer 推断所有子测试资源已结束。

采用建议很明确：串行测试、按调用读取环境、设置只需持续到当前子测试结束时，使用该子测试的 Setenv；需要并行私有状态、重跑初始化或跨进程隔离时，应换用相应设计，而不是扩大 Setenv 的承诺。