Function TestUpdateSession_LazyCreation_InMemory

pkg/session/store_test.go:434–474

(t *testing.T)

}

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

	store := NewInMemorySessionStore()

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

	// Create a session but don't add it to the store
	session := &Session{
		ID:        "lazy-session",
		CreatedAt: time.Now(),
	}

	// Verify session doesn't exist yet
	_, err := store.GetSession(t.Context(), "lazy-session")
	require.ErrorIs(t, err, ErrNotFound)

	// UpdateSession creates the session (upsert) without messages
	// Messages must be added separately via AddMessage (like SQLite behavior)
	err = store.UpdateSession(t.Context(), session)
	require.NoError(t, err)

	// Session exists but has no messages yet
	retrieved, err := store.GetSession(t.Context(), "lazy-session")
	require.NoError(t, err)
	assert.Empty(t, retrieved.Messages)

	// Add messages via AddMessage
	_, err = store.AddMessage(t.Context(), "lazy-session", UserMessage("Hello"))
	require.NoError(t, err)
	_, err = store.AddMessage(t.Context(), "lazy-session", NewAgentMessage(testAgent.Name(), &chat.Message{
		Role:    chat.MessageRoleAssistant,
		Content: "Hi there!",
	}))
	require.NoError(t, err)

	// Now the session has 2 messages
	retrieved, err = store.GetSession(t.Context(), "lazy-session")
	require.NoError(t, err)
	assert.Len(t, retrieved.Messages, 2)
}

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