package config import ( "os" "path/filepath" "testing" ) func TestLoadReadsCORSAllowedOrigins(t *testing.T) { path := filepath.Join(t.TempDir(), "config.yaml") content := []byte(`server: port: 8083 cors: allowedOrigins: - https://console.example.com - http://localhost:3000 `) if err := os.WriteFile(path, content, 0600); err != nil { t.Fatalf("WriteFile() error = %v", err) } cfg, err := Load(path) if err != nil { t.Fatalf("Load() error = %v", err) } got := cfg.Server.CORS.AllowedOrigins want := []string{"https://console.example.com", "http://localhost:3000"} if len(got) != len(want) { t.Fatalf("len(AllowedOrigins)=%d want %d", len(got), len(want)) } for i := range want { if got[i] != want[i] { t.Fatalf("AllowedOrigins[%d]=%q want %q", i, got[i], want[i]) } } } func TestLoadOverridesValuesFromEnvironment(t *testing.T) { path := filepath.Join(t.TempDir(), "config.yaml") content := []byte(`server: port: 8083 db: type: postgres dsn: postgres-dsn storage: local: baseUrl: /storage `) if err := os.WriteFile(path, content, 0600); err != nil { t.Fatalf("WriteFile() error = %v", err) } t.Setenv("AGENT_DESK_SERVER_PORT", "8090") t.Setenv("AGENT_DESK_DB_DSN", "postgres-override-dsn") t.Setenv("AGENT_DESK_STORAGE_LOCAL_BASEURL", "/files") cfg, err := Load(path) if err != nil { t.Fatalf("Load() error = %v", err) } if cfg.Server.Port != 8090 { t.Fatalf("Server.Port=%d want 8090", cfg.Server.Port) } if cfg.DB.Type != "postgres" { t.Fatalf("DB.Type=%q want postgres", cfg.DB.Type) } if cfg.DB.DSN != "postgres-override-dsn" { t.Fatalf("DB.DSN=%q want postgres-override-dsn", cfg.DB.DSN) } if cfg.Storage.Local.BaseURL != "/files" { t.Fatalf("Storage.Local.BaseURL=%q want /files", cfg.Storage.Local.BaseURL) } } func TestFromSettingsBuildsRuntimeConfig(t *testing.T) { cfg, err := FromSettings(map[string]string{ "ai_agent_language": "en-US", "ai_agent_storage_default": "oss", "ai_agent_storage_max_upload_size_mb": "64", "ai_agent_storage_oss_private": "true", "ai_vector_path": "/var/lib/agent-desk/vectors.db", "ai_agent_wxwork_notify_to_users": `[1,2,3]`, "ai_agent_wxwork_notify_dedupe_interval": "900", }) if err != nil { t.Fatalf("FromSettings() error = %v", err) } if cfg.Language != "en-US" || cfg.Storage.Default != "oss" { t.Fatalf("unexpected basic config: %+v", cfg) } if cfg.Storage.MaxUploadSizeMB != 64 || !cfg.Storage.OSS.Private { t.Fatalf("unexpected storage config: %+v", cfg.Storage) } if cfg.VectorDB.Path != "/var/lib/agent-desk/vectors.db" { t.Fatalf("VectorDB.Path=%q", cfg.VectorDB.Path) } if len(cfg.WxWork.Notify.ToUsers) != 3 || cfg.WxWork.Notify.DuplicateCheckInterval != 900 { t.Fatalf("unexpected WeCom notify config: %+v", cfg.WxWork.Notify) } } func TestFromSettingsRejectsInvalidValues(t *testing.T) { if _, err := FromSettings(map[string]string{ "ai_agent_wxwork_notify_dedupe_interval": "not-a-number", }); err == nil { t.Fatal("FromSettings() error = nil, want invalid setting error") } }