session_agent_root_test.go RawHistoryFolder

1package modes
2
3import (
4	"context"
5	"testing"
6
7	"github.com/patriceckhart/zot/packages/core"
8	"github.com/patriceckhart/zot/packages/provider"
9)
10
11func TestSessionsSlashUsesAgentSessionRoot(t *testing.T) {
12	zotHome := t.TempDir()
13	agentRoot := t.TempDir()
14	cwd := t.TempDir()
15
16	session, err := core.NewSession(agentRoot, cwd, "test", "test-model", "test-version")
17	if err != nil {
18		t.Fatal(err)
19	}
20	if err := session.AppendMessage(provider.Message{
21		Role:    provider.RoleUser,
22		Content: []provider.Content{provider.TextBlock{Text: "agent session"}},
23	}); err != nil {
24		t.Fatal(err)
25	}
26	if err := session.Close(); err != nil {
27		t.Fatal(err)
28	}
29
30	i := NewInteractive(InteractiveConfig{
31		ZotHome:      zotHome,
32		SessionsRoot: agentRoot,
33		CWD:          cwd,
34	})
35	i.runSlash(context.Background(), "/sessions")
36
37	if got := len(i.sessionDialog.sessions); got != 1 {
38		t.Fatalf("session picker entries = %d, want 1 from agent session root", got)
39	}
40	if got := i.sessionDialog.sessions[0].Path; got != session.Path {
41		t.Fatalf("session picker path = %q, want %q", got, session.Path)
42	}
43}
44
45func TestSessionsRootDefaultsToZotHome(t *testing.T) {
46	i := &Interactive{cfg: InteractiveConfig{ZotHome: "/zot/home"}}
47	if got := i.sessionsRoot(); got != "/zot/home" {
48		t.Fatalf("sessions root = %q, want ZotHome fallback", got)
49	}
50}