mlogclub
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
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: sqlite
|
|
dsn: file:./data/app.db?_busy_timeout=5000
|
|
storage:
|
|
local:
|
|
baseUrl: /storage
|
|
mcp:
|
|
servers:
|
|
system:
|
|
endpoint: http://127.0.0.1:8083/api/mcp
|
|
`)
|
|
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", "mysql-dsn")
|
|
t.Setenv("AGENT_DESK_STORAGE_LOCAL_BASEURL", "/files")
|
|
t.Setenv("AGENT_DESK_MCP_SERVERS_SYSTEM_ENDPOINT", "http://127.0.0.1:8090/api/mcp")
|
|
|
|
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 != "sqlite" {
|
|
t.Fatalf("DB.Type=%q want sqlite", cfg.DB.Type)
|
|
}
|
|
if cfg.DB.DSN != "mysql-dsn" {
|
|
t.Fatalf("DB.DSN=%q want mysql-dsn", cfg.DB.DSN)
|
|
}
|
|
if cfg.Storage.Local.BaseURL != "/files" {
|
|
t.Fatalf("Storage.Local.BaseURL=%q want /files", cfg.Storage.Local.BaseURL)
|
|
}
|
|
if cfg.MCP.Servers["system"].Endpoint != "http://127.0.0.1:8090/api/mcp" {
|
|
t.Fatalf("MCP system endpoint=%q", cfg.MCP.Servers["system"].Endpoint)
|
|
}
|
|
}
|