2008-05-18 06:22:34 +02:00
|
|
|
require "cases/helper"
|
|
|
|
require 'models/post'
|
|
|
|
require 'models/person'
|
2009-03-16 15:55:30 +01:00
|
|
|
require 'models/reference'
|
|
|
|
require 'models/job'
|
2008-05-18 06:22:34 +02:00
|
|
|
require 'models/reader'
|
2008-10-27 07:47:01 +01:00
|
|
|
require 'models/comment'
|
2009-02-04 21:26:08 +01:00
|
|
|
require 'models/tag'
|
|
|
|
require 'models/tagging'
|
|
|
|
require 'models/author'
|
2009-03-16 15:55:30 +01:00
|
|
|
require 'models/owner'
|
|
|
|
require 'models/pet'
|
|
|
|
require 'models/toy'
|
2009-09-05 09:01:46 +02:00
|
|
|
require 'models/contract'
|
|
|
|
require 'models/company'
|
|
|
|
require 'models/developer'
|
2008-05-18 06:22:34 +02:00
|
|
|
|
|
|
|
class HasManyThroughAssociationsTest < ActiveRecord::TestCase
|
2009-09-05 09:01:46 +02:00
|
|
|
fixtures :posts, :readers, :people, :comments, :authors, :owners, :pets, :toys,
|
|
|
|
:companies
|
2008-05-18 06:22:34 +02:00
|
|
|
|
|
|
|
def test_associate_existing
|
|
|
|
assert_queries(2) { posts(:thinking);people(:david) }
|
2008-10-27 07:47:01 +01:00
|
|
|
|
|
|
|
posts(:thinking).people
|
|
|
|
|
2008-05-18 06:22:34 +02:00
|
|
|
assert_queries(1) do
|
2008-10-27 07:47:01 +01:00
|
|
|
posts(:thinking).people << people(:david)
|
2008-05-18 06:22:34 +02:00
|
|
|
end
|
|
|
|
|
|
|
|
assert_queries(1) do
|
|
|
|
assert posts(:thinking).people.include?(people(:david))
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:thinking).reload.people(true).include?(people(:david))
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associating_new
|
|
|
|
assert_queries(1) { posts(:thinking) }
|
|
|
|
new_person = nil # so block binding catches it
|
|
|
|
|
|
|
|
assert_queries(0) do
|
|
|
|
new_person = Person.new :first_name => 'bob'
|
|
|
|
end
|
|
|
|
|
|
|
|
# Associating new records always saves them
|
|
|
|
# Thus, 1 query for the new person record, 1 query for the new join table record
|
|
|
|
assert_queries(2) do
|
|
|
|
posts(:thinking).people << new_person
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_queries(1) do
|
|
|
|
assert posts(:thinking).people.include?(new_person)
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:thinking).reload.people(true).include?(new_person)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_new_by_building
|
|
|
|
assert_queries(1) { posts(:thinking) }
|
|
|
|
|
|
|
|
assert_queries(0) do
|
|
|
|
posts(:thinking).people.build(:first_name=>"Bob")
|
|
|
|
posts(:thinking).people.new(:first_name=>"Ted")
|
|
|
|
end
|
|
|
|
|
|
|
|
# Should only need to load the association once
|
|
|
|
assert_queries(1) do
|
|
|
|
assert posts(:thinking).people.collect(&:first_name).include?("Bob")
|
|
|
|
assert posts(:thinking).people.collect(&:first_name).include?("Ted")
|
|
|
|
end
|
|
|
|
|
|
|
|
# 2 queries for each new record (1 to save the record itself, 1 for the join model)
|
|
|
|
# * 2 new records = 4
|
|
|
|
# + 1 query to save the actual post = 5
|
|
|
|
assert_queries(5) do
|
2008-06-02 08:35:38 +02:00
|
|
|
posts(:thinking).body += '-changed'
|
2008-05-18 06:22:34 +02:00
|
|
|
posts(:thinking).save
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:thinking).reload.people(true).collect(&:first_name).include?("Bob")
|
|
|
|
assert posts(:thinking).reload.people(true).collect(&:first_name).include?("Ted")
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_delete_association
|
|
|
|
assert_queries(2){posts(:welcome);people(:michael); }
|
|
|
|
|
|
|
|
assert_queries(1) do
|
|
|
|
posts(:welcome).people.delete(people(:michael))
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_queries(1) do
|
|
|
|
assert posts(:welcome).people.empty?
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:welcome).reload.people(true).empty?
|
|
|
|
end
|
|
|
|
|
2009-03-16 15:55:30 +01:00
|
|
|
def test_destroy_association
|
|
|
|
assert_difference "Person.count", -1 do
|
|
|
|
posts(:welcome).people.destroy(people(:michael))
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:welcome).reload.people.empty?
|
|
|
|
assert posts(:welcome).people(true).empty?
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_destroy_all
|
|
|
|
assert_difference "Person.count", -1 do
|
|
|
|
posts(:welcome).people.destroy_all
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:welcome).reload.people.empty?
|
|
|
|
assert posts(:welcome).people(true).empty?
|
|
|
|
end
|
|
|
|
|
2008-05-18 06:22:34 +02:00
|
|
|
def test_replace_association
|
|
|
|
assert_queries(4){posts(:welcome);people(:david);people(:michael); posts(:welcome).people(true)}
|
|
|
|
|
|
|
|
# 1 query to delete the existing reader (michael)
|
|
|
|
# 1 query to associate the new reader (david)
|
|
|
|
assert_queries(2) do
|
|
|
|
posts(:welcome).people = [people(:david)]
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_queries(0){
|
|
|
|
assert posts(:welcome).people.include?(people(:david))
|
|
|
|
assert !posts(:welcome).people.include?(people(:michael))
|
|
|
|
}
|
|
|
|
|
|
|
|
assert posts(:welcome).reload.people(true).include?(people(:david))
|
|
|
|
assert !posts(:welcome).reload.people(true).include?(people(:michael))
|
|
|
|
end
|
|
|
|
|
2009-12-01 02:38:34 +01:00
|
|
|
def test_replace_order_is_preserved
|
|
|
|
posts(:welcome).people.clear
|
|
|
|
posts(:welcome).people = [people(:david), people(:michael)]
|
|
|
|
assert_equal [people(:david).id, people(:michael).id], posts(:welcome).readers.all(:order => 'id').map(&:person_id)
|
|
|
|
|
|
|
|
# Test the inverse order in case the first success was a coincidence
|
|
|
|
posts(:welcome).people.clear
|
|
|
|
posts(:welcome).people = [people(:michael), people(:david)]
|
|
|
|
assert_equal [people(:michael).id, people(:david).id], posts(:welcome).readers.all(:order => 'id').map(&:person_id)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_replace_by_id_order_is_preserved
|
|
|
|
posts(:welcome).people.clear
|
|
|
|
posts(:welcome).person_ids = [people(:david).id, people(:michael).id]
|
|
|
|
assert_equal [people(:david).id, people(:michael).id], posts(:welcome).readers.all(:order => 'id').map(&:person_id)
|
|
|
|
|
|
|
|
# Test the inverse order in case the first success was a coincidence
|
|
|
|
posts(:welcome).people.clear
|
|
|
|
posts(:welcome).person_ids = [people(:michael).id, people(:david).id]
|
|
|
|
assert_equal [people(:michael).id, people(:david).id], posts(:welcome).readers.all(:order => 'id').map(&:person_id)
|
|
|
|
end
|
|
|
|
|
2008-05-18 06:22:34 +02:00
|
|
|
def test_associate_with_create
|
|
|
|
assert_queries(1) { posts(:thinking) }
|
|
|
|
|
|
|
|
# 1 query for the new record, 1 for the join table record
|
|
|
|
# No need to update the actual collection yet!
|
|
|
|
assert_queries(2) do
|
|
|
|
posts(:thinking).people.create(:first_name=>"Jeb")
|
|
|
|
end
|
|
|
|
|
|
|
|
# *Now* we actually need the collection so it's loaded
|
|
|
|
assert_queries(1) do
|
|
|
|
assert posts(:thinking).people.collect(&:first_name).include?("Jeb")
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:thinking).reload.people(true).collect(&:first_name).include?("Jeb")
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_with_create_and_no_options
|
|
|
|
peeps = posts(:thinking).people.count
|
|
|
|
posts(:thinking).people.create(:first_name => 'foo')
|
|
|
|
assert_equal peeps + 1, posts(:thinking).people.count
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_with_create_exclamation_and_no_options
|
|
|
|
peeps = posts(:thinking).people.count
|
|
|
|
posts(:thinking).people.create!(:first_name => 'foo')
|
|
|
|
assert_equal peeps + 1, posts(:thinking).people.count
|
|
|
|
end
|
|
|
|
|
2009-09-05 09:01:46 +02:00
|
|
|
def test_associate_with_create_and_invalid_options
|
|
|
|
peeps = companies(:first_firm).developers.count
|
|
|
|
assert_nothing_raised { companies(:first_firm).developers.create(:name => '0') }
|
|
|
|
assert_equal peeps, companies(:first_firm).developers.count
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_with_create_and_valid_options
|
|
|
|
peeps = companies(:first_firm).developers.count
|
|
|
|
assert_nothing_raised { companies(:first_firm).developers.create(:name => 'developer') }
|
|
|
|
assert_equal peeps + 1, companies(:first_firm).developers.count
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_with_create_bang_and_invalid_options
|
|
|
|
peeps = companies(:first_firm).developers.count
|
|
|
|
assert_raises(ActiveRecord::RecordInvalid) { companies(:first_firm).developers.create!(:name => '0') }
|
|
|
|
assert_equal peeps, companies(:first_firm).developers.count
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_associate_with_create_bang_and_valid_options
|
|
|
|
peeps = companies(:first_firm).developers.count
|
|
|
|
assert_nothing_raised { companies(:first_firm).developers.create!(:name => 'developer') }
|
|
|
|
assert_equal peeps + 1, companies(:first_firm).developers.count
|
|
|
|
end
|
|
|
|
|
2008-05-18 06:22:34 +02:00
|
|
|
def test_clear_associations
|
|
|
|
assert_queries(2) { posts(:welcome);posts(:welcome).people(true) }
|
|
|
|
|
|
|
|
assert_queries(1) do
|
|
|
|
posts(:welcome).people.clear
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_queries(0) do
|
|
|
|
assert posts(:welcome).people.empty?
|
|
|
|
end
|
|
|
|
|
|
|
|
assert posts(:welcome).reload.people(true).empty?
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_association_callback_ordering
|
|
|
|
Post.reset_log
|
|
|
|
log = Post.log
|
|
|
|
post = posts(:thinking)
|
|
|
|
|
|
|
|
post.people_with_callbacks << people(:michael)
|
|
|
|
assert_equal [
|
|
|
|
[:added, :before, "Michael"],
|
|
|
|
[:added, :after, "Michael"]
|
|
|
|
], log.last(2)
|
|
|
|
|
|
|
|
post.people_with_callbacks.push(people(:david), Person.create!(:first_name => "Bob"), Person.new(:first_name => "Lary"))
|
|
|
|
assert_equal [
|
|
|
|
[:added, :before, "David"],
|
|
|
|
[:added, :after, "David"],
|
|
|
|
[:added, :before, "Bob"],
|
|
|
|
[:added, :after, "Bob"],
|
|
|
|
[:added, :before, "Lary"],
|
|
|
|
[:added, :after, "Lary"]
|
|
|
|
],log.last(6)
|
|
|
|
|
|
|
|
post.people_with_callbacks.build(:first_name => "Ted")
|
|
|
|
assert_equal [
|
|
|
|
[:added, :before, "Ted"],
|
|
|
|
[:added, :after, "Ted"]
|
|
|
|
], log.last(2)
|
|
|
|
|
|
|
|
post.people_with_callbacks.create(:first_name => "Sam")
|
|
|
|
assert_equal [
|
|
|
|
[:added, :before, "Sam"],
|
|
|
|
[:added, :after, "Sam"]
|
|
|
|
], log.last(2)
|
|
|
|
|
|
|
|
post.people_with_callbacks = [people(:michael),people(:david), Person.new(:first_name => "Julian"), Person.create!(:first_name => "Roger")]
|
|
|
|
assert_equal (%w(Ted Bob Sam Lary) * 2).sort, log[-12..-5].collect(&:last).sort
|
|
|
|
assert_equal [
|
|
|
|
[:added, :before, "Julian"],
|
|
|
|
[:added, :after, "Julian"],
|
|
|
|
[:added, :before, "Roger"],
|
|
|
|
[:added, :after, "Roger"]
|
|
|
|
], log.last(4)
|
|
|
|
|
|
|
|
post.people_with_callbacks.clear
|
|
|
|
assert_equal (%w(Michael David Julian Roger) * 2).sort, log.last(8).collect(&:last).sort
|
|
|
|
end
|
2008-09-07 07:54:05 +02:00
|
|
|
|
|
|
|
def test_dynamic_find_should_respect_association_include
|
|
|
|
# SQL error in sort clause if :include is not included
|
|
|
|
# due to Unknown column 'comments.id'
|
|
|
|
assert Person.find(1).posts_with_comments_sorted_by_comment_id.find_by_title('Welcome to the weblog')
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_count_with_include_should_alias_join_table
|
|
|
|
assert_equal 2, people(:michael).posts.count(:include => :readers)
|
|
|
|
end
|
2008-10-27 07:47:01 +01:00
|
|
|
|
2009-02-04 21:26:08 +01:00
|
|
|
def test_inner_join_with_quoted_table_name
|
|
|
|
assert_equal 2, people(:michael).jobs.size
|
|
|
|
end
|
|
|
|
|
2008-10-27 07:47:01 +01:00
|
|
|
def test_get_ids
|
|
|
|
assert_equal [posts(:welcome).id, posts(:authorless).id].sort, people(:michael).post_ids.sort
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_get_ids_for_loaded_associations
|
|
|
|
person = people(:michael)
|
|
|
|
person.posts(true)
|
|
|
|
assert_queries(0) do
|
|
|
|
person.post_ids
|
|
|
|
person.post_ids
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_get_ids_for_unloaded_associations_does_not_load_them
|
|
|
|
person = people(:michael)
|
|
|
|
assert !person.posts.loaded?
|
|
|
|
assert_equal [posts(:welcome).id, posts(:authorless).id].sort, person.post_ids.sort
|
|
|
|
assert !person.posts.loaded?
|
|
|
|
end
|
|
|
|
|
2009-02-28 02:23:00 +01:00
|
|
|
def test_association_proxy_transaction_method_starts_transaction_in_association_class
|
|
|
|
Tag.expects(:transaction)
|
|
|
|
Post.find(:first).tags.transaction do
|
|
|
|
# nothing
|
2008-10-27 07:47:01 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_has_many_association_through_a_belongs_to_association_where_the_association_doesnt_exist
|
|
|
|
author = authors(:mary)
|
|
|
|
post = Post.create!(:title => "TITLE", :body => "BODY")
|
|
|
|
assert_equal [], post.author_favorites
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_has_many_association_through_a_belongs_to_association
|
|
|
|
author = authors(:mary)
|
|
|
|
post = Post.create!(:author => author, :title => "TITLE", :body => "BODY")
|
|
|
|
author.author_favorites.create(:favorite_author_id => 1)
|
|
|
|
author.author_favorites.create(:favorite_author_id => 2)
|
|
|
|
author.author_favorites.create(:favorite_author_id => 3)
|
|
|
|
assert_equal post.author.author_favorites, post.author_favorites
|
|
|
|
end
|
2009-03-16 15:55:30 +01:00
|
|
|
|
|
|
|
def test_has_many_association_through_a_has_many_association_with_nonstandard_primary_keys
|
|
|
|
assert_equal 1, owners(:blackbeard).toys.count
|
|
|
|
end
|
2009-09-05 09:01:46 +02:00
|
|
|
|
|
|
|
def test_find_on_has_many_association_collection_with_include_and_conditions
|
|
|
|
post_with_no_comments = people(:michael).posts_with_no_comments.first
|
|
|
|
assert_equal post_with_no_comments, posts(:authorless)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_has_many_through_has_one_reflection
|
|
|
|
assert_equal [comments(:eager_sti_on_associations_vs_comment)], authors(:david).very_special_comments
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_modifying_has_many_through_has_one_reflection_should_raise
|
|
|
|
[
|
|
|
|
lambda { authors(:david).very_special_comments = [VerySpecialComment.create!(:body => "Gorp!", :post_id => 1011), VerySpecialComment.create!(:body => "Eep!", :post_id => 1012)] },
|
|
|
|
lambda { authors(:david).very_special_comments << VerySpecialComment.create!(:body => "Hoohah!", :post_id => 1013) },
|
|
|
|
lambda { authors(:david).very_special_comments.delete(authors(:david).very_special_comments.first) },
|
|
|
|
].each {|block| assert_raise(ActiveRecord::HasManyThroughCantAssociateThroughHasOneOrManyReflection, &block) }
|
|
|
|
end
|
2010-10-15 17:47:59 +02:00
|
|
|
|
|
|
|
def test_include_method_in_association_through_should_return_true_for_instance_added_with_build
|
|
|
|
person = Person.new
|
|
|
|
reference = person.references.build
|
|
|
|
job = reference.build_job
|
|
|
|
assert person.jobs.include?(job)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_include_method_in_association_through_should_return_true_for_instance_added_with_nested_builds
|
|
|
|
author = Author.new
|
|
|
|
post = author.posts.build
|
|
|
|
comment = post.comments.build
|
|
|
|
assert author.comments.include?(comment)
|
|
|
|
end
|
2008-05-18 06:22:34 +02:00
|
|
|
end
|