3587d0ce27
Brings 5 packages from low/zero coverage to solid test suites: - pkg/lab: 0% → 100% (Store pub/sub, Config env loading) - pkg/session: 0% → 89.9% (transcript parser, HTML renderer, search, video) - pkg/io/sigil: 43.8% → 98.5% (XOR/ShuffleMask obfuscators, ChaCha20-Poly1305) - pkg/repos: 18.9% → 81.9% (registry, topo sort, directory scan, org detection) - pkg/plugin: 54.8% → 67.1% (installer error paths, Remove, registry Load/Save) Co-Authored-By: Virgil <virgil@lethean.io>
129 lines
3.7 KiB
Go
129 lines
3.7 KiB
Go
package lab
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
// ── LoadConfig defaults ────────────────────────────────────────────
|
|
|
|
func TestLoadConfig_Good_Defaults(t *testing.T) {
|
|
cfg := LoadConfig()
|
|
|
|
if cfg.Addr != ":8080" {
|
|
t.Fatalf("expected :8080, got %s", cfg.Addr)
|
|
}
|
|
if cfg.PrometheusURL != "http://prometheus:9090" {
|
|
t.Fatalf("unexpected PrometheusURL: %s", cfg.PrometheusURL)
|
|
}
|
|
if cfg.PrometheusInterval != 15 {
|
|
t.Fatalf("expected 15, got %d", cfg.PrometheusInterval)
|
|
}
|
|
if cfg.ForgeURL != "https://forge.lthn.io" {
|
|
t.Fatalf("unexpected ForgeURL: %s", cfg.ForgeURL)
|
|
}
|
|
if cfg.ForgeInterval != 60 {
|
|
t.Fatalf("expected 60, got %d", cfg.ForgeInterval)
|
|
}
|
|
if cfg.HFAuthor != "lthn" {
|
|
t.Fatalf("expected lthn, got %s", cfg.HFAuthor)
|
|
}
|
|
if cfg.HFInterval != 300 {
|
|
t.Fatalf("expected 300, got %d", cfg.HFInterval)
|
|
}
|
|
if cfg.TrainingDataDir != "/data/training" {
|
|
t.Fatalf("unexpected TrainingDataDir: %s", cfg.TrainingDataDir)
|
|
}
|
|
if cfg.InfluxDB != "training" {
|
|
t.Fatalf("expected training, got %s", cfg.InfluxDB)
|
|
}
|
|
}
|
|
|
|
// ── env override ───────────────────────────────────────────────────
|
|
|
|
func TestLoadConfig_Good_EnvOverride(t *testing.T) {
|
|
os.Setenv("ADDR", ":9090")
|
|
os.Setenv("FORGE_URL", "https://forge.lthn.ai")
|
|
os.Setenv("HF_AUTHOR", "snider")
|
|
defer func() {
|
|
os.Unsetenv("ADDR")
|
|
os.Unsetenv("FORGE_URL")
|
|
os.Unsetenv("HF_AUTHOR")
|
|
}()
|
|
|
|
cfg := LoadConfig()
|
|
if cfg.Addr != ":9090" {
|
|
t.Fatalf("expected :9090, got %s", cfg.Addr)
|
|
}
|
|
if cfg.ForgeURL != "https://forge.lthn.ai" {
|
|
t.Fatalf("expected forge.lthn.ai, got %s", cfg.ForgeURL)
|
|
}
|
|
if cfg.HFAuthor != "snider" {
|
|
t.Fatalf("expected snider, got %s", cfg.HFAuthor)
|
|
}
|
|
}
|
|
|
|
// ── envInt ─────────────────────────────────────────────────────────
|
|
|
|
func TestLoadConfig_Good_IntEnvOverride(t *testing.T) {
|
|
os.Setenv("PROMETHEUS_INTERVAL", "30")
|
|
defer os.Unsetenv("PROMETHEUS_INTERVAL")
|
|
|
|
cfg := LoadConfig()
|
|
if cfg.PrometheusInterval != 30 {
|
|
t.Fatalf("expected 30, got %d", cfg.PrometheusInterval)
|
|
}
|
|
}
|
|
|
|
func TestLoadConfig_Bad_InvalidIntFallsBack(t *testing.T) {
|
|
os.Setenv("PROMETHEUS_INTERVAL", "not-a-number")
|
|
defer os.Unsetenv("PROMETHEUS_INTERVAL")
|
|
|
|
cfg := LoadConfig()
|
|
if cfg.PrometheusInterval != 15 {
|
|
t.Fatalf("expected fallback 15, got %d", cfg.PrometheusInterval)
|
|
}
|
|
}
|
|
|
|
// ── env / envInt helpers directly ──────────────────────────────────
|
|
|
|
func TestEnv_Good(t *testing.T) {
|
|
os.Setenv("TEST_LAB_KEY", "hello")
|
|
defer os.Unsetenv("TEST_LAB_KEY")
|
|
|
|
if got := env("TEST_LAB_KEY", "default"); got != "hello" {
|
|
t.Fatalf("expected hello, got %s", got)
|
|
}
|
|
}
|
|
|
|
func TestEnv_Good_Fallback(t *testing.T) {
|
|
os.Unsetenv("TEST_LAB_MISSING")
|
|
if got := env("TEST_LAB_MISSING", "fallback"); got != "fallback" {
|
|
t.Fatalf("expected fallback, got %s", got)
|
|
}
|
|
}
|
|
|
|
func TestEnvInt_Good(t *testing.T) {
|
|
os.Setenv("TEST_LAB_INT", "42")
|
|
defer os.Unsetenv("TEST_LAB_INT")
|
|
|
|
if got := envInt("TEST_LAB_INT", 0); got != 42 {
|
|
t.Fatalf("expected 42, got %d", got)
|
|
}
|
|
}
|
|
|
|
func TestEnvInt_Bad_Fallback(t *testing.T) {
|
|
os.Unsetenv("TEST_LAB_INT_MISSING")
|
|
if got := envInt("TEST_LAB_INT_MISSING", 99); got != 99 {
|
|
t.Fatalf("expected 99, got %d", got)
|
|
}
|
|
}
|
|
|
|
func TestEnvInt_Bad_InvalidString(t *testing.T) {
|
|
os.Setenv("TEST_LAB_INT_BAD", "xyz")
|
|
defer os.Unsetenv("TEST_LAB_INT_BAD")
|
|
|
|
if got := envInt("TEST_LAB_INT_BAD", 7); got != 7 {
|
|
t.Fatalf("expected fallback 7, got %d", got)
|
|
}
|
|
}
|