mirror of
https://github.com/mostlygeek/llama-swap.git
synced 2026-06-09 06:46:34 +02:00
Benson Wong
a4b91e08cf
- update README.md with new docker instructions - update docs/configuration.md - update .github/workflows to have pinned action versions - gofmt events package - fix small bugs in CI scripts - reduce config options for internal/perf/monitor and config. A ring buffer is used to keep 1hr of entries at max 5s granularity. For long term stats use prometheus monitoring on /metrics Fixes #744
99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPerformanceConfig_Defaults(t *testing.T) {
|
|
content := `
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
config, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.NoError(t, err)
|
|
|
|
// When performance section is missing, defaults should be applied
|
|
assert.False(t, config.Performance.Disabled)
|
|
assert.Equal(t, 5*time.Second, config.Performance.Every)
|
|
}
|
|
|
|
func TestPerformanceConfig_CustomValues(t *testing.T) {
|
|
content := `
|
|
performance:
|
|
enable: true
|
|
every: 30s
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
config, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.NoError(t, err)
|
|
|
|
assert.False(t, config.Performance.Disabled)
|
|
assert.Equal(t, 30*time.Second, config.Performance.Every)
|
|
}
|
|
|
|
func TestPerformanceConfig_Disabled(t *testing.T) {
|
|
content := `
|
|
performance:
|
|
disabled: true
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
config, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.NoError(t, err)
|
|
|
|
assert.True(t, config.Performance.Disabled)
|
|
// Duration defaults should still apply
|
|
assert.Equal(t, 5*time.Second, config.Performance.Every)
|
|
}
|
|
|
|
func TestPerformanceConfig_PartialValues(t *testing.T) {
|
|
content := `
|
|
performance:
|
|
every: 10s
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
config, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.NoError(t, err)
|
|
|
|
// enable should default to true
|
|
assert.False(t, config.Performance.Disabled)
|
|
assert.Equal(t, 10*time.Second, config.Performance.Every)
|
|
}
|
|
|
|
func TestPerformanceConfig_InvalidEvery(t *testing.T) {
|
|
content := `
|
|
performance:
|
|
every: 4s
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
_, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "every must be at least 5s")
|
|
}
|
|
|
|
func TestPerformanceConfig_ComplexDurations(t *testing.T) {
|
|
content := `
|
|
performance:
|
|
every: 1m30s
|
|
models:
|
|
model1:
|
|
cmd: path/to/cmd --port ${PORT}
|
|
`
|
|
config, err := LoadConfigFromReader(strings.NewReader(content))
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, 90*time.Second, config.Performance.Every)
|
|
}
|