Test case for asserting on integer conditional

Shopify · Feb 14, 2024 · f71d0ac · f71d0ac
1 parent ff5935d
commit f71d0ac
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/unit/active_record_enumerator_test.rb b/test/unit/active_record_enumerator_test.rb
@@ -4,6 +4,8 @@
 
 module JobIteration
   class ActiveRecordEnumeratorTest < IterationUnitTest
+    include ActiveRecordHelpers
+
     SQL_TIME_FORMAT = "%Y-%m-%d %H:%M:%S.%N"
     test "#records yields every record with their cursor position" do
       enum = build_enumerator.records
@@ -133,6 +135,13 @@ class ActiveRecordEnumeratorTest < IterationUnitTest
       end
     end
 
+    test "enumerator paginates using integer conditionals for primary key when no columns are defined" do
+      enum = build_enumerator(relation: Product.all, batch_size: 1).records
+      assert_sql(/`products`\.`id` > 1/) do
+        enum.take(2)
+      end
+    end
+
     private
 
     def build_enumerator(relation: Product.all, batch_size: 2, columns: nil, cursor: nil)