Go单元测试如何在config包init方法执行前设置环境变量

问题根因 #

Go程序的初始化执行顺序固定为：包级变量初始化 → 包内init()函数执行 → 入口函数（测试场景下为TestMain）执行。你在测试文件中定义的TestMain运行时，config包的init()已经执行完成，此时再设置环境变量不会改变已经加载完成的全局配置值。

可行解决方案 #

方案1：懒加载（改动最小，最匹配现有需求） #

取消init()函数，把配置加载逻辑延后到第一次访问配置时执行，既保留首次使用自动加载的特性，又能适配测试场景：

// config.go 核心代码示例
package config

import "os"

var (
    PropertyX string
    PropertyY string
    PropertyZ string
    isLoaded  bool
)

// 私有方法，访问配置前先调用判断是否需要加载
func loadIfNeeded() {
    if !isLoaded {
        LoadSettings()
        isLoaded = true
    }
}

func LoadSettings() {
    PropertyX = os.Getenv("X")
    PropertyY = os.Getenv("Y")
    PropertyZ = os.Getenv("Z")
}

// 对外暴露配置访问方法，替换原来直接访问全局变量的逻辑
func GetX() string {
    loadIfNeeded()
    return PropertyX
}

func GetY() string {
    loadIfNeeded()
    return PropertyY
}

func GetZ() string {
    loadIfNeeded()
    return PropertyZ
}

测试时就可以先设置环境变量再访问配置，此时才会触发加载逻辑：

// config_test.go 示例
package config

import (
    "os"
    "testing"
)

func TestMain(m *testing.M) {
    // 先设置环境变量
    os.Setenv("X", "500")
    // 执行测试用例，首次访问配置时才会加载
    exitCode := m.Run()
    // 清理环境变量
    os.Unsetenv("X")
    os.Exit(exitCode)
}

func TestLoadConfig(t *testing.T) {
    x := GetX()
    if x != "500" {
        t.Fatalf("expected X = 500, got %s", x)
    }
}

方案2：执行测试命令时提前传入环境变量 #

如果不想修改现有业务代码，可以直接在执行测试命令时注入环境变量：

X=500 Y=test Z=dev go test ./config -v

该方案无需改动业务逻辑，缺点是测试配置和代码分离，不支持单个测试用例使用不同的环境变量值。

方案3：重构为依赖注入模式（长期可维护性更高） #

放弃全局配置变量+隐式初始化的设计，改成显式加载、传递配置实例的模式，完全解耦测试和环境依赖：

// config.go 示例
package config

import "os"

type Config struct {
    X string
    Y string
    Z string
}

// 生产环境加载逻辑
func Load() (*Config, error) {
    return &Config{
        X: os.Getenv("X"),
        Y: os.Getenv("Y"),
        Z: os.Getenv("Z"),
    }, nil
}

// 测试用构造方法，直接传参生成配置实例，不需要依赖环境变量
func NewForTest(x, y, z string) *Config {
    return &Config{X: x, Y: y, Z: z}
}

测试时可以直接生成需要的配置实例，不需要处理环境变量和初始化逻辑，业务代码启动时统一加载一次配置实例，传递给下游业务模块即可。

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