Ilker ManapandClaude Opus 5.5
0033a530c7
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
86 lines
3.0 KiB
Go
86 lines
3.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestAPeerIsAddedToTheMirrorsUnitOnce(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "llmgw-mirror.service")
|
|
unit := mirrorUnitFor("/srv/models", ":8082", "http://10.0.0.10:8081", []string{"10.0.0.10"}, nil, "test")
|
|
if err := os.WriteFile(path, []byte(unit), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
changed, err := addToMirrorUnit(path, "http://10.0.0.12:8082", "10.0.0.12")
|
|
if err != nil || !changed {
|
|
t.Fatalf("changed %v %v", changed, err)
|
|
}
|
|
b, _ := os.ReadFile(path)
|
|
if !strings.Contains(string(b), "--peer http://10.0.0.12:8082") || !strings.Contains(string(b), "--api-allow 10.0.0.12") {
|
|
t.Errorf("unit:\n%s", b)
|
|
}
|
|
if changed, _ := addToMirrorUnit(path, "http://10.0.0.12:8082", "10.0.0.12"); changed {
|
|
t.Error("added twice")
|
|
}
|
|
// A mirror whose API is on loopback alone is not opened by the back door.
|
|
lo := mirrorUnitFor("/srv/models", "127.0.0.1:8082", "http://10.0.0.10:8081", nil, nil, "test")
|
|
_ = os.WriteFile(path, []byte(lo), 0o644)
|
|
_, _ = addToMirrorUnit(path, "http://10.0.0.12:8082", "10.0.0.12")
|
|
if b, _ := os.ReadFile(path); strings.Contains(string(b), "--api-allow") {
|
|
t.Errorf("a loopback mirror was opened:\n%s", b)
|
|
}
|
|
}
|
|
|
|
func TestASecondMirrorIsNamedInTheSettings(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "server.yaml")
|
|
if err := os.WriteFile(path, []byte("listen: \":8080\"\nmirror:\n api: http://10.0.0.10:8082\n"), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for range 2 {
|
|
if err := addMirrorPeerSetting(path, "http://10.0.0.12:8082"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
b, _ := os.ReadFile(path)
|
|
if strings.Count(string(b), "10.0.0.12:8082") != 1 || !strings.Contains(string(b), "peers") {
|
|
t.Errorf("server.yaml:\n%s", b)
|
|
}
|
|
}
|
|
|
|
func TestTheSetupOffersASecondMirrorOnAStoreMachine(t *testing.T) {
|
|
in := newDefaultServerInit()
|
|
f := newInstallForm(&in, "ctl", []string{"10.0.0.5"}, true)
|
|
f.secondMirror.Choice = 1
|
|
if !f.secondMirror.Show() {
|
|
t.Fatal("not offered with the mirror on this machine")
|
|
}
|
|
f.mirrorWhere.Choice = 1
|
|
if f.secondMirror.Show() {
|
|
t.Error("offered with the mirror elsewhere")
|
|
}
|
|
f.mirrorWhere.Choice = 0
|
|
in.secondMirror = 1
|
|
in.patroni = []string{"ubuntu@10.0.0.11", "ubuntu@10.0.0.12"}
|
|
if got := strings.Join(in.initFlags(), " "); !strings.Contains(got, "--second-mirror 1") {
|
|
t.Errorf("flags %s", got)
|
|
}
|
|
// Without standbys this mirror answers itself alone: the second could not read it.
|
|
in.standbys = false
|
|
if err := in.ensureSecondMirror(t.Context()); err == nil || !strings.Contains(err.Error(), "standbys") {
|
|
t.Errorf("without standbys: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestAMirrorOnAMachineOfItsOwnSaysWhatToRunOnAControlPlane(t *testing.T) {
|
|
var out bytes.Buffer
|
|
if err := nameMirrorPeer(&out, filepath.Join(t.TempDir(), "server.yaml"), "http://10.0.0.12:8082"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(out.String(), "llmgw ctl config set mirror.peers '[http://10.0.0.12:8082]'") {
|
|
t.Errorf("%s", out.String())
|
|
}
|
|
}
|