same for tests

Shopify · Nov 23, 2023 · 28edb79 · 28edb79
1 parent 7b63945
commit 28edb79
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/fetch_test.rb b/test/fetch_test.rb
@@ -282,11 +282,11 @@ def test_fetch_cache_hit_does_not_checkout_database_connection
  @record.save!
  record = Item.fetch(@record.id)
 
- ActiveRecord::Base.clear_active_connections!
+ ActiveRecord::Base.clear_active_connections!(ActiveRecord::Base.current_role)
 
  assert_equal(record, Item.fetch(@record.id))
 
- assert_equal(false, ActiveRecord::Base.connection_handler.active_connections?)
+ assert_equal(false, ActiveRecord::Base.connection_handler.active_connections?(ActiveRecord::Base.current_role))
  end
 
  def test_fetch_raises_when_called_on_a_scope