interactive_queue_test.go RawHistoryFolder

1package modes
2
3import (
4	"context"
5	"testing"
6
7	"github.com/bnema/zut/packages/core"
8	"github.com/bnema/zut/packages/tui"
9)
10
11func TestEscapeRestoresMostRecentQueuedMessageToEditor(t *testing.T) {
12	agent := core.NewAgent(nil, "test-model", "", nil)
13	agent.QueueMessage("older follow-up")
14	agent.QueueMessage("recover this draft")
15
16	i := NewInteractive(InteractiveConfig{Agent: agent})
17	i.ed.SetValue("existing draft")
18	cancelled := 0
19	i.mu.Lock()
20	i.busy = true
21	i.cancelTurn = func() { cancelled++ }
22	i.mu.Unlock()
23
24	if done := i.handleKey(context.Background(), tui.Key{Kind: tui.KeyEsc}); done {
25		t.Fatal("Escape exited")
26	}
27	if cancelled != 1 {
28		t.Fatalf("cancelled = %d, want 1", cancelled)
29	}
30	if got, want := i.ed.Value(), "recover this draft"; got != want {
31		t.Errorf("editor = %q, want %q", got, want)
32	}
33	if got := agent.PendingQueuedMessages(); len(got) != 1 || got[0] != "older follow-up" {
34		t.Errorf("remaining queued messages = %v, want [older follow-up]", got)
35	}
36}
37
38func TestEscapeRestoresHostQueuedMessageToEditor(t *testing.T) {
39	i := NewInteractive(InteractiveConfig{})
40	cancelled := 0
41	i.mu.Lock()
42	i.busy = true
43	i.cancelTurn = func() { cancelled++ }
44	i.queued = []string{"recover this draft"}
45	i.mu.Unlock()
46
47	i.handleKey(context.Background(), tui.Key{Kind: tui.KeyEsc})
48
49	if cancelled != 1 {
50		t.Fatalf("cancelled = %d, want 1", cancelled)
51	}
52	if got, want := i.ed.Value(), "recover this draft"; got != want {
53		t.Errorf("editor = %q, want %q", got, want)
54	}
55}