Function TestGetSessions

pkg/session/store_test.go:123–171

(t *testing.T)

}

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

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

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

	testAgent := agent.New("test-agent", "test prompt")

	session1 := &Session{
		ID: "session-1",
		Messages: []Item{
			NewMessageItem(NewAgentMessage(testAgent.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Message from session 1",
			})),
		},
		CreatedAt: time.Now().Add(-1 * time.Hour),
	}

	session2 := &Session{
		ID: "session-2",
		Messages: []Item{
			NewMessageItem(NewAgentMessage(testAgent.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Message from session 2",
			})),
		},
		CreatedAt: time.Now(),
	}

	// Store the sessions
	err = store.AddSession(t.Context(), session1)
	require.NoError(t, err)
	err = store.AddSession(t.Context(), session2)
	require.NoError(t, err)

	// Retrieve all sessions
	sessions, err := store.GetSessions(t.Context())
	require.NoError(t, err)
	assert.Len(t, sessions, 2)

	for _, session := range sessions {
		assert.Len(t, session.Messages, 1)
		assert.Equal(t, "test-agent", session.Messages[0].Message.AgentName)
	}
}

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