Makes a minor fix to PredicateBuilder respect custom primary keys when

calling `Relation#where`

Author: Rick Song

activerecord/CHANGELOG.md

@@ -1,3 +1,9 @@
+*   Fixes custom primary keys for associations when calling `Relation#where`
+
+    Fixes #23327.
+
+    *Rick Song*
+
 *   No longer pass deprecated option `-i` to `pg_dump`.
 
     *Paul Sadauskas*

activerecord/lib/active_record/relation/predicate_builder.rb

@@ -56,7 +56,8 @@ def self.expand(klass, table, column, value)
       # For polymorphic relationships, find the foreign key and type:
       # PriceEstimate.where(estimate_of: treasure)
       if klass && reflection = klass._reflect_on_association(column.to_sym)
-        if reflection.polymorphic? && base_class = polymorphic_base_class_from_value(value)
+        base_class = polymorphic_base_class_from_value(value)
+        if reflection.polymorphic? && base_class
           queries << build(table[reflection.foreign_type], base_class)
         end
 

activerecord/test/cases/relation/where_test.rb

@@ -2,6 +2,7 @@
 require "models/author"
 require "models/binary"
 require "models/cake_designer"
+require "models/category"
 require "models/chef"
 require "models/comment"
 require "models/edge"
@@ -14,7 +15,7 @@
 
 module ActiveRecord
   class WhereTest < ActiveRecord::TestCase
-    fixtures :posts, :edges, :authors, :binaries, :essays
+    fixtures :posts, :edges, :authors, :categories, :binaries, :essays
 
     def test_where_copies_bind_params
       author = authors(:david)
@@ -207,35 +208,70 @@ def test_where_with_blank_conditions
     end
 
     def test_where_on_association_with_custom_primary_key
+      category = categories(:general)
+      essay = Essay.where(category: category).first
+
+      assert_equal essays(:david_modest_proposal), essay
+    end
+
+    def test_where_on_association_with_custom_primary_key_with_relation
+      category = categories(:general)
+      essay = Essay.where(category: Category.where(id: category.id)).first
+
+      assert_equal essays(:david_modest_proposal), essay
+    end
+
+    def test_where_on_association_with_relation_performs_subselect_not_two_queries
+      category = categories(:general)
+
+      assert_queries(1) do
+        Essay.where(category: Category.where(id: category.id)).to_a
+      end
+    end
+
+    def test_where_on_association_with_custom_primary_key_with_array_of_base
+      category = categories(:general)
+      essay = Essay.where(category: [category]).first
+
+      assert_equal essays(:david_modest_proposal), essay
+    end
+
+    def test_where_on_association_with_custom_primary_key_with_array_of_ids
+      essay = Essay.where(category: ["General"]).first
+
+      assert_equal essays(:david_modest_proposal), essay
+    end
+
+    def test_where_on_polymorphic_association_with_custom_primary_key
       author = authors(:david)
       essay = Essay.where(writer: author).first
 
       assert_equal essays(:david_modest_proposal), essay
     end
 
-    def test_where_on_association_with_custom_primary_key_with_relation
+    def test_where_on_polymorphic_association_with_custom_primary_key_with_relation
       author = authors(:david)
       essay = Essay.where(writer: Author.where(id: author.id)).first
 
       assert_equal essays(:david_modest_proposal), essay
     end
 
-    def test_where_on_association_with_relation_performs_subselect_not_two_queries
+    def test_where_on_polymorphic_association_with_relation_performs_subselect_not_two_queries
       author = authors(:david)
 
       assert_queries(1) do
         Essay.where(writer: Author.where(id: author.id)).to_a
       end
     end
 
-    def test_where_on_association_with_custom_primary_key_with_array_of_base
+    def test_where_on_polymorphic_association_with_custom_primary_key_with_array_of_base
       author = authors(:david)
       essay = Essay.where(writer: [author]).first
 
       assert_equal essays(:david_modest_proposal), essay
     end
 
-    def test_where_on_association_with_custom_primary_key_with_array_of_ids
+    def test_where_on_polymorphic_association_with_custom_primary_key_with_array_of_ids
       essay = Essay.where(writer: ["David"]).first
 
       assert_equal essays(:david_modest_proposal), essay