Go 单元测试处理全局变量

在Golang中测试包含全局变量的代码时，需特别注意全局状态带来的测试污染和并发安全问题。

一、测试环境隔离

1. ​​Setup/Teardown模式​​
   每个测试用例执行前重置全局变量，避免状态污染：

   var globalConfig *Config
   
   func TestMain(m *testing.M) {
       setup()
       code := m.Run()
       teardown()
       os.Exit(code)
   }
   
   func setup() { 
       globalConfig = &Config{Env: "test"} // 初始化测试环境
   }
   
   func teardown() {
       globalConfig = nil // 清理全局状态
   }
   

2. ​​并行测试隔离​​
   使用t.Parallel()时需配合深拷贝或独立初始化：

   func TestService(t *testing.T) {
       t.Run("subtest", func(t *testing.T) {
           original := globalConfig
           globalConfig = &Config{Env: "subtest"} // 创建副本
           defer func() { globalConfig = original }()
           // 测试逻辑
       })
   }
   

二、并发安全测试

1. ​​竞争条件检测​​
   使用-race参数强制检测数据竞争：

   go test -race ./...
   

2. ​​锁机制验证​​
   测试包含锁操作的全局变量时，需验证锁的生效范围：

   func TestConcurrentAccess(t *testing.T) {
       var wg sync.WaitGroup
       for i := 0; i < 100; i++ {
           wg.Add(1)
           go func() {
               defer wg.Done()
               // 触发需要加锁的全局变量操作
           }()
       }
       wg.Wait()
       assert.Equal(t, expectedValue, globalVar) 
   }
   

三、状态有效性验证

1. ​​初始化检查​​
   验证全局变量是否被正确初始化：

   func TestGlobalVarInit(t *testing.T) {
       require.NotNil(t, globalDB, "数据库连接未初始化")
       assert.IsType(t, &sql.DB{}, globalDB)
   }
   

2. ​​值范围校验​​
   对数值型全局变量进行边界检查：

   func TestThresholdValidation(t *testing.T) {
       assert.Greater(t, MaxConnections, 0, "连接数必须为正数")
       assert.LessOrEqual(t, MaxRetries, 5, "重试次数超限")
   }
   

四、依赖解耦策略

1. ​​依赖注入模式​​
   通过接口替代直接访问全局变量：

   type ConfigProvider interface {
       GetConfig() *Config
   }
   
   var configProvider ConfigProvider = &defaultProvider{}
   
   func TestConfigOverride(t *testing.T) {
       oldProvider := configProvider
       configProvider = &mockProvider{Config: testConfig}
       defer func() { configProvider = oldProvider }()
       // 执行测试逻辑
   }
   

2. ​​环境变量模拟​​
   测试环境相关的全局变量时，使用t.Setenv临时修改：

   func TestEnvConfig(t *testing.T) {
       t.Setenv("API_ENDPOINT", "http://localhost:8080")
       ReloadConfig() // 重新加载全局配置
       assert.Contains(t, globalConfig.APIEndpoint, "localhost")
   }
   

五、调试与监控

1. ​​expvar集成测试​​
   验证通过expvar暴露的全局变量：

   func TestExpvarMetrics(t *testing.T) {
       resp := httptest.NewRecorder()
       expvar.Handler().ServeHTTP(resp, httptest.NewRequest("GET", "/debug/vars", nil))
       vars := make(map[string]interface{})
       json.Unmarshal(resp.Body.Bytes(), &vars)
       assert.Greater(t, vars["request_count"], 0)
   }
   

2. ​​日志追踪​​
   在全局变量修改处添加日志断言：

   func TestConfigChangeLog(t *testing.T) {
       logOutput := captureLogs(func() {
           UpdateConfig(newConfig)
       })
       assert.Contains(t, logOutput, "配置已更新")
   }
   

关键注意事项

1. ​​避免测试间干扰​​：全局变量必须在每个测试用例执行前重置到已知状态
2. ​​禁止:=误操作​​：在测试函数中修改全局变量时，必须使用=赋值而非:=声明
3. ​​原子操作验证​​：对sync/atomic操作的全局变量，需测试中断场景下的原子性
4. ​​文档化全局状态​​：为每个全局变量添加注释说明其测试依赖关系

通过结合状态隔离、并发验证、依赖解耦等策略，可有效测试包含全局变量的代码模块。建议配合testify断言库和-coverprofile参数实现高覆盖率验证。