Go 中全局 init() 函数会在包导入时自动执行，导致测试前无法设置环境变量；正确做法是使用 TestMain 在所有测试运行前统一配置环境，并在结束后清理，确保测试可重复、隔离且可靠。
go 中全局 `init()` 函数会在包导入时自动执行，导致测试前无法设置环境变量；正确做法是使用 `testmain` 在所有测试运行前统一配置环境，并在结束后清理，确保测试可重复、隔离且可靠。
在 Go 项目中，常借助 init() 函数完成一次性初始化（如加载环境变量），但这也给单元测试带来挑战：init() 在任何测试函数执行前即已运行，无法被测试用例控制或重置。你遇到的问题正是典型表现——测试中通过 os.Setenv() 设置的环境变量，晚于 init() 执行，因此 env.Parse(&envConfig) 实际读取的是空值或旧值，导致断言失败。
✅ 正确解法：使用 TestMain
Go 提供了 func TestMain(m *testing.M) 作为测试生命周期的入口钩子。它在所有测试开始前执行，且可完全控制测试流程（包括环境准备与清理），是替代“测试前手动调用 init”的标准、推荐方式。
以下是一个完整、健壮的测试示例：
// config_test.go
package main
import (
"os"
"testing"
"github.com/caarlos0/env"
"github.com/stretchr/testify/assert"
)
type envVars struct {
Host string `env:"APP_HOST"`
Username string `env:"APP_USERNAME"`
Password string `env:"APP_PASSWORD"`
}
var envConfig envVars
func init() {
if err := env.Parse(&envConfig); err != nil {
// 注意：生产代码中建议记录错误后 panic，避免静默失败
panic("failed to parse env: " + err.Error())
}
}
func TestMain(m *testing.M) {
// ? 测试前：设置所需环境变量（必须在 m.Run() 之前）
os.Setenv("APP_HOST", "http://localhost:9999")
os.Setenv("APP_USERNAME", "john")
os.Setenv("APP_PASSWORD", "doe")
// 运行全部测试
exitCode := m.Run()
// ? 测试后：清理环境变量（防止污染其他测试或后续执行）
os.Unsetenv("APP_HOST")
os.Unsetenv("APP_USERNAME")
os.Unsetenv("APP_PASSWORD")
os.Exit(exitCode)
}
func TestEnvConfig(t *testing.T) {
assert.NotNil(t, &envConfig)
assert.Equal(t, "http://localhost:9999", envConfig.Host)
assert.Equal(t, "john", envConfig.Username)
assert.Equal(t, "doe", envConfig.Password)
}
⚠️ 关键注意事项
- TestMain 是包级唯一入口：每个测试包最多定义一个 TestMain，否则编译报错；
- 环境变量需提前设置：务必在 m.Run() 调用前完成 os.Setenv()，否则 init() 已执行完毕；
- 务必清理环境：使用 os.Unsetenv() 避免跨测试干扰（尤其在并行测试或 CI 环境中）；
- 避免在 init() 中调用不可控逻辑：如网络请求、文件 I/O 或依赖外部状态的操作——这会使测试更难隔离；
- 替代方案思考：长期来看，更推荐将配置加载逻辑从 init() 移出，改为显式调用（如 LoadConfig()），便于注入 mock、控制时机和提升可测性。
通过 TestMain，你既尊重了 Go 的初始化机制，又获得了对测试环境的完全掌控力——这是编写可靠、可维护 Go 单元测试的关键实践之一。
