diff --git a/lib/replicate/active_record.rb b/lib/replicate/active_record.rb index 0889fdc..2980cef 100644 --- a/lib/replicate/active_record.rb +++ b/lib/replicate/active_record.rb @@ -300,12 +300,12 @@ def create_or_update_replicant(instance, attributes) def replicate_disable_callbacks(instance) if ::ActiveRecord::VERSION::MAJOR >= 3 # AR 3.1.x - def instance.run_callbacks(*args); yield; end + def instance.run_callbacks(*args); yield if block_given?; end # AR 3.0.x - def instance._run_save_callbacks(*args); yield; end - def instance._run_create_callbacks(*args); yield; end - def instance._run_update_callbacks(*args); yield; end + def instance._run_save_callbacks(*args); yield if block_given?; end + def instance._run_create_callbacks(*args); yield if block_given?; end + def instance._run_update_callbacks(*args); yield if block_given?; end else # AR 2.x def instance.callback(*args)