56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
|
package cli
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLoadConfigDefaults(t *testing.T) {
|
||
|
|
t.Setenv("HOME", t.TempDir())
|
||
|
|
t.Setenv("KMSVC_SERVER", "")
|
||
|
|
t.Setenv("KMSVC_TOKEN", "")
|
||
|
|
t.Setenv("KMSVC_OUTPUT", "")
|
||
|
|
|
||
|
|
cfg := LoadConfig()
|
||
|
|
if cfg.Output != "table" {
|
||
|
|
t.Errorf("Output = %q, want default %q", cfg.Output, "table")
|
||
|
|
}
|
||
|
|
if cfg.Server != "" || cfg.Token != "" {
|
||
|
|
t.Errorf("expected empty Server/Token with no file or env, got %+v", cfg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLoadConfigFileThenEnvOverride(t *testing.T) {
|
||
|
|
home := t.TempDir()
|
||
|
|
t.Setenv("HOME", home)
|
||
|
|
|
||
|
|
dir := filepath.Join(home, ".kmsvc")
|
||
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
content := "server: file-server:443\noutput: json\n"
|
||
|
|
if err := os.WriteFile(filepath.Join(dir, "config.yaml"), []byte(content), 0o644); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Setenv("KMSVC_SERVER", "")
|
||
|
|
t.Setenv("KMSVC_OUTPUT", "")
|
||
|
|
t.Setenv("KMSVC_TOKEN", "")
|
||
|
|
|
||
|
|
cfg := LoadConfig()
|
||
|
|
if cfg.Server != "file-server:443" || cfg.Output != "json" {
|
||
|
|
t.Fatalf("expected file values, got %+v", cfg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Env var must win over the file.
|
||
|
|
t.Setenv("KMSVC_SERVER", "env-server:443")
|
||
|
|
cfg = LoadConfig()
|
||
|
|
if cfg.Server != "env-server:443" {
|
||
|
|
t.Errorf("Server = %q, want env override %q", cfg.Server, "env-server:443")
|
||
|
|
}
|
||
|
|
if cfg.Output != "json" {
|
||
|
|
t.Errorf("Output = %q, want file value %q to survive (env unset)", cfg.Output, "json")
|
||
|
|
}
|
||
|
|
}
|