Function TestBranchSessionCopiesPrefix

pkg/session/store_test.go:234–273

(t *testing.T)

}

func TestBranchSessionCopiesPrefix(t *testing.T) {
	t.Parallel()

	tempDB := filepath.Join(t.TempDir(), "test_branch_prefix.db")

	store, err := NewSQLiteSessionStore(t.Context(), tempDB)
	require.NoError(t, err)
	defer store.(*SQLiteSessionStore).Close()

	testAgent := agent.New("test-agent", "test prompt")
	parent := &Session{
		ID:        "parent-session",
		CreatedAt: time.Now(),
		Messages: []Item{
			NewMessageItem(UserMessage("Hello")),
			NewMessageItem(NewAgentMessage(testAgent.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Response",
			})),
			NewMessageItem(UserMessage("Edited")),
		},
	}

	require.NoError(t, store.AddSession(t.Context(), parent))

	parentLoaded, err := store.GetSession(t.Context(), parent.ID)
	require.NoError(t, err)

	branched, err := BranchSession(parentLoaded, 2)
	require.NoError(t, err)

	require.NoError(t, store.AddSession(t.Context(), branched))

	loaded, err := store.GetSession(t.Context(), branched.ID)
	require.NoError(t, err)

	require.Len(t, loaded.Messages, 2)
	assert.Equal(t, "Hello", loaded.Messages[0].Message.Message.Content)
	assert.Equal(t, "Response", loaded.Messages[1].Message.Message.Content)
}

func TestBranchSessionClonesSubSession(t *testing.T) {
	t.Parallel()