Billy D.
f1dd96a42b
All checks were successful
CI / Test (push) Successful in 3m2s
CI / Lint (push) Successful in 3m7s
CI / Release (push) Successful in 1m55s
CI / Notify Downstream (stt-module) (push) Successful in 1s
CI / Notify Downstream (voice-assistant) (push) Successful in 1s
CI / Notify (push) Successful in 2s
CI / Notify Downstream (chat-handler) (push) Successful in 1s
CI / Notify Downstream (pipeline-bridge) (push) Successful in 1s
CI / Notify Downstream (tts-module) (push) Successful in 1s
124 lines
3.0 KiB
Go
124 lines
3.0 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoadDefaults(t *testing.T) {
|
|
s := Load()
|
|
if s.ServiceName != "handler" {
|
|
t.Errorf("expected default ServiceName 'handler', got %q", s.ServiceName)
|
|
}
|
|
if s.HealthPort != 8080 {
|
|
t.Errorf("expected default HealthPort 8080, got %d", s.HealthPort)
|
|
}
|
|
if s.HTTPTimeout != 60*time.Second {
|
|
t.Errorf("expected default HTTPTimeout 60s, got %v", s.HTTPTimeout)
|
|
}
|
|
}
|
|
|
|
func TestLoadFromEnv(t *testing.T) {
|
|
t.Setenv("SERVICE_NAME", "test-svc")
|
|
t.Setenv("HEALTH_PORT", "9090")
|
|
t.Setenv("OTEL_ENABLED", "false")
|
|
|
|
s := Load()
|
|
if s.ServiceName != "test-svc" {
|
|
t.Errorf("expected ServiceName 'test-svc', got %q", s.ServiceName)
|
|
}
|
|
if s.HealthPort != 9090 {
|
|
t.Errorf("expected HealthPort 9090, got %d", s.HealthPort)
|
|
}
|
|
if s.OTELEnabled {
|
|
t.Error("expected OTELEnabled false")
|
|
}
|
|
}
|
|
|
|
func TestURLGetters(t *testing.T) {
|
|
s := Load()
|
|
if s.EmbeddingsURL() == "" {
|
|
t.Error("EmbeddingsURL should have a default")
|
|
}
|
|
if s.RerankerURL() == "" {
|
|
t.Error("RerankerURL should have a default")
|
|
}
|
|
if s.LLMURL() == "" {
|
|
t.Error("LLMURL should have a default")
|
|
}
|
|
if s.TTSURL() == "" {
|
|
t.Error("TTSURL should have a default")
|
|
}
|
|
if s.STTURL() == "" {
|
|
t.Error("STTURL should have a default")
|
|
}
|
|
}
|
|
|
|
func TestURLGettersFromEnv(t *testing.T) {
|
|
t.Setenv("EMBEDDINGS_URL", "http://embed:8000")
|
|
t.Setenv("LLM_URL", "http://llm:9000")
|
|
|
|
s := Load()
|
|
if s.EmbeddingsURL() != "http://embed:8000" {
|
|
t.Errorf("expected custom EmbeddingsURL, got %q", s.EmbeddingsURL())
|
|
}
|
|
if s.LLMURL() != "http://llm:9000" {
|
|
t.Errorf("expected custom LLMURL, got %q", s.LLMURL())
|
|
}
|
|
}
|
|
|
|
func TestReloadFromSecrets(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
// Write initial secret files
|
|
writeSecret(t, dir, "embeddings-url", "http://old-embed:8000")
|
|
writeSecret(t, dir, "llm-url", "http://old-llm:9000")
|
|
|
|
s := Load()
|
|
s.SecretsPath = dir
|
|
s.reloadFromSecrets()
|
|
|
|
if s.EmbeddingsURL() != "http://old-embed:8000" {
|
|
t.Errorf("expected reloaded EmbeddingsURL, got %q", s.EmbeddingsURL())
|
|
}
|
|
if s.LLMURL() != "http://old-llm:9000" {
|
|
t.Errorf("expected reloaded LLMURL, got %q", s.LLMURL())
|
|
}
|
|
|
|
// Simulate secret update
|
|
writeSecret(t, dir, "embeddings-url", "http://new-embed:8000")
|
|
s.reloadFromSecrets()
|
|
|
|
if s.EmbeddingsURL() != "http://new-embed:8000" {
|
|
t.Errorf("expected updated EmbeddingsURL, got %q", s.EmbeddingsURL())
|
|
}
|
|
// LLM should remain unchanged
|
|
if s.LLMURL() != "http://old-llm:9000" {
|
|
t.Errorf("expected unchanged LLMURL, got %q", s.LLMURL())
|
|
}
|
|
}
|
|
|
|
func TestReloadFromSecretsNoPath(t *testing.T) {
|
|
s := Load()
|
|
s.SecretsPath = ""
|
|
// Should not panic
|
|
s.reloadFromSecrets()
|
|
}
|
|
|
|
func TestGetEnvDuration(t *testing.T) {
|
|
t.Setenv("TEST_DUR", "30")
|
|
d := getEnvDuration("TEST_DUR", 10*time.Second)
|
|
if d != 30*time.Second {
|
|
t.Errorf("expected 30s, got %v", d)
|
|
}
|
|
}
|
|
|
|
func writeSecret(t *testing.T, dir, name, value string) {
|
|
t.Helper()
|
|
if err := os.WriteFile(filepath.Join(dir, name), []byte(value), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|