Merge pull request #21 from healthie/fix-total_count

Fix `Paginator#total_count` to not change when paginating

Author: fatkodima

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## master (unreleased)
 
+- Fix `Paginator#total_count` to not change when paginating
 - Fix `Paginator#limit` to return default page size by default
 
 ## 0.4.0 (2025-03-10)

lib/activerecord_cursor_paginate/paginator.rb

@@ -184,7 +184,7 @@ def pages
     # @return [Integer]
     #
     def total_count
-      @total_count ||= build_cursor_relation(@cursor).count(:all)
+      @total_count ||= @relation.count(:all)
     end
 
     private

test/paginator_test.rb

@@ -149,10 +149,16 @@ def test_does_not_append_id_if_asked
   end
 
   def test_paginates_forward_after_cursor
+    total_count = User.count
+
     p1 = User.cursor_paginate(limit: 3)
+    assert_equal total_count, p1.total_count
+
     page1 = p1.fetch
 
     p2 = User.cursor_paginate(after: page1.cursor, limit: 4)
+    assert_equal total_count, p2.total_count
+
     page2 = p2.fetch
     assert_equal([4, 5, 6, 7], page2.records.pluck(:id))
   end