61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
package config
|
|
|
|
import "testing"
|
|
|
|
func TestLoadFromEnvDefaults(t *testing.T) {
|
|
t.Setenv("TUTOR_HTTP_ADDR", "")
|
|
t.Setenv("TUTOR_ENV", "")
|
|
t.Setenv("TUTOR_WORKFLOW_RUNTIME", "")
|
|
t.Setenv("TUTOR_MODEL_KEY", "")
|
|
t.Setenv("TUTOR_IMAGE_MODEL_KEY", "")
|
|
t.Setenv("THIRDONE_BIN", "")
|
|
|
|
cfg := LoadFromEnv()
|
|
|
|
if cfg.HTTPAddr != defaultHTTPAddr {
|
|
t.Fatalf("HTTPAddr = %q, want %q", cfg.HTTPAddr, defaultHTTPAddr)
|
|
}
|
|
if cfg.Environment != defaultEnvironment {
|
|
t.Fatalf("Environment = %q, want %q", cfg.Environment, defaultEnvironment)
|
|
}
|
|
if cfg.ModelKey != defaultModelKey {
|
|
t.Fatalf("ModelKey = %q, want %q", cfg.ModelKey, defaultModelKey)
|
|
}
|
|
if cfg.ImageModelKey != defaultImageModelKey {
|
|
t.Fatalf("ImageModelKey = %q, want %q", cfg.ImageModelKey, defaultImageModelKey)
|
|
}
|
|
if cfg.ThirdOneBin != defaultThirdOneBin {
|
|
t.Fatalf("ThirdOneBin = %q, want %q", cfg.ThirdOneBin, defaultThirdOneBin)
|
|
}
|
|
}
|
|
|
|
func TestLoadFromEnvOverrides(t *testing.T) {
|
|
t.Setenv("TUTOR_HTTP_ADDR", ":9090")
|
|
t.Setenv("TUTOR_ENV", "test")
|
|
t.Setenv("TUTOR_WORKFLOW_RUNTIME", "runtime.yaml")
|
|
t.Setenv("TUTOR_MODEL_KEY", "other-model")
|
|
t.Setenv("TUTOR_IMAGE_MODEL_KEY", "other-image-model")
|
|
t.Setenv("THIRDONE_BIN", "C:/bin/thirdone.exe")
|
|
|
|
cfg := LoadFromEnv()
|
|
|
|
if cfg.HTTPAddr != ":9090" {
|
|
t.Fatalf("HTTPAddr = %q", cfg.HTTPAddr)
|
|
}
|
|
if cfg.Environment != "test" {
|
|
t.Fatalf("Environment = %q", cfg.Environment)
|
|
}
|
|
if cfg.WorkflowRuntime != "runtime.yaml" {
|
|
t.Fatalf("WorkflowRuntime = %q", cfg.WorkflowRuntime)
|
|
}
|
|
if cfg.ModelKey != "other-model" {
|
|
t.Fatalf("ModelKey = %q", cfg.ModelKey)
|
|
}
|
|
if cfg.ImageModelKey != "other-image-model" {
|
|
t.Fatalf("ImageModelKey = %q", cfg.ImageModelKey)
|
|
}
|
|
if cfg.ThirdOneBin != "C:/bin/thirdone.exe" {
|
|
t.Fatalf("ThirdOneBin = %q", cfg.ThirdOneBin)
|
|
}
|
|
}
|