89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
const baseConfig = `
|
|
app:
|
|
name: test-app
|
|
env: development
|
|
server:
|
|
host: 127.0.0.1
|
|
port: 8080
|
|
database:
|
|
host: 127.0.0.1
|
|
port: 3306
|
|
name: base_db
|
|
user: base_user
|
|
password: base_password
|
|
auth:
|
|
jwt_secret: base-development-secret
|
|
seed:
|
|
admin_username: admin
|
|
admin_password: admin123
|
|
admin_display_name: Admin
|
|
`
|
|
|
|
func TestLoadProfileThenEnvironmentOverride(t *testing.T) {
|
|
dir := writeConfigs(t, `
|
|
app:
|
|
env: test
|
|
server:
|
|
port: 8081
|
|
database:
|
|
name: profile_db
|
|
auth:
|
|
jwt_secret: test-profile-secret
|
|
`)
|
|
t.Setenv("APP_DATABASE_NAME", "environment_db")
|
|
t.Setenv("APP_SERVER_PORT", "9090")
|
|
|
|
cfg, err := Load(LoadOptions{Environment: "test", ConfigDir: dir})
|
|
if err != nil {
|
|
t.Fatalf("Load() error = %v", err)
|
|
}
|
|
if cfg.App.Env != "test" || cfg.Database.Name != "environment_db" || cfg.Database.User != "base_user" || cfg.Server.Port != 9090 {
|
|
t.Fatalf("unexpected merged config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadProductionRequiresEnvironmentSecrets(t *testing.T) {
|
|
dir := writeConfigs(t, "app:\n env: production\n")
|
|
t.Setenv("APP_DATABASE_PASSWORD", "")
|
|
t.Setenv("APP_AUTH_JWT_SECRET", "")
|
|
|
|
_, err := Load(LoadOptions{Environment: "prod", ConfigDir: dir})
|
|
if err == nil || !strings.Contains(err.Error(), "APP_DATABASE_PASSWORD") {
|
|
t.Fatalf("Load() error = %v, want production password requirement", err)
|
|
}
|
|
|
|
t.Setenv("APP_DATABASE_PASSWORD", "production-password")
|
|
t.Setenv("APP_AUTH_JWT_SECRET", "production-jwt-secret")
|
|
cfg, err := Load(LoadOptions{Environment: "prod", ConfigDir: dir})
|
|
if err != nil {
|
|
t.Fatalf("Load() with environment secrets error = %v", err)
|
|
}
|
|
if cfg.Database.Password != "production-password" || cfg.Auth.JWTSecret != "production-jwt-secret" {
|
|
t.Fatalf("environment secrets were not applied")
|
|
}
|
|
}
|
|
|
|
func writeConfigs(t *testing.T, profile string) string {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
if err := os.WriteFile(filepath.Join(dir, "config.yml"), []byte(baseConfig), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, "config.test.yml"), []byte(profile), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, "config.prod.yml"), []byte(profile), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return dir
|
|
}
|