Update mocks for Rails 7+

Shopify · Nov 23, 2023 · 82aff4d · 82aff4d
1 parent 8315e4e
commit 82aff4d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/fetch_test.rb b/test/fetch_test.rb
@@ -238,7 +238,7 @@ def test_fetch_by_title_hit
     end
 
     # Id not found, use sql, SELECT id FROM records WHERE title = '...' LIMIT 1"
-    Item.connection.expects(:exec_query).returns(ActiveRecord::Result.new(["id"], [[1]]))
+    Item.connection.expects(:internal_exec_query).returns(ActiveRecord::Result.new(["id"], [[1]]))
 
     result = Item.fetch_by_title("bob")
     assert_instance_of(Item, result)

diff --git a/test/index_cache_test.rb b/test/index_cache_test.rb
@@ -34,7 +34,7 @@ def test_fetch_with_garbage_input
   def test_fetch_with_unique_adds_limit_clause
     Item.cache_index(:title, :id, unique: true)
 
-    Item.connection.expects(:exec_query)
+    Item.connection.expects(:internal_exec_query)
       .with(regexp_matches(/ LIMIT [1?]\Z/i), any_parameters)
       .returns(ActiveRecord::Result.new([], []))