Coerce tests

Author: aidanharan

test/cases/coerced_tests.rb

@@ -2959,4 +2959,30 @@ def test_in_batches_executes_range_queries_when_constrained_and_opted_in_into_ra
       relations.each { |relation| assert_kind_of Post, relation.first }
     end
   end
+
+  # Match SQL Server SQL format.
+  coerce_tests! :test_in_batches_should_unscope_cursor_after_pluck
+  def test_in_batches_should_unscope_cursor_after_pluck_coerced
+    all_ids = Post.limit(2).pluck(:id)
+    found_ids = []
+    # only a single clause on id (i.e. not 'id IN (?,?) AND id = ?', but only 'id = ?')
+    assert_queries_match(/WHERE #{Regexp.escape(quote_table_name("posts.id"))} = \S+ ORDER BY/) do
+      Post.where(id: all_ids).in_batches(of: 1) do |relation|
+        found_ids << relation.pick(:id)
+      end
+    end
+    assert_equal all_ids.sort, found_ids
+  end
+
+  # Match SQL Server SQL format.
+  coerce_tests! :test_in_batches_loaded_should_unscope_cursor_after_pluck
+  def test_in_batches_loaded_should_unscope_cursor_after_pluck_coerced
+    all_ids = Post.limit(2).pluck(:id)
+    # only a single clause on id (i.e. not 'id IN (?,?) AND id = ?', but only 'id = ?')
+    assert_queries_match(/WHERE #{Regexp.escape(quote_table_name("posts.id"))} = \S+;/) do
+      Post.where(id: all_ids).in_batches(of: 1, load: true) do |relation|
+        relation.delete_all
+      end
+    end
+  end
 end