login_lmstudio_test.go RawHistoryFolder

1package modes
2
3import (
4	"testing"
5
6	"github.com/patriceckhart/zot/packages/tui"
7)
8
9func TestLoginLMStudioRequiresServerURL(t *testing.T) {
10	d := newLoginDialog()
11	d.step = loginStepProvider
12	d.method = "apikey"
13	d.status = map[string]string{}
14	d.providerQuery = "lmstudio"
15	d.HandleKey(tui.Key{Kind: tui.KeyEnter})
16	if d.step != loginStepLlamaURL || d.provider != "lmstudio" {
17		t.Fatalf("wrong login step: %v", d.step)
18	}
19	d.HandleKey(tui.Key{Kind: tui.KeyEnter})
20	if d.step != loginStepLlamaURL {
21		t.Fatal("empty URL enabled discovery")
22	}
23	d.llamaEd.SetValue("http://localhost:1234/api/v1")
24	d.HandleKey(tui.Key{Kind: tui.KeyEnter})
25	if d.step != loginStepLlamaKey {
26		t.Fatalf("URL rejected: %s", d.message)
27	}
28	a := d.HandleKey(tui.Key{Kind: tui.KeyEnter})
29	if !a.SaveLlama || a.Provider != "lmstudio" || a.LlamaURL != "http://localhost:1234" || a.LlamaAPIKey != "" {
30		t.Fatalf("action=%+v", a)
31	}
32}