mirror of
https://github.com/mostlygeek/llama-swap.git
synced 2026-09-07 16:37:22 +02:00
Benson WongandGitHub
8a06d56b54
Expose llama-swap through Tailcat virtual TCP listeners and route peer requests through Tailcat transports. - validate server identities, caller allowlists, and published model IDs - restrict the default remote HTTP surface and retain Tailcat request sources - add UI status, activity attribution, configuration docs, and schema support - make Tailcat transport diagnostics opt-in with server.tailcat.debug fixes: #1073
92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mostlygeek/llama-swap/internal/config"
|
|
)
|
|
|
|
func TestConfigureTailcatListener_RequiresModels(t *testing.T) {
|
|
cfg := config.Config{}
|
|
err := configureTailcatListener(&cfg, "/path/to/server.private.json")
|
|
require.ErrorContains(t, err, "tailcat.models")
|
|
|
|
cfg.Tailcat = &config.TailcatConfig{}
|
|
err = configureTailcatListener(&cfg, "/path/to/server.private.json")
|
|
require.ErrorContains(t, err, "tailcat.models")
|
|
|
|
cfg.Tailcat.Models = []string{"model"}
|
|
require.NoError(t, configureTailcatListener(&cfg, "/path/to/server.private.json"))
|
|
assert.True(t, cfg.TailcatEnabled())
|
|
}
|
|
|
|
func TestConfigureTailcatListener_DisabledWithoutFlag(t *testing.T) {
|
|
cfg := config.Config{}
|
|
require.NoError(t, configureTailcatListener(&cfg, ""))
|
|
assert.False(t, cfg.TailcatEnabled())
|
|
}
|
|
|
|
func TestLoadTailcatListenerKey_InvalidFile(t *testing.T) {
|
|
_, err := loadTailcatListenerKey(filepath.Join(t.TempDir(), "missing.private.json"), true)
|
|
require.ErrorContains(t, err, "-listen-tailcat")
|
|
}
|
|
|
|
func TestRunValidate_ValidConfig(t *testing.T) {
|
|
dir := t.TempDir()
|
|
configPath := filepath.Join(dir, "config.yml")
|
|
require.NoError(t, os.WriteFile(configPath, []byte(`
|
|
models:
|
|
model1:
|
|
cmd: server --port ${PORT}
|
|
proxy: http://localhost:8080
|
|
`), 0644))
|
|
|
|
var buf strings.Builder
|
|
code := runValidate(configPath, "", &buf)
|
|
|
|
assert.Equal(t, 0, code)
|
|
assert.NotEmpty(t, buf.String())
|
|
assert.Contains(t, buf.String(), "config is valid")
|
|
}
|
|
|
|
func TestRunValidate_BrokenConfig(t *testing.T) {
|
|
dir := t.TempDir()
|
|
configPath := filepath.Join(dir, "config.yml")
|
|
require.NoError(t, os.WriteFile(configPath, []byte(`
|
|
models:
|
|
model1:
|
|
cmd: server --port ${PORT} ${UNKNOWN_MACRO}
|
|
proxy: http://localhost:8080
|
|
`), 0644))
|
|
|
|
var buf strings.Builder
|
|
code := runValidate(configPath, "", &buf)
|
|
|
|
assert.Equal(t, 1, code)
|
|
assert.NotEmpty(t, buf.String())
|
|
assert.Contains(t, buf.String(), "failed")
|
|
}
|
|
|
|
func TestRunValidate_ConfigDir(t *testing.T) {
|
|
dir := t.TempDir()
|
|
require.NoError(t, os.WriteFile(filepath.Join(dir, "a.yml"), []byte(`
|
|
models:
|
|
model1:
|
|
cmd: server --port ${PORT}
|
|
proxy: http://localhost:8080
|
|
`), 0644))
|
|
|
|
var buf strings.Builder
|
|
code := runValidate("", dir, &buf)
|
|
|
|
assert.Equal(t, 0, code)
|
|
assert.NotEmpty(t, buf.String())
|
|
assert.Contains(t, buf.String(), "config is valid")
|
|
}
|