43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package config_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mathiasbq/supervisor/internal/config"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLoadDefaults(t *testing.T) {
|
|
t.Setenv("SUPERVISOR_PORT", "")
|
|
t.Setenv("LITELLM_BASE_URL", "")
|
|
t.Setenv("LITELLM_API_KEY", "")
|
|
t.Setenv("SUPERVISOR_CONFIG_DIR", "")
|
|
t.Setenv("INGEST_BASE_URL", "")
|
|
t.Setenv("SUPERVISOR_SESSIONS_DIR", "")
|
|
t.Setenv("SUPERVISOR_BRAIN_DIR", "")
|
|
|
|
cfg, err := config.Load()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "3200", cfg.Port)
|
|
assert.Equal(t, "http://iguana:4000", cfg.LiteLLMBaseURL)
|
|
assert.Equal(t, "./config/supervisor", cfg.ConfigDir)
|
|
assert.Equal(t, "http://localhost:3300", cfg.IngestBaseURL)
|
|
assert.Equal(t, "./brain/sessions", cfg.SessionsDir)
|
|
assert.Equal(t, "./brain", cfg.BrainDir)
|
|
}
|
|
|
|
func TestLoadFromEnv(t *testing.T) {
|
|
t.Setenv("SUPERVISOR_PORT", "4000")
|
|
t.Setenv("LITELLM_BASE_URL", "http://localhost:4000")
|
|
t.Setenv("LITELLM_API_KEY", "test-key")
|
|
t.Setenv("SUPERVISOR_CONFIG_DIR", "/etc/supervisor")
|
|
|
|
cfg, err := config.Load()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "4000", cfg.Port)
|
|
assert.Equal(t, "http://localhost:4000", cfg.LiteLLMBaseURL)
|
|
assert.Equal(t, "test-key", cfg.LiteLLMAPIKey)
|
|
assert.Equal(t, "/etc/supervisor", cfg.ConfigDir)
|
|
}
|