From ad389286bbce5a43de34dc2df3968fa1f47181e1 Mon Sep 17 00:00:00 2001 From: Rob Looby Date: Fri, 8 May 2015 13:52:33 -0500 Subject: [PATCH] put dynamic matchers on the GeneratedAssociationMethods instead of model --- activerecord/lib/active_record/dynamic_matchers.rb | 6 +++--- activerecord/test/cases/finder_test.rb | 13 +++++++++++++ activerecord/test/models/electron.rb | 8 ++++++++ 3 files changed, 24 insertions(+), 3 deletions(-) diff --git a/activerecord/lib/active_record/dynamic_matchers.rb b/activerecord/lib/active_record/dynamic_matchers.rb index b6dd6814db..78fac1f3bb 100644 --- a/activerecord/lib/active_record/dynamic_matchers.rb +++ b/activerecord/lib/active_record/dynamic_matchers.rb @@ -16,7 +16,7 @@ module ActiveRecord if match && match.valid? match.define - send(name, *arguments, &block) + generated_association_methods.send(name, *arguments, &block) else super end @@ -60,7 +60,7 @@ module ActiveRecord end def define - model.class_eval <<-CODE, __FILE__, __LINE__ + 1 + model.generated_association_methods.class_eval <<-CODE, __FILE__, __LINE__ + 1 def self.#{name}(#{signature}) #{body} end @@ -70,7 +70,7 @@ module ActiveRecord private def body - "#{finder}(#{attributes_hash})" + "#{model}.#{finder}(#{attributes_hash})" end # The parameters in the signature may have reserved Ruby words, in order diff --git a/activerecord/test/cases/finder_test.rb b/activerecord/test/cases/finder_test.rb index 4b819a82e8..daf8299242 100644 --- a/activerecord/test/cases/finder_test.rb +++ b/activerecord/test/cases/finder_test.rb @@ -17,6 +17,7 @@ require 'models/matey' require 'models/dog' require 'models/car' require 'models/tyre' +require 'models/electron' class FinderTest < ActiveRecord::TestCase fixtures :companies, :topics, :entrants, :developers, :developers_projects, :posts, :comments, :accounts, :authors, :customers, :categories, :categorizations, :cars @@ -799,6 +800,18 @@ class FinderTest < ActiveRecord::TestCase assert_nil Topic.find_by_title("The First Topic!") end + def test_find_by_does_not_overwrite_method_on_class + Electron.create(name: 'First Electron') + + assert_equal 0, Electron.times_called_find_by_name + + Electron.find_by_name('First Electron') + assert_equal 1, Electron.times_called_find_by_name + + Electron.find_by_name('Some Other Electron') + assert_equal 2, Electron.times_called_find_by_name + end + def test_find_by_one_attribute_bang assert_equal topics(:first), Topic.find_by_title!("The First Topic") assert_raises_with_message(ActiveRecord::RecordNotFound, "Couldn't find Topic") do diff --git a/activerecord/test/models/electron.rb b/activerecord/test/models/electron.rb index 6fc270673f..aacdaa71b7 100644 --- a/activerecord/test/models/electron.rb +++ b/activerecord/test/models/electron.rb @@ -2,4 +2,12 @@ class Electron < ActiveRecord::Base belongs_to :molecule validates_presence_of :name + + cattr_reader :times_called_find_by_name + @@times_called_find_by_name = 0 + + def self.find_by_name(name) + @@times_called_find_by_name += 1 + super(name) + end end