mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
cdb5132c41
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@5305 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
480 lines
18 KiB
Ruby
480 lines
18 KiB
Ruby
require 'abstract_unit'
|
|
require 'fixtures/tag'
|
|
require 'fixtures/tagging'
|
|
require 'fixtures/post'
|
|
require 'fixtures/comment'
|
|
require 'fixtures/author'
|
|
require 'fixtures/category'
|
|
require 'fixtures/categorization'
|
|
|
|
class AssociationsJoinModelTest < Test::Unit::TestCase
|
|
self.use_transactional_fixtures = false
|
|
fixtures :posts, :authors, :categories, :categorizations, :comments, :tags, :taggings, :author_favorites
|
|
|
|
def test_has_many
|
|
assert authors(:david).categories.include?(categories(:general))
|
|
end
|
|
|
|
def test_has_many_inherited
|
|
assert authors(:mary).categories.include?(categories(:sti_test))
|
|
end
|
|
|
|
def test_inherited_has_many
|
|
assert categories(:sti_test).authors.include?(authors(:mary))
|
|
end
|
|
|
|
def test_has_many_uniq_through_join_model
|
|
assert_equal 2, authors(:mary).categorized_posts.size
|
|
assert_equal 1, authors(:mary).unique_categorized_posts.size
|
|
end
|
|
|
|
def test_polymorphic_has_many
|
|
assert posts(:welcome).taggings.include?(taggings(:welcome_general))
|
|
end
|
|
|
|
def test_polymorphic_has_one
|
|
assert_equal taggings(:welcome_general), posts(:welcome).tagging
|
|
end
|
|
|
|
def test_polymorphic_belongs_to
|
|
assert_equal posts(:welcome), posts(:welcome).taggings.first.taggable
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model
|
|
assert_equal tags(:general), tag = posts(:welcome).tags.first
|
|
assert_no_queries do
|
|
tag.tagging
|
|
end
|
|
end
|
|
|
|
def test_count_polymorphic_has_many
|
|
assert_equal 1, posts(:welcome).taggings.count
|
|
assert_equal 1, posts(:welcome).tags.count
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_find
|
|
assert_equal tags(:general), tag = posts(:welcome).tags.find(:first)
|
|
assert_no_queries do
|
|
tag.tagging
|
|
end
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_include_on_source_reflection
|
|
assert_equal tags(:general), tag = posts(:welcome).funky_tags.first
|
|
assert_no_queries do
|
|
tag.tagging
|
|
end
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_include_on_source_reflection_with_find
|
|
assert_equal tags(:general), tag = posts(:welcome).funky_tags.find(:first)
|
|
assert_no_queries do
|
|
tag.tagging
|
|
end
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_disabled_include
|
|
assert_equal tags(:general), tag = posts(:welcome).tags.add_joins_and_select.first
|
|
assert_queries 1 do
|
|
tag.tagging
|
|
end
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_custom_select_and_joins
|
|
assert_equal tags(:general), tag = posts(:welcome).tags.add_joins_and_select.first
|
|
tag.author_id
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_custom_foreign_key
|
|
assert_equal tags(:misc), taggings(:welcome_general).super_tag
|
|
assert_equal tags(:misc), posts(:welcome).super_tags.first
|
|
end
|
|
|
|
def test_polymorphic_has_many_create_model_with_inheritance_and_custom_base_class
|
|
post = SubStiPost.create :title => 'SubStiPost', :body => 'SubStiPost body'
|
|
assert_instance_of SubStiPost, post
|
|
|
|
tagging = tags(:misc).taggings.create(:taggable => post)
|
|
assert_equal "SubStiPost", tagging.taggable_type
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_inheritance
|
|
assert_equal tags(:general), posts(:thinking).tags.first
|
|
end
|
|
|
|
def test_polymorphic_has_many_going_through_join_model_with_inheritance_with_custom_class_name
|
|
assert_equal tags(:general), posts(:thinking).funky_tags.first
|
|
end
|
|
|
|
def test_polymorphic_has_many_create_model_with_inheritance
|
|
post = posts(:thinking)
|
|
assert_instance_of SpecialPost, post
|
|
|
|
tagging = tags(:misc).taggings.create(:taggable => post)
|
|
assert_equal "Post", tagging.taggable_type
|
|
end
|
|
|
|
def test_polymorphic_has_one_create_model_with_inheritance
|
|
tagging = tags(:misc).create_tagging(:taggable => posts(:thinking))
|
|
assert_equal "Post", tagging.taggable_type
|
|
end
|
|
|
|
def test_set_polymorphic_has_many
|
|
tagging = tags(:misc).taggings.create
|
|
posts(:thinking).taggings << tagging
|
|
assert_equal "Post", tagging.taggable_type
|
|
end
|
|
|
|
def test_set_polymorphic_has_one
|
|
tagging = tags(:misc).taggings.create
|
|
posts(:thinking).tagging = tagging
|
|
assert_equal "Post", tagging.taggable_type
|
|
end
|
|
|
|
def test_create_polymorphic_has_many_with_scope
|
|
old_count = posts(:welcome).taggings.count
|
|
tagging = posts(:welcome).taggings.create(:tag => tags(:misc))
|
|
assert_equal "Post", tagging.taggable_type
|
|
assert_equal old_count+1, posts(:welcome).taggings.count
|
|
end
|
|
|
|
def test_create_bang_polymorphic_with_has_many_scope
|
|
old_count = posts(:welcome).taggings.count
|
|
tagging = posts(:welcome).taggings.create!(:tag => tags(:misc))
|
|
assert_equal "Post", tagging.taggable_type
|
|
assert_equal old_count+1, posts(:welcome).taggings.count
|
|
end
|
|
|
|
def test_create_polymorphic_has_one_with_scope
|
|
old_count = Tagging.count
|
|
tagging = posts(:welcome).tagging.create(:tag => tags(:misc))
|
|
assert_equal "Post", tagging.taggable_type
|
|
assert_equal old_count+1, Tagging.count
|
|
end
|
|
|
|
def test_delete_polymorphic_has_many_with_delete_all
|
|
assert_equal 1, posts(:welcome).taggings.count
|
|
posts(:welcome).taggings.first.update_attribute :taggable_type, 'PostWithHasManyDeleteAll'
|
|
post = find_post_with_dependency(1, :has_many, :taggings, :delete_all)
|
|
|
|
old_count = Tagging.count
|
|
post.destroy
|
|
assert_equal old_count-1, Tagging.count
|
|
assert_equal 0, posts(:welcome).taggings.count
|
|
end
|
|
|
|
def test_delete_polymorphic_has_many_with_destroy
|
|
assert_equal 1, posts(:welcome).taggings.count
|
|
posts(:welcome).taggings.first.update_attribute :taggable_type, 'PostWithHasManyDestroy'
|
|
post = find_post_with_dependency(1, :has_many, :taggings, :destroy)
|
|
|
|
old_count = Tagging.count
|
|
post.destroy
|
|
assert_equal old_count-1, Tagging.count
|
|
assert_equal 0, posts(:welcome).taggings.count
|
|
end
|
|
|
|
def test_delete_polymorphic_has_many_with_nullify
|
|
assert_equal 1, posts(:welcome).taggings.count
|
|
posts(:welcome).taggings.first.update_attribute :taggable_type, 'PostWithHasManyNullify'
|
|
post = find_post_with_dependency(1, :has_many, :taggings, :nullify)
|
|
|
|
old_count = Tagging.count
|
|
post.destroy
|
|
assert_equal old_count, Tagging.count
|
|
assert_equal 0, posts(:welcome).taggings.count
|
|
end
|
|
|
|
def test_delete_polymorphic_has_one_with_destroy
|
|
assert posts(:welcome).tagging
|
|
posts(:welcome).tagging.update_attribute :taggable_type, 'PostWithHasOneDestroy'
|
|
post = find_post_with_dependency(1, :has_one, :tagging, :destroy)
|
|
|
|
old_count = Tagging.count
|
|
post.destroy
|
|
assert_equal old_count-1, Tagging.count
|
|
assert_nil posts(:welcome).tagging(true)
|
|
end
|
|
|
|
def test_delete_polymorphic_has_one_with_nullify
|
|
assert posts(:welcome).tagging
|
|
posts(:welcome).tagging.update_attribute :taggable_type, 'PostWithHasOneNullify'
|
|
post = find_post_with_dependency(1, :has_one, :tagging, :nullify)
|
|
|
|
old_count = Tagging.count
|
|
post.destroy
|
|
assert_equal old_count, Tagging.count
|
|
assert_nil posts(:welcome).tagging(true)
|
|
end
|
|
|
|
def test_has_many_with_piggyback
|
|
assert_equal "2", categories(:sti_test).authors.first.post_id.to_s
|
|
end
|
|
|
|
def test_include_has_many_through
|
|
posts = Post.find(:all, :order => 'posts.id')
|
|
posts_with_authors = Post.find(:all, :include => :authors, :order => 'posts.id')
|
|
assert_equal posts.length, posts_with_authors.length
|
|
posts.length.times do |i|
|
|
assert_equal posts[i].authors.length, assert_no_queries { posts_with_authors[i].authors.length }
|
|
end
|
|
end
|
|
|
|
def test_include_polymorphic_has_one
|
|
post = Post.find_by_id(posts(:welcome).id, :include => :tagging)
|
|
tagging = taggings(:welcome_general)
|
|
assert_no_queries do
|
|
assert_equal tagging, post.tagging
|
|
end
|
|
end
|
|
|
|
def test_include_polymorphic_has_many_through
|
|
posts = Post.find(:all, :order => 'posts.id')
|
|
posts_with_tags = Post.find(:all, :include => :tags, :order => 'posts.id')
|
|
assert_equal posts.length, posts_with_tags.length
|
|
posts.length.times do |i|
|
|
assert_equal posts[i].tags.length, assert_no_queries { posts_with_tags[i].tags.length }
|
|
end
|
|
end
|
|
|
|
def test_include_polymorphic_has_many
|
|
posts = Post.find(:all, :order => 'posts.id')
|
|
posts_with_taggings = Post.find(:all, :include => :taggings, :order => 'posts.id')
|
|
assert_equal posts.length, posts_with_taggings.length
|
|
posts.length.times do |i|
|
|
assert_equal posts[i].taggings.length, assert_no_queries { posts_with_taggings[i].taggings.length }
|
|
end
|
|
end
|
|
|
|
def test_has_many_find_all
|
|
assert_equal [categories(:general)], authors(:david).categories.find(:all)
|
|
end
|
|
|
|
def test_has_many_find_first
|
|
assert_equal categories(:general), authors(:david).categories.find(:first)
|
|
end
|
|
|
|
def test_has_many_with_hash_conditions
|
|
assert_equal categories(:general), authors(:david).categories_like_general.find(:first)
|
|
end
|
|
|
|
def test_has_many_find_conditions
|
|
assert_equal categories(:general), authors(:david).categories.find(:first, :conditions => "categories.name = 'General'")
|
|
assert_equal nil, authors(:david).categories.find(:first, :conditions => "categories.name = 'Technology'")
|
|
end
|
|
|
|
def test_has_many_class_methods_called_by_method_missing
|
|
assert_equal categories(:general), authors(:david).categories.find_all_by_name('General').first
|
|
# assert_equal nil, authors(:david).categories.find_by_name('Technology')
|
|
end
|
|
|
|
def test_has_many_going_through_join_model_with_custom_foreign_key
|
|
assert_equal [], posts(:thinking).authors
|
|
assert_equal [authors(:mary)], posts(:authorless).authors
|
|
end
|
|
|
|
def test_belongs_to_polymorphic_with_counter_cache
|
|
assert_equal 0, posts(:welcome)[:taggings_count]
|
|
tagging = posts(:welcome).taggings.create(:tag => tags(:general))
|
|
assert_equal 1, posts(:welcome, :reload)[:taggings_count]
|
|
tagging.destroy
|
|
assert posts(:welcome, :reload)[:taggings_count].zero?
|
|
end
|
|
|
|
def test_unavailable_through_reflection
|
|
assert_raises (ActiveRecord::HasManyThroughAssociationNotFoundError) { authors(:david).nothings }
|
|
end
|
|
|
|
def test_has_many_through_join_model_with_conditions
|
|
assert_equal [], posts(:welcome).invalid_taggings
|
|
assert_equal [], posts(:welcome).invalid_tags
|
|
end
|
|
|
|
def test_has_many_polymorphic
|
|
assert_raises ActiveRecord::HasManyThroughAssociationPolymorphicError do
|
|
assert_equal [posts(:welcome), posts(:thinking)], tags(:general).taggables
|
|
end
|
|
assert_raises ActiveRecord::EagerLoadPolymorphicError do
|
|
assert_equal [posts(:welcome), posts(:thinking)], tags(:general).taggings.find(:all, :include => :taggable)
|
|
end
|
|
end
|
|
|
|
def test_has_many_through_has_many_find_all
|
|
assert_equal comments(:greetings), authors(:david).comments.find(:all, :order => 'comments.id').first
|
|
end
|
|
|
|
def test_has_many_through_has_many_find_all_with_custom_class
|
|
assert_equal comments(:greetings), authors(:david).funky_comments.find(:all, :order => 'comments.id').first
|
|
end
|
|
|
|
def test_has_many_through_has_many_find_first
|
|
assert_equal comments(:greetings), authors(:david).comments.find(:first, :order => 'comments.id')
|
|
end
|
|
|
|
def test_has_many_through_has_many_find_conditions
|
|
options = { :conditions => "comments.#{QUOTED_TYPE}='SpecialComment'", :order => 'comments.id' }
|
|
assert_equal comments(:does_it_hurt), authors(:david).comments.find(:first, options)
|
|
end
|
|
|
|
def test_has_many_through_has_many_find_by_id
|
|
assert_equal comments(:more_greetings), authors(:david).comments.find(2)
|
|
end
|
|
|
|
def test_has_many_through_polymorphic_has_one
|
|
assert_raise(ActiveRecord::HasManyThroughSourceAssociationMacroError) { authors(:david).tagging }
|
|
end
|
|
|
|
def test_has_many_through_polymorphic_has_many
|
|
assert_equal [taggings(:welcome_general), taggings(:thinking_general)], authors(:david).taggings.uniq.sort_by { |t| t.id }
|
|
end
|
|
|
|
def test_include_has_many_through_polymorphic_has_many
|
|
author = Author.find_by_id(authors(:david).id, :include => :taggings)
|
|
expected_taggings = [taggings(:welcome_general), taggings(:thinking_general)]
|
|
assert_no_queries do
|
|
assert_equal expected_taggings, author.taggings.uniq.sort_by { |t| t.id }
|
|
end
|
|
end
|
|
|
|
def test_has_many_through_has_many_through
|
|
assert_raise(ActiveRecord::HasManyThroughSourceAssociationMacroError) { authors(:david).tags }
|
|
end
|
|
|
|
def test_has_many_through_habtm
|
|
assert_raise(ActiveRecord::HasManyThroughSourceAssociationMacroError) { authors(:david).post_categories }
|
|
end
|
|
|
|
def test_eager_load_has_many_through_has_many
|
|
author = Author.find :first, :conditions => ['name = ?', 'David'], :include => :comments, :order => 'comments.id'
|
|
SpecialComment.new; VerySpecialComment.new
|
|
assert_no_queries do
|
|
assert_equal [1,2,3,5,6,7,8,9,10], author.comments.collect(&:id)
|
|
end
|
|
end
|
|
|
|
def test_eager_load_has_many_through_has_many_with_conditions
|
|
post = Post.find(:first, :include => :invalid_tags)
|
|
assert_no_queries do
|
|
post.invalid_tags
|
|
end
|
|
end
|
|
|
|
def test_eager_belongs_to_and_has_one_not_singularized
|
|
assert_nothing_raised do
|
|
Author.find(:first, :include => :author_address)
|
|
AuthorAddress.find(:first, :include => :author)
|
|
end
|
|
end
|
|
|
|
def test_self_referential_has_many_through
|
|
assert_equal [authors(:mary)], authors(:david).favorite_authors
|
|
assert_equal [], authors(:mary).favorite_authors
|
|
end
|
|
|
|
def test_add_to_self_referential_has_many_through
|
|
new_author = Author.create(:name => "Bob")
|
|
authors(:david).author_favorites.create :favorite_author => new_author
|
|
assert_equal new_author, authors(:david).reload.favorite_authors.first
|
|
end
|
|
|
|
def test_has_many_through_uses_correct_attributes
|
|
assert_nil posts(:thinking).tags.find_by_name("General").attributes["tag_id"]
|
|
end
|
|
|
|
def test_raise_error_when_adding_new_record_to_has_many_through
|
|
assert_raise(ActiveRecord::HasManyThroughCantAssociateNewRecords) { posts(:thinking).tags << tags(:general).clone }
|
|
assert_raise(ActiveRecord::HasManyThroughCantAssociateNewRecords) { posts(:thinking).clone.tags << tags(:general) }
|
|
assert_raise(ActiveRecord::HasManyThroughCantAssociateNewRecords) { posts(:thinking).tags.build }
|
|
end
|
|
|
|
def test_create_associate_when_adding_to_has_many_through
|
|
count = posts(:thinking).tags.count
|
|
push = Tag.create!(:name => 'pushme')
|
|
post_thinking = posts(:thinking)
|
|
assert_nothing_raised { post_thinking.tags << push }
|
|
assert_nil( wrong = post_thinking.tags.detect { |t| t.class != Tag },
|
|
message = "Expected a Tag in tags collection, got #{wrong.class}.")
|
|
assert_nil( wrong = post_thinking.taggings.detect { |t| t.class != Tagging },
|
|
message = "Expected a Tagging in taggings collection, got #{wrong.class}.")
|
|
assert_equal(count + 1, post_thinking.tags.size)
|
|
assert_equal(count + 1, post_thinking.tags(true).size)
|
|
|
|
assert_nothing_raised { post_thinking.tags.create!(:name => 'foo') }
|
|
assert_nil( wrong = post_thinking.tags.detect { |t| t.class != Tag },
|
|
message = "Expected a Tag in tags collection, got #{wrong.class}.")
|
|
assert_nil( wrong = post_thinking.taggings.detect { |t| t.class != Tagging },
|
|
message = "Expected a Tagging in taggings collection, got #{wrong.class}.")
|
|
assert_equal(count + 2, post_thinking.tags.size)
|
|
assert_equal(count + 2, post_thinking.tags(true).size)
|
|
|
|
assert_nothing_raised { post_thinking.tags.concat(Tag.create!(:name => 'abc'), Tag.create!(:name => 'def')) }
|
|
assert_nil( wrong = post_thinking.tags.detect { |t| t.class != Tag },
|
|
message = "Expected a Tag in tags collection, got #{wrong.class}.")
|
|
assert_nil( wrong = post_thinking.taggings.detect { |t| t.class != Tagging },
|
|
message = "Expected a Tagging in taggings collection, got #{wrong.class}.")
|
|
assert_equal(count + 4, post_thinking.tags.size)
|
|
assert_equal(count + 4, post_thinking.tags(true).size)
|
|
end
|
|
|
|
def test_adding_junk_to_has_many_through_should_raise_type_mismatch
|
|
assert_raise(ActiveRecord::AssociationTypeMismatch) { posts(:thinking).tags << "Uhh what now?" }
|
|
end
|
|
|
|
def test_adding_to_has_many_through_should_return_self
|
|
tags = posts(:thinking).tags
|
|
assert_equal tags, posts(:thinking).tags.push(tags(:general))
|
|
end
|
|
|
|
def test_delete_associate_when_deleting_from_has_many_through
|
|
count = posts(:thinking).tags.count
|
|
tags_before = posts(:thinking).tags
|
|
tag = Tag.create!(:name => 'doomed')
|
|
post_thinking = posts(:thinking)
|
|
post_thinking.tags << tag
|
|
assert_equal(count + 1, post_thinking.tags(true).size)
|
|
|
|
assert_nothing_raised { post_thinking.tags.delete(tag) }
|
|
assert_equal(count, post_thinking.tags.size)
|
|
assert_equal(count, post_thinking.tags(true).size)
|
|
assert_equal(tags_before.sort, post_thinking.tags.sort)
|
|
end
|
|
|
|
def test_delete_associate_when_deleting_from_has_many_through_with_multiple_tags
|
|
count = posts(:thinking).tags.count
|
|
tags_before = posts(:thinking).tags
|
|
doomed = Tag.create!(:name => 'doomed')
|
|
doomed2 = Tag.create!(:name => 'doomed2')
|
|
quaked = Tag.create!(:name => 'quaked')
|
|
post_thinking = posts(:thinking)
|
|
post_thinking.tags << doomed << doomed2
|
|
assert_equal(count + 2, post_thinking.tags(true).size)
|
|
|
|
assert_nothing_raised { post_thinking.tags.delete(doomed, doomed2, quaked) }
|
|
assert_equal(count, post_thinking.tags.size)
|
|
assert_equal(count, post_thinking.tags(true).size)
|
|
assert_equal(tags_before.sort, post_thinking.tags.sort)
|
|
end
|
|
|
|
def test_deleting_junk_from_has_many_through_should_raise_type_mismatch
|
|
assert_raise(ActiveRecord::AssociationTypeMismatch) { posts(:thinking).tags.delete("Uhh what now?") }
|
|
end
|
|
|
|
def test_has_many_through_sum_uses_calculations
|
|
assert_nothing_raised { authors(:david).comments.sum(:post_id) }
|
|
end
|
|
|
|
def test_has_many_through_has_many_with_sti
|
|
assert_equal [comments(:does_it_hurt)], authors(:david).special_post_comments
|
|
end
|
|
|
|
private
|
|
# create dynamic Post models to allow different dependency options
|
|
def find_post_with_dependency(post_id, association, association_name, dependency)
|
|
class_name = "PostWith#{association.to_s.classify}#{dependency.to_s.classify}"
|
|
Post.find(post_id).update_attribute :type, class_name
|
|
klass = Object.const_set(class_name, Class.new(ActiveRecord::Base))
|
|
klass.set_table_name 'posts'
|
|
klass.send(association, association_name, :as => :taggable, :dependent => dependency)
|
|
klass.find(post_id)
|
|
end
|
|
end
|