dynamic_context_test.go RawHistoryFolder

1package core
2
3import (
4	"context"
5	"testing"
6
7	"github.com/bnema/zut/packages/provider"
8)
9
10func TestFirstDynamicContextPersistsAsTranscriptReplacement(t *testing.T) {
11	client := &compactLifecycleClient{}
12	agent := NewAgent(client, "test-model", "system", Registry{})
13	session, err := NewSession(t.TempDir(), "/workspace", "test", "test-model", "test")
14	if err != nil {
15		t.Fatal(err)
16	}
17	agent.OnMessageAppended = func(message provider.Message) {
18		if err := session.AppendMessage(message); err != nil {
19			t.Fatal(err)
20		}
21	}
22	agent.OnTranscriptCompacted = func(messages []provider.Message) {
23		if err := session.AppendCompaction(messages); err != nil {
24			t.Fatal(err)
25		}
26	}
27	agent.BeforeTurnContext = func(context.Context, int) (bool, string, string) {
28		return true, "", "accepted extension context"
29	}
30	if err := agent.Prompt(context.Background(), "accepted user message", nil, nil); err != nil {
31		t.Fatal(err)
32	}
33	path := session.Path
34	if err := session.Close(); err != nil {
35		t.Fatal(err)
36	}
37
38	opened, messages, err := OpenSession(path)
39	if err != nil {
40		t.Fatal(err)
41	}
42	defer opened.Close()
43	if len(messages) < 2 || messages[0].Role != provider.RoleDeveloper || messages[1].Role != provider.RoleUser {
44		t.Fatalf("restored messages = %#v, want developer context followed by accepted user message", messages)
45	}
46}