aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord
diff options
context:
space:
mode:
Diffstat (limited to 'activerecord')
-rw-r--r--activerecord/test/cases/attribute_methods_test.rb5
1 files changed, 3 insertions, 2 deletions
diff --git a/activerecord/test/cases/attribute_methods_test.rb b/activerecord/test/cases/attribute_methods_test.rb
index 7c9383b194..17cf1ec907 100644
--- a/activerecord/test/cases/attribute_methods_test.rb
+++ b/activerecord/test/cases/attribute_methods_test.rb
@@ -54,8 +54,9 @@ class AttributeMethodsTest < ActiveRecord::TestCase
end
def test_caching_nil_primary_key
- Minimalistic.expects(:reset_primary_key).returns(nil).once
- Minimalistic.create!
+ klass = Class.new(Minimalistic)
+ klass.expects(:reset_primary_key).returns(nil).once
+ 2.times { klass.primary_key }
end
def test_attribute_keys_on_new_instance