JezK
Edit File: eager_load_nested_include_test.rb
require 'cases/helper' class ShapeExpression < ActiveRecord::Base belongs_to :shape, :polymorphic => true belongs_to :paint, :polymorphic => true end class Circle < ActiveRecord::Base has_many :shape_expressions, :as => :shape end class Square < ActiveRecord::Base has_many :shape_expressions, :as => :shape end class Triangle < ActiveRecord::Base has_many :shape_expressions, :as => :shape end class PaintColor < ActiveRecord::Base has_many :shape_expressions, :as => :paint belongs_to :non_poly, :foreign_key => "non_poly_one_id", :class_name => "NonPolyOne" end class PaintTexture < ActiveRecord::Base has_many :shape_expressions, :as => :paint belongs_to :non_poly, :foreign_key => "non_poly_two_id", :class_name => "NonPolyTwo" end class NonPolyOne < ActiveRecord::Base has_many :paint_colors end class NonPolyTwo < ActiveRecord::Base has_many :paint_textures end class EagerLoadPolyAssocsTest < ActiveRecord::TestCase NUM_SIMPLE_OBJS = 50 NUM_SHAPE_EXPRESSIONS = 100 def setup generate_test_object_graphs end def teardown [Circle, Square, Triangle, PaintColor, PaintTexture, ShapeExpression, NonPolyOne, NonPolyTwo].each do |c| c.delete_all end end # meant to be supplied as an ID, never returns 0 def rand_simple val = (NUM_SIMPLE_OBJS * rand).round val == 0 ? 1 : val end def generate_test_object_graphs 1.upto(NUM_SIMPLE_OBJS) do [Circle, Square, Triangle, NonPolyOne, NonPolyTwo].map(&:create!) end 1.upto(NUM_SIMPLE_OBJS) do |i| PaintColor.create!(:non_poly_one_id => rand_simple) PaintTexture.create!(:non_poly_two_id => rand_simple) end 1.upto(NUM_SHAPE_EXPRESSIONS) do |i| ShapeExpression.create!(:shape_type => [Circle, Square, Triangle].rand.to_s, :shape_id => rand_simple, :paint_type => [PaintColor, PaintTexture].rand.to_s, :paint_id => rand_simple) end end def test_include_query res = 0 res = ShapeExpression.find :all, :include => [ :shape, { :paint => :non_poly } ] assert_equal NUM_SHAPE_EXPRESSIONS, res.size assert_queries(0) do res.each do |se| assert_not_nil se.paint.non_poly, "this is the association that was loading incorrectly before the change" assert_not_nil se.shape, "just making sure other associations still work" end end end end