Function TestBranchSessionClonesSubSession

pkg/session/store_test.go:275–321

(t *testing.T)

}

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

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

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

	subSession := &Session{
		ID:        "sub-session",
		CreatedAt: time.Now(),
		Messages: []Item{
			NewMessageItem(UserMessage("Sub message")),
		},
	}
	parent := &Session{
		ID:        "parent-session",
		CreatedAt: time.Now(),
		Messages: []Item{
			NewMessageItem(UserMessage("Start")),
			NewSubSessionItem(subSession),
			NewMessageItem(UserMessage("After")),
		},
	}

	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)

	subItem := loaded.Messages[1]
	require.NotNil(t, subItem.SubSession)
	assert.NotEqual(t, subSession.ID, subItem.SubSession.ID)
	assert.Equal(t, loaded.ID, subItem.SubSession.ParentID)
	require.Len(t, subItem.SubSession.Messages, 1)
	assert.Equal(t, "Sub message", subItem.SubSession.Messages[0].Message.Message.Content)
}

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