From 82df12e293591534e636ca004ff2a8a7ad35a326 Mon Sep 17 00:00:00 2001 From: Levi Neely Date: Wed, 20 May 2026 10:50:20 +0200 Subject: [PATCH] add test for SetSessionID preamble update --- pkg/agent/core_test.go | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/pkg/agent/core_test.go b/pkg/agent/core_test.go index 8064819..e9c8b43 100644 --- a/pkg/agent/core_test.go +++ b/pkg/agent/core_test.go @@ -1705,6 +1705,33 @@ func TestSetSessionID_Rename(t *testing.T) { } } +func TestSetSessionID_UpdatesPreamble(t *testing.T) { + oldID := NewSessionID() + env := AgentEnv{ + preamble: "session is " + oldID + " end", + } + c := NewAgentCore(AgentCoreConfig{ + Backend: defaultBE(), + AgentName: "test", + AgentsDir: t.TempDir(), + SessionsDir: t.TempDir(), + SessionID: oldID, + CWD: t.TempDir(), + Env: env, + NewDispatcher: tools.NewDispatcher, + }).(*agent) + t.Cleanup(c.Close) + + newID := NewSessionID() + if err := c.SetSessionID(newID); err != nil { + t.Fatalf("SetSessionID: %v", err) + } + want := "session is " + newID + " end" + if got := c.SystemPrompt(); got != want { + t.Errorf("SystemPrompt() = %q; want %q", got, want) + } +} + func TestSetSessionID_SameID(t *testing.T) { c := newCore(t, nil, nil) id := c.sessionID