Function TestUpdateSession_LazyCreation

pkg/session/store_test.go:385–432

(t *testing.T)

}

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

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

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

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

	// Create a session but don't add it to the store (simulating lazy creation)
	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) but does NOT persist messages
	// Messages must be added separately via AddMessage
	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 (the proper way)
	_, 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 should have messages
	retrieved, err = store.GetSession(t.Context(), "lazy-session")
	require.NoError(t, err)
	assert.Len(t, retrieved.Messages, 2)
	assert.Equal(t, "Hello", retrieved.Messages[0].Message.Message.Content)
	assert.Equal(t, "Hi there!", retrieved.Messages[1].Message.Message.Content)
}

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