agent_context_test.go RawHistoryFolder

1package core
2
3import (
4	"testing"
5
6	"github.com/bnema/zut/packages/provider"
7)
8
9func TestAppendUserContextAddsAndPublishesMessage(t *testing.T) {
10	agent := NewAgent(nil, "", "", nil)
11	var appended []provider.Message
12	agent.OnMessageAppended = func(msg provider.Message) {
13		appended = append(appended, msg)
14	}
15
16	before := agent.Revision()
17	agent.AppendUserContext("$ pwd\n\n/tmp\n\n[exit 0]", map[string]string{"shell_escape": "true"})
18
19	messages := agent.Messages()
20	if len(messages) != 1 {
21		t.Fatalf("Messages() length = %d, want 1", len(messages))
22	}
23	msg := messages[0]
24	if msg.Role != provider.RoleUser {
25		t.Fatalf("message role = %q, want %q", msg.Role, provider.RoleUser)
26	}
27	if msg.Meta["shell_escape"] != "true" {
28		t.Fatalf("message metadata = %v, want shell_escape marker", msg.Meta)
29	}
30	if len(msg.Content) != 1 {
31		t.Fatalf("message content length = %d, want 1", len(msg.Content))
32	}
33	text, ok := msg.Content[0].(provider.TextBlock)
34	if !ok || text.Text != "$ pwd\n\n/tmp\n\n[exit 0]" {
35		t.Fatalf("message content = %#v", msg.Content[0])
36	}
37	if msg.Time.IsZero() {
38		t.Fatal("message timestamp is zero")
39	}
40	if agent.Revision() != before+1 {
41		t.Fatalf("Revision() = %d, want %d", agent.Revision(), before+1)
42	}
43	if len(appended) != 1 || appended[0].Meta["shell_escape"] != "true" {
44		t.Fatalf("OnMessageAppended messages = %#v", appended)
45	}
46}