diff options
author | Andrew White <andyw@pixeltrix.co.uk> | 2013-03-15 03:31:30 -0700 |
---|---|---|
committer | Andrew White <andyw@pixeltrix.co.uk> | 2013-03-15 03:31:30 -0700 |
commit | aa073188ae5d896542fb26a3211c6938c94c263c (patch) | |
tree | be99d39ad7a99fd103f9625e2429111b1f8c2e1b | |
parent | ddb9040b433b00aa0206442aa718eee12bceb517 (diff) | |
parent | b601399b72ab56cc01368f02615af99f45d14f02 (diff) | |
download | rails-aa073188ae5d896542fb26a3211c6938c94c263c.tar.gz rails-aa073188ae5d896542fb26a3211c6938c94c263c.tar.bz2 rails-aa073188ae5d896542fb26a3211c6938c94c263c.zip |
Merge pull request #9731 from senny/9724_argument_error_on_reset_counters
`#reset_counters` verifies counter names.
-rw-r--r-- | activerecord/lib/active_record/counter_cache.rb | 3 | ||||
-rw-r--r-- | activerecord/test/cases/counter_cache_test.rb | 7 |
2 files changed, 9 insertions, 1 deletions
diff --git a/activerecord/lib/active_record/counter_cache.rb b/activerecord/lib/active_record/counter_cache.rb index 81f92db271..81cca37939 100644 --- a/activerecord/lib/active_record/counter_cache.rb +++ b/activerecord/lib/active_record/counter_cache.rb @@ -11,7 +11,7 @@ module ActiveRecord # ==== Parameters # # * +id+ - The id of the object you wish to reset a counter on. - # * +counters+ - One or more counter names to reset + # * +counters+ - One or more association counters to reset # # ==== Examples # @@ -21,6 +21,7 @@ module ActiveRecord object = find(id) counters.each do |association| has_many_association = reflect_on_association(association.to_sym) + raise ArgumentError, "'#{self.name}' has no association called '#{association}'" unless has_many_association if has_many_association.is_a? ActiveRecord::Reflection::ThroughReflection has_many_association = has_many_association.through_reflection diff --git a/activerecord/test/cases/counter_cache_test.rb b/activerecord/test/cases/counter_cache_test.rb index fc46a249c8..7de2ceae88 100644 --- a/activerecord/test/cases/counter_cache_test.rb +++ b/activerecord/test/cases/counter_cache_test.rb @@ -131,4 +131,11 @@ class CounterCacheTest < ActiveRecord::TestCase Subscriber.reset_counters(subscriber.id, 'books') end end + + test "the passed symbol needs to be an association name" do + e = assert_raises(ArgumentError) do + Topic.reset_counters(@topic.id, :replies_count) + end + assert_equal "'Topic' has no association called 'replies_count'", e.message + end end |