Fix while_preventing_writes for legacy version

We need to pass the block along to the connection handler's `while_preventing_writes`.
rails · Nov 2, 2020 · 9c8a7f4 · 9c8a7f4
1 parent 0919871
commit 9c8a7f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/activerecord/lib/active_record/connection_handling.rb b/activerecord/lib/active_record/connection_handling.rb
@@ -202,7 +202,7 @@ def connecting_to(role: default_role, shard: default_shard, prevent_writes: fals
     # method.
     def while_preventing_writes(enabled = true, &block)
       if legacy_connection_handling
-        connection_handler.while_preventing_writes(enabled)
+        connection_handler.while_preventing_writes(enabled, &block)
       else
         connected_to(role: current_role, prevent_writes: enabled, &block)
       end