Golang测试配置解析实例教程

type Config struct {
    Server struct {
        Host string `yaml:"host"`
        Port int    `yaml:"port"`
    } `yaml:"server"`
    Database struct {
        DSN string `yaml:"dsn"`
    } `yaml:"database"`
}

func LoadConfig(filename string) (*Config, error) {
    data, err := os.ReadFile(filename)
    if err != nil {
        return nil, err
    }
var cfg Config
err = yaml.Unmarshal(data, &cfg)
if err != nil {
    return nil, err
}
return &cfg, nil
}

var configPath = flag.String("config", "config.test.yaml", "path to config file")
func TestMain(m *testing.M) {
flag.Parse()
os.Exit(m.Run())
}
func TestLoadConfig(t testing.T) {
cfg, err := LoadConfig(configPath)
if err != nil {
t.Fatalf("failed to load config: %v", err)
}
if cfg.Server.Port == 0 {
    t.Error("expected server port > 0")
}
if cfg.Database.DSN == "" {
    t.Error("expected non-empty DSN")
}
}

go test -v
# 或指定特定配置
go test -v -config ./testdata/config.integration.yaml

func TestLoadConfig_InvalidYAML(t *testing.T) {
    tmpfile, cleanup := createTempConfig(t, "{ invalid: yaml [ }")
    defer cleanup()
_, err := LoadConfig(tmpfile)
if err == nil {
    t.Fatal("expected error for invalid YAML")
}
}
func createTempConfig(t testing.T, content string) (string, func()) {
tmpfile, err := os.CreateTemp("", "config-.yaml")
if err != nil {
t.Fatal(err)
}
if _, err := tmpfile.Write([]byte(content)); err != nil {
t.Fatal(err)
}
if err := tmpfile.Close(); err != nil {
t.Fatal(err)
}
return tmpfile.Name(), func() { os.Remove(tmpfile.Name()) }
}

func TestLoadConfig_ProdEnv(t *testing.T) {
    t.Setenv("ENV", "prod")
    // 假设 LoadConfig 根据 ENV 变量选择文件
    cfg, err := LoadConfigFromEnv()
    if err != nil {
        t.Fatalf("unexpected error: %v", err)
    }
    // 断言生产配置正确加载
    if cfg.Server.Host != "api.example.com" {
        t.Errorf("wrong host for prod: got %s", cfg.Server.Host)
    }
}