88 lines
3.2 KiB
Go
88 lines
3.2 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")
|
|
t.Setenv("BRAIN_MAX_DISPLAY_NODES", "12000")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.LearningEnabled || cfg.ThinkingEnabled || cfg.DefaultView != "honeycomb" || cfg.MaxDisplayNodes != 12000 {
|
|
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)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsInvalidMaxDisplayNodes(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_MAX_DISPLAY_NODES", "500001")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected max display nodes validation error")
|
|
}
|
|
}
|
|
|
|
func TestLoadArticleSynthesisQualitySettings(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_ARTICLE_SYNTHESIS_ENABLED", "true")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_SOURCES", "4")
|
|
t.Setenv("BRAIN_ARTICLE_MAX_SOURCES", "7")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_PRODUCTION_RATIO", "0.8")
|
|
t.Setenv("BRAIN_ARTICLE_MAX_GENERATION_DEPTH", "2")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_CONFIDENCE", "0.81")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_TEXT_CHARS", "220")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_ANSWER_CHARS", "500")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !cfg.ArticleSynthesisEnabled || cfg.ArticleMinSources != 4 || cfg.ArticleMaxSources != 7 || cfg.ArticleMinProductionRatio != 0.8 || cfg.ArticleMaxGenerationDepth != 2 || cfg.ArticleMinConfidence != 0.81 || cfg.ArticleMinTextChars != 220 || cfg.ArticleMinAnswerChars != 500 {
|
|
t.Fatalf("unexpected article synthesis config: %+v", cfg)
|
|
}
|
|
}
|