Function TestStoreMultipleAgents

pkg/session/store_test.go:72–121

(t *testing.T)

}

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

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

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

	agent1 := agent.New("agent-1", "agent 1 prompt")
	agent2 := agent.New("agent-2", "agent 2 prompt")

	session := &Session{
		ID:        "multi-agent-session",
		CreatedAt: time.Now(),
		Messages: []Item{
			NewMessageItem(UserMessage("Start conversation")),
			NewMessageItem(NewAgentMessage(agent1.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Response from agent 1",
			})),
			NewMessageItem(NewAgentMessage(agent2.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Response from agent 2",
			})),
		},
	}

	// Store the session
	err = store.AddSession(t.Context(), session)
	require.NoError(t, err)

	// Retrieve the session
	retrievedSession, err := store.GetSession(t.Context(), "multi-agent-session")
	require.NoError(t, err)
	require.NotNil(t, retrievedSession)

	assert.Len(t, retrievedSession.Messages, 3)

	// First message should be user message with empty agent name
	assert.Empty(t, retrievedSession.Messages[0].Message.AgentName)

	// Second message should have agent-1 name
	assert.Equal(t, "agent-1", retrievedSession.Messages[1].Message.AgentName)
	assert.Equal(t, "Response from agent 1", retrievedSession.Messages[1].Message.Message.Content)

	// Third message should have agent-2 name
	assert.Equal(t, "agent-2", retrievedSession.Messages[2].Message.AgentName)
	assert.Equal(t, "Response from agent 2", retrievedSession.Messages[2].Message.Message.Content)
}

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