63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/diagnostics"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/framework/pipeline"
|
|
)
|
|
|
|
func TestApplyEnvOverridesOperationalValues(t *testing.T) {
|
|
cfg := Default()
|
|
cfg.Pipelines["example"] = pipeline.PipelineProfile{ID: "example", Input: pipeline.Binding("before")}
|
|
|
|
err := cfg.applyEnvOverridesWithLookup(mapLookup(map[string]string{
|
|
"NOTARIUS_TOTAL_LLM_CONCURRENCY": "3",
|
|
"NOTARIUS_WORK_DIR": "/tmp/notarius-env",
|
|
"NOTARIUS_DIAGNOSTICS_RETENTION": "never",
|
|
"NOTARIUS_PIPELINE_INPUT": "after",
|
|
}))
|
|
if err != nil {
|
|
t.Fatalf("ApplyEnvOverrides: %v", err)
|
|
}
|
|
|
|
if cfg.Scriptorium.ProfileDir != "" || cfg.Scriptorium.ProfileFile != "" {
|
|
t.Fatalf("LLM environment overrides must not change Scriptorium config: %+v", cfg.Scriptorium)
|
|
}
|
|
if cfg.Concurrency.TotalLLM != 3 {
|
|
t.Fatalf("unexpected total concurrency: %d", cfg.Concurrency.TotalLLM)
|
|
}
|
|
if cfg.Diagnostics.WorkDir != "/tmp/notarius-env" || cfg.Diagnostics.Retention != diagnostics.RetentionNever {
|
|
t.Fatalf("unexpected diagnostics config: %+v", cfg.Diagnostics)
|
|
}
|
|
if cfg.Pipelines["example"].Input.Module != "before" {
|
|
t.Fatalf("environment overrides must not change pipeline wiring: %+v", cfg.Pipelines["example"])
|
|
}
|
|
}
|
|
|
|
func TestApplyEnvOverridesRejectsInvalidIntegers(t *testing.T) {
|
|
cfg := Default()
|
|
err := cfg.applyEnvOverridesWithLookup(mapLookup(map[string]string{
|
|
"NOTARIUS_TOTAL_LLM_CONCURRENCY": "many",
|
|
}))
|
|
if err == nil || !strings.Contains(err.Error(), "NOTARIUS_TOTAL_LLM_CONCURRENCY") {
|
|
t.Fatalf("expected named integer error, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestLoadFromEnvUsesDefaultConfig(t *testing.T) {
|
|
t.Setenv("NOTARIUS_TOTAL_LLM_CONCURRENCY", "2")
|
|
|
|
cfg, err := LoadFromEnv()
|
|
if err != nil {
|
|
t.Fatalf("LoadFromEnv: %v", err)
|
|
}
|
|
if cfg.Scriptorium.ProfileDir != "" || cfg.Scriptorium.ProfileFile != "" {
|
|
t.Fatalf("unexpected Scriptorium config from env: %+v", cfg.Scriptorium)
|
|
}
|
|
if cfg.Concurrency.TotalLLM != 2 {
|
|
t.Fatalf("expected env concurrency override, got %+v", cfg.Concurrency)
|
|
}
|
|
}
|