diff --git a/spec/mocha_extensions.rb b/spec/mocha_extensions.rb index 73862746af7..d7f199080ad 100644 --- a/spec/mocha_extensions.rb +++ b/spec/mocha_extensions.rb @@ -49,11 +49,11 @@ class ActiveRecord::Base ar_obj end - def self.instantiate_with_any_instantiation(a) - if obj = @@any_instantiation[[base_class, a['id'].to_i]] + def self.instantiate_with_any_instantiation(*args) + if obj = @@any_instantiation[[base_class, args.first['id'].to_i]] obj else - instantiate_without_any_instantiation(a) + instantiate_without_any_instantiation(*args) end end class << self