Function TestStoreAgentNameJSON

pkg/session/store_test.go:323–363

(t *testing.T)

}

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

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

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

	agent1 := agent.New("my-agent", "test prompt")
	agent2 := agent.New("another-agent", "another prompt")

	session := &Session{
		ID: "json-test-session",
		Messages: []Item{
			NewMessageItem(UserMessage("User input")),
			NewMessageItem(NewAgentMessage(agent1.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Response from my-agent",
			})),
			NewMessageItem(NewAgentMessage(agent2.Name(), &chat.Message{
				Role:    chat.MessageRoleAssistant,
				Content: "Response from another-agent",
			})),
		},
		CreatedAt: time.Now(),
	}

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

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

	assert.Empty(t, retrievedSession.Messages[0].Message.AgentName)                  // User message
	assert.Equal(t, "my-agent", retrievedSession.Messages[1].Message.AgentName)      // First agent
	assert.Equal(t, "another-agent", retrievedSession.Messages[2].Message.AgentName) // Second agent
}

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