getenv在单元测试中的模拟

在单元测试中，我们通常需要模拟环境变量，以便在不影响实际环境的情况下测试代码

package main

import (
	"fmt"
	"os"
)

func main() {
	// 设置一个环境变量
	os.Setenv("MY_ENV_VAR", "test_value")

	// 获取环境变量的值
	value := os.Getenv("MY_ENV_VAR")

	// 输出环境变量的值
	fmt.Println("MY_ENV_VAR:", value)
}

在单元测试中，我们可以使用os.Setenv和os.Unsetenv函数来模拟环境变量。这里是一个使用Go语言编写的单元测试示例：

package main

import (
	"os"
	"testing"
)

func TestGetEnv(t *testing.T) {
	// 设置一个环境变量
	os.Setenv("MY_ENV_VAR", "test_value")

	// 获取环境变量的值
	value := os.Getenv("MY_ENV_VAR")

	// 检查环境变量的值是否正确
	if value != "test_value" {
		t.Errorf("Expected MY_ENV_VAR to be 'test_value', but got '%s'", value)
	}

	// 删除环境变量
	os.Unsetenv("MY_ENV_VAR")

	// 再次获取环境变量的值
	value = os.Getenv("MY_ENV_VAR")

	// 检查环境变量是否已被删除
	if value != "" {
		t.Errorf("Expected MY_ENV_VAR to be empty, but got '%s'", value)
	}
}

在这个示例中，我们首先使用os.Setenv设置了一个环境变量，然后使用os.Getenv获取其值并进行检查。接下来，我们使用os.Unsetenv删除环境变量，并再次获取其值以确保它已被删除。这样，我们就可以在不影响实际环境的情况下测试代码。