interactive_host_test.go RawHistoryFolder

1package modes
2
3import (
4	"testing"
5
6	"github.com/bnema/zut/packages/agent/internal/orchestration"
7	"github.com/bnema/zut/packages/agent/subagents"
8	"github.com/bnema/zut/packages/core"
9	"github.com/bnema/zut/packages/provider"
10)
11
12func TestSubmitOrQueuePreservesImagesInHostQueue(t *testing.T) {
13	i := &Interactive{agent: &core.Agent{}, busy: true, compacting: true}
14	image := provider.ImageBlock{MimeType: "image/png", Data: []byte("png-1")}
15
16	i.submitOrQueue("inspect", []provider.ImageBlock{image}, false)
17
18	if len(i.queued) != 1 || i.queued[0].Text != "inspect" {
19		t.Fatalf("queued messages = %#v, want one inspect prompt", i.queued)
20	}
21	if len(i.queued[0].Images) != 1 || string(i.queued[0].Images[0].Data) != "png-1" {
22		t.Fatalf("queued images = %#v, want png-1", i.queued[0].Images)
23	}
24}
25
26func TestSubmitOrQueuePreservesSealedWorkerWaveWithPendingWorker(t *testing.T) {
27	i := &Interactive{agent: &core.Agent{}}
28	i.applyCoordinator(orchestration.Event{Kind: orchestration.EventManagerStarted})
29	i.applyCoordinator(orchestration.Event{Kind: orchestration.EventWorkerRegistered, WorkerID: "worker#1"})
30	i.applyCoordinator(orchestration.Event{Kind: orchestration.EventManagerFinished})
31
32	image := provider.ImageBlock{MimeType: "image/png", Data: []byte("png-1")}
33	i.submitOrQueue("review the result", []provider.ImageBlock{image}, true)
34
35	actions := i.applyCoordinator(orchestration.Event{
36		Kind:     orchestration.EventWorkerFinished,
37		WorkerID: "worker#1",
38		Completion: subagents.Completion{
39			AgentID: "worker",
40			Status:  "completed",
41		},
42	})
43	if len(actions) != 1 {
44		t.Fatalf("completion actions = %d, want 1", len(actions))
45	}
46	if got := actions[0].Text; got != "review the result" {
47		t.Fatalf("completion action text = %q, want queued user input", got)
48	}
49	if len(actions[0].Images) != 1 || string(actions[0].Images[0].Data) != "png-1" {
50		t.Fatalf("completion action images = %#v, want queued image", actions[0].Images)
51	}
52}