config/config_test.go

package config_test

import (
	"os"
	"testing"
	"time"

	"apigo.cc/go/config"
)

type TestList struct {
	Name string
}

func (l *TestList) ConfigureBy(setting string) {
	l.Name = setting
}

type testConfType struct {
	Name     string
	Sets     []int
	List     map[string]*TestList
	Duration config.Duration
}

func TestConfig_ComplexStruct(t *testing.T) {
	os.Clearenv()
	// 测试复杂嵌套、Map、指针以及 Duration
	os.WriteFile("complex.json", []byte(`{
		"name": "test-config",
		"sets": [1, 2, 3],
		"list": {
			"aaa": {"name": "222"}
		},
		"duration": "100s"
	}`), 0644)
	defer os.Remove("complex.json")

	// 模拟 env 覆盖
	os.Setenv("LIST_BBB", "{\"name\":\"xxx\"}")

	var conf testConfType
	if err := config.Load(&conf, "complex"); err != nil {
		t.Errorf("Load failed: %v", err)
	}

	if conf.Name != "test-config" {
		t.Errorf("Name failed, got %s", conf.Name)
	}
	if len(conf.Sets) != 3 || conf.Sets[1] != 2 {
		t.Errorf("Sets failed, got %v", conf.Sets)
	}
	if conf.List == nil || conf.List["aaa"] == nil || conf.List["aaa"].Name != "222" {
		t.Errorf("Map aaa failed, got %v", conf.List["aaa"])
	}
	if conf.List == nil || conf.List["bbb"] == nil || conf.List["bbb"].Name != "xxx" {
		t.Errorf("Map bbb (env) failed, got %v", conf.List["bbb"])
	}
	if conf.Duration.TimeDuration() != 100*time.Second {
		t.Errorf("Duration failed, got %v", conf.Duration.TimeDuration())
	}
}

func TestConfig_EnvYamlWithPrefix(t *testing.T) {
	os.Clearenv()
	// Create env.yaml with a prefixed block
	os.WriteFile("env.yaml", []byte(`
service:
  app: "emp"
  listen: ":3000"
`), 0644)
	defer os.Remove("env.yaml")

	type ServiceConfig struct {
		App    string
		Listen string
	}

	var conf ServiceConfig
	if err := config.Load(&conf, "service"); err != nil {
		t.Errorf("Load failed: %v", err)
	}

	if conf.App != "emp" {
		t.Errorf("App mismatch: expected emp, got %s", conf.App)
	}
	if conf.Listen != ":3000" {
		t.Errorf("Listen mismatch: expected :3000, got %s", conf.Listen)
	}
}