60 lines
2.0 KiB
Go
60 lines
2.0 KiB
Go
package config
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestLoadOllamaPoolAndGLPIKB(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("OLLAMA_URLS", "http://gpu-a:11434,http://gpu-b:11434")
|
|
t.Setenv("OLLAMA_NODE_NAMES", "a,b")
|
|
t.Setenv("OLLAMA_NODE_WEIGHTS", "1,3")
|
|
t.Setenv("OLLAMA_ROUTING_MODE", "weighted")
|
|
t.Setenv("GLPI_KB_ENABLED", "true")
|
|
t.Setenv("GLPI_URL", "http://glpi.test")
|
|
t.Setenv("GLPI_ALLOW_INSECURE_HTTP", "true")
|
|
t.Setenv("GLPI_CLIENT_ID", "id")
|
|
t.Setenv("GLPI_CLIENT_SECRET", "secret")
|
|
t.Setenv("GLPI_USERNAME", "user")
|
|
t.Setenv("GLPI_PASSWORD", "password")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(cfg.OllamaURLs) != 2 || cfg.OllamaNodeNames[1] != "b" || cfg.OllamaNodeWeights[1] != 3 || cfg.OllamaRoutingMode != "weighted" {
|
|
t.Fatalf("unexpected pool config: %+v", cfg)
|
|
}
|
|
if !cfg.GLPIKBEnabled || cfg.GLPIURL != "http://glpi.test" {
|
|
t.Fatalf("unexpected GLPI config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsMismatchedPoolMetadata(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("OLLAMA_URLS", "http://gpu-a:11434,http://gpu-b:11434")
|
|
t.Setenv("OLLAMA_NODE_NAMES", "only-one")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected pool metadata validation error")
|
|
}
|
|
}
|
|
|
|
func TestLoadRuntimeControlDefaultsAndFilters(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_LEARNING_ENABLED", "false")
|
|
t.Setenv("BRAIN_THINKING_ENABLED", "false")
|
|
t.Setenv("BRAIN_LEARNING_CATEGORIES", "Netzwerk,GLPI KB")
|
|
t.Setenv("BRAIN_DISPLAY_CATEGORIES", "GLPI KB")
|
|
t.Setenv("BRAIN_THINKING_CATEGORIES", "Netzwerk")
|
|
t.Setenv("BRAIN_DEFAULT_VIEW", "honeycomb")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.LearningEnabled || cfg.ThinkingEnabled || cfg.DefaultView != "honeycomb" {
|
|
t.Fatalf("unexpected runtime defaults: %+v", cfg)
|
|
}
|
|
if len(cfg.LearningCategories) != 2 || len(cfg.DisplayCategories) != 1 || len(cfg.ThinkingCategories) != 1 {
|
|
t.Fatalf("unexpected category defaults: %+v", cfg)
|
|
}
|
|
}
|