Allow preload and eager_load to work on relations at the same time.

This commit is contained in:
Emilio Tagua 2009-10-07 11:57:59 -03:00
parent c3b4da7796
commit 1d5854826b
2 changed files with 30 additions and 6 deletions

View File

@ -26,9 +26,9 @@ module ActiveRecord
end
def to_a
if @eager_load_associations.any?
records = catch :invalid_query do
@klass.send(:find_with_associations, {
records = if @eager_load_associations.any?
catch :invalid_query do
return @klass.send(:find_with_associations, {
:select => @relation.send(:select_clauses).join(', '),
:joins => @relation.joins(relation),
:group => @relation.send(:group_clauses).join(', '),
@ -38,12 +38,14 @@ module ActiveRecord
},
ActiveRecord::Associations::ClassMethods::JoinDependency.new(@klass, @eager_load_associations, nil))
end
[]
else
records = @klass.find_by_sql(@relation.to_sql)
@klass.send(:preload_associations, records, @associations_to_preload) unless @associations_to_preload.empty?
records.each { |record| record.readonly! } if @readonly
@klass.find_by_sql(@relation.to_sql)
end
@klass.send(:preload_associations, records, @associations_to_preload) unless @associations_to_preload.empty?
records.each { |record| record.readonly! } if @readonly
records
end

View File

@ -4,6 +4,7 @@ require 'models/topic'
require 'models/comment'
require 'models/reply'
require 'models/author'
require 'models/comment'
require 'models/entrant'
require 'models/developer'
require 'models/company'
@ -125,5 +126,26 @@ class RelationTest < ActiveRecord::TestCase
assert_equal 'Jamis', DeveloperCalledJamis.create!.name
end
def test_loading_with_one_association
posts = Post.all(:include => :comments).to_a
post = posts.find { |p| p.id == 1 }
assert_equal 2, post.comments.size
assert post.comments.include?(comments(:greetings))
post = Post.find(:first, :include => :comments, :conditions => "posts.title = 'Welcome to the weblog'")
assert_equal 2, post.comments.size
assert post.comments.include?(comments(:greetings))
posts = Post.all(:include => :last_comment).to_a
post = posts.find { |p| p.id == 1 }
assert_equal Post.find(1).last_comment, post.last_comment
end
def test_loading_with_one_association_with_non_preload
posts = Post.all(:include => :last_comment, :order => 'comments.id DESC').to_a
post = posts.find { |p| p.id == 1 }
assert_equal Post.find(1).last_comment, post.last_comment
end
end