require 'abstract_unit' require 'fixtures/developer' require 'fixtures/project' require 'fixtures/company' require 'fixtures/topic' require 'fixtures/reply' require 'fixtures/computer' require 'fixtures/customer' require 'fixtures/order' require 'fixtures/category' require 'fixtures/post' require 'fixtures/author' # Can't declare new classes in test case methods, so tests before that bad_collection_keys = false begin class Car < ActiveRecord::Base; has_many :wheels, :name => "wheels"; end rescue ArgumentError bad_collection_keys = true end raise "ActiveRecord should have barked on bad collection keys" unless bad_collection_keys class AssociationsTest < Test::Unit::TestCase fixtures :accounts, :companies, :developers, :projects, :developers_projects, :computers def test_force_reload firm = Firm.new("name" => "A New Firm, Inc") firm.save firm.clients.each {|c|} # forcing to load all clients assert firm.clients.empty?, "New firm shouldn't have client objects" assert !firm.has_clients?, "New firm shouldn't have clients" assert_equal 0, firm.clients.size, "New firm should have 0 clients" client = Client.new("name" => "TheClient.com", "firm_id" => firm.id) client.save assert firm.clients.empty?, "New firm should have cached no client objects" assert !firm.has_clients?, "New firm should have cached a no-clients response" assert_equal 0, firm.clients.size, "New firm should have cached 0 clients count" assert !firm.clients(true).empty?, "New firm should have reloaded client objects" assert_equal 1, firm.clients(true).size, "New firm should have reloaded clients count" end def test_storing_in_pstore require "tmpdir" store_filename = File.join(Dir.tmpdir, "ar-pstore-association-test") File.delete(store_filename) if File.exists?(store_filename) require "pstore" apple = Firm.create("name" => "Apple") natural = Client.new("name" => "Natural Company") apple.clients << natural db = PStore.new(store_filename) db.transaction do db["apple"] = apple end db = PStore.new(store_filename) db.transaction do assert_equal "Natural Company", db["apple"].clients.first.name end end end class AssociationProxyTest < Test::Unit::TestCase fixtures :authors, :posts def test_proxy_accessors welcome = posts(:welcome) assert_equal welcome, welcome.author.proxy_owner assert_equal welcome.class.reflect_on_association(:author), welcome.author.proxy_reflection welcome.author.class # force load target assert_equal welcome.author, welcome.author.proxy_target david = authors(:david) assert_equal david, david.posts.proxy_owner assert_equal david.class.reflect_on_association(:posts), david.posts.proxy_reflection david.posts.first # force load target assert_equal david.posts, david.posts.proxy_target assert_equal david, david.posts_with_extension.testing_proxy_owner assert_equal david.class.reflect_on_association(:posts_with_extension), david.posts_with_extension.testing_proxy_reflection david.posts_with_extension.first # force load target assert_equal david.posts_with_extension, david.posts_with_extension.testing_proxy_target end end class HasOneAssociationsTest < Test::Unit::TestCase fixtures :accounts, :companies, :developers, :projects, :developers_projects def test_has_one assert_equal companies(:first_firm).account, Account.find(1) assert_equal Account.find(1).credit_limit, companies(:first_firm).account.credit_limit end def test_proxy_assignment company = companies(:first_firm) assert_nothing_raised { company.account = company.account } end def test_triple_equality assert Account === companies(:first_firm).account assert companies(:first_firm).account === Account end def test_type_mismatch assert_raises(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).account = 1 } assert_raises(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).account = Project.find(1) } end def test_natural_assignment apple = Firm.create("name" => "Apple") citibank = Account.create("credit_limit" => 10) apple.account = citibank assert_equal apple.id, citibank.firm_id end def test_natural_assignment_to_nil old_account_id = companies(:first_firm).account.id companies(:first_firm).account = nil companies(:first_firm).save assert_nil companies(:first_firm).account # account is dependent, therefore is destroyed when reference to owner is lost assert_raises(ActiveRecord::RecordNotFound) { Account.find(old_account_id) } end def test_assignment_without_replacement apple = Firm.create("name" => "Apple") citibank = Account.create("credit_limit" => 10) apple.account = citibank assert_equal apple.id, citibank.firm_id hsbc = apple.build_account({ :credit_limit => 20}, false) assert_equal apple.id, hsbc.firm_id hsbc.save assert_equal apple.id, citibank.firm_id nykredit = apple.create_account({ :credit_limit => 30}, false) assert_equal apple.id, nykredit.firm_id assert_equal apple.id, citibank.firm_id assert_equal apple.id, hsbc.firm_id end def test_assignment_without_replacement_on_create apple = Firm.create("name" => "Apple") citibank = Account.create("credit_limit" => 10) apple.account = citibank assert_equal apple.id, citibank.firm_id hsbc = apple.create_account({:credit_limit => 10}, false) assert_equal apple.id, hsbc.firm_id hsbc.save assert_equal apple.id, citibank.firm_id end def test_dependence num_accounts = Account.count firm = Firm.find(1) assert !firm.account.nil? firm.destroy assert_equal num_accounts - 1, Account.count end def test_succesful_build_association firm = Firm.new("name" => "GlobalMegaCorp") firm.save account = firm.build_account("credit_limit" => 1000) assert account.save assert_equal account, firm.account end def test_failing_build_association firm = Firm.new("name" => "GlobalMegaCorp") firm.save account = firm.build_account assert !account.save assert_equal "can't be empty", account.errors.on("credit_limit") end def test_build_association_twice_without_saving_affects_nothing count_of_account = Account.count firm = Firm.find(:first) account1 = firm.build_account("credit_limit" => 1000) account2 = firm.build_account("credit_limit" => 2000) assert_equal count_of_account, Account.count end def test_create_association firm = Firm.new("name" => "GlobalMegaCorp") firm.save assert_equal firm.create_account("credit_limit" => 1000), firm.account end def test_build firm = Firm.new("name" => "GlobalMegaCorp") firm.save firm.account = account = Account.new("credit_limit" => 1000) assert_equal account, firm.account assert account.save assert_equal account, firm.account end def test_build_before_child_saved firm = Firm.find(1) account = firm.account.build("credit_limit" => 1000) assert_equal account, firm.account assert account.new_record? assert firm.save assert_equal account, firm.account assert !account.new_record? end def test_build_before_either_saved firm = Firm.new("name" => "GlobalMegaCorp") firm.account = account = Account.new("credit_limit" => 1000) assert_equal account, firm.account assert account.new_record? assert firm.save assert_equal account, firm.account assert !account.new_record? end def test_failing_build_association firm = Firm.new("name" => "GlobalMegaCorp") firm.save firm.account = account = Account.new assert_equal account, firm.account assert !account.save assert_equal account, firm.account assert_equal "can't be empty", account.errors.on("credit_limit") end def test_create firm = Firm.new("name" => "GlobalMegaCorp") firm.save firm.account = account = Account.create("credit_limit" => 1000) assert_equal account, firm.account end def test_create_before_save firm = Firm.new("name" => "GlobalMegaCorp") firm.account = account = Account.create("credit_limit" => 1000) assert_equal account, firm.account end def test_dependence_with_missing_association Account.destroy_all firm = Firm.find(1) assert firm.account.nil? firm.destroy end def test_assignment_before_parent_saved firm = Firm.new("name" => "GlobalMegaCorp") firm.account = a = Account.find(1) assert firm.new_record? assert_equal a, firm.account assert firm.save assert_equal a, firm.account assert_equal a, firm.account(true) end def test_assignment_before_child_saved firm = Firm.find(1) firm.account = a = Account.new("credit_limit" => 1000) assert !a.new_record? assert_equal a, firm.account assert_equal a, firm.account assert_equal a, firm.account(true) end def test_assignment_before_either_saved firm = Firm.new("name" => "GlobalMegaCorp") firm.account = a = Account.new("credit_limit" => 1000) assert firm.new_record? assert a.new_record? assert_equal a, firm.account assert firm.save assert !firm.new_record? assert !a.new_record? assert_equal a, firm.account assert_equal a, firm.account(true) end end class HasManyAssociationsTest < Test::Unit::TestCase fixtures :accounts, :companies, :developers, :projects, :developers_projects, :topics def setup Client.destroyed_client_ids.clear end def force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.each {|f| } end def test_counting_with_counter_sql assert_equal 2, Firm.find(:first).clients.count end def test_counting assert_equal 2, Firm.find(:first).plain_clients.count end def test_counting_with_single_conditions assert_equal 2, Firm.find(:first).plain_clients.count('1=1') end def test_counting_with_single_hash assert_equal 2, Firm.find(:first).plain_clients.count(:conditions => '1=1') end def test_counting_with_column_name_and_hash assert_equal 2, Firm.find(:first).plain_clients.count(:all, :conditions => '1=1') end def test_finding assert_equal 2, Firm.find(:first).clients.length end def test_find_many_with_merged_options assert_equal 1, companies(:first_firm).limited_clients.size assert_equal 1, companies(:first_firm).limited_clients.find(:all).size assert_equal 2, companies(:first_firm).limited_clients.find(:all, :limit => nil).size end def test_triple_equality assert !(Array === Firm.find(:first).clients) assert Firm.find(:first).clients === Array end def test_finding_default_orders assert_equal "Summit", Firm.find(:first).clients.first.name end def test_finding_with_different_class_name_and_order assert_equal "Microsoft", Firm.find(:first).clients_sorted_desc.first.name end def test_finding_with_foreign_key assert_equal "Microsoft", Firm.find(:first).clients_of_firm.first.name end def test_finding_with_condition assert_equal "Microsoft", Firm.find(:first).clients_like_ms.first.name end def test_finding_using_sql firm = Firm.find(:first) first_client = firm.clients_using_sql.first assert_not_nil first_client assert_equal "Microsoft", first_client.name assert_equal 1, firm.clients_using_sql.size assert_equal 1, Firm.find(:first).clients_using_sql.size end def test_counting_using_sql assert_equal 1, Firm.find(:first).clients_using_counter_sql.size assert_equal 0, Firm.find(:first).clients_using_zero_counter_sql.size end def test_counting_non_existant_items_using_sql assert_equal 0, Firm.find(:first).no_clients_using_counter_sql.size end def test_belongs_to_sanity c = Client.new assert_nil c.firm if c.firm assert false, "belongs_to failed if check" end unless c.firm else assert false, "belongs_to failed unless check" end end def test_find_ids firm = Firm.find(:first) assert_raises(ActiveRecord::RecordNotFound) { firm.clients.find } client = firm.clients.find(2) assert_kind_of Client, client client_ary = firm.clients.find([2]) assert_kind_of Array, client_ary assert_equal client, client_ary.first client_ary = firm.clients.find(2, 3) assert_kind_of Array, client_ary assert_equal 2, client_ary.size assert_equal client, client_ary.first assert_raises(ActiveRecord::RecordNotFound) { firm.clients.find(2, 99) } end def test_find_all firm = Firm.find_first assert_equal firm.clients, firm.clients.find_all assert_equal 2, firm.clients.find(:all, :conditions => "#{QUOTED_TYPE} = 'Client'").length assert_equal 1, firm.clients.find(:all, :conditions => "name = 'Summit'").length end def test_find_all_sanitized firm = Firm.find_first assert_equal firm.clients.find_all("name = 'Summit'"), firm.clients.find_all(["name = '%s'", "Summit"]) summit = firm.clients.find(:all, :conditions => "name = 'Summit'") assert_equal summit, firm.clients.find(:all, :conditions => ["name = ?", "Summit"]) assert_equal summit, firm.clients.find(:all, :conditions => ["name = :name", { :name => "Summit" }]) end def test_find_first firm = Firm.find_first client2 = Client.find(2) assert_equal firm.clients.first, firm.clients.find_first assert_equal client2, firm.clients.find_first("#{QUOTED_TYPE} = 'Client'") assert_equal client2, firm.clients.find(:first, :conditions => "#{QUOTED_TYPE} = 'Client'") end def test_find_first_sanitized firm = Firm.find_first client2 = Client.find(2) assert_equal client2, firm.clients.find_first(["#{QUOTED_TYPE} = ?", "Client"]) assert_equal client2, firm.clients.find(:first, :conditions => ["#{QUOTED_TYPE} = ?", 'Client']) assert_equal client2, firm.clients.find(:first, :conditions => ["#{QUOTED_TYPE} = :type", { :type => 'Client' }]) end def test_find_in_collection assert_equal Client.find(2).name, companies(:first_firm).clients.find(2).name assert_raises(ActiveRecord::RecordNotFound) { companies(:first_firm).clients.find(6) } end def test_find_grouped all_clients_of_firm1 = Client.find(:all, :conditions => "firm_id = 1") grouped_clients_of_firm1 = Client.find(:all, :conditions => "firm_id = 1", :group => "firm_id", :select => 'firm_id, count(id) as clients_count') assert_equal 2, all_clients_of_firm1.size assert_equal 1, grouped_clients_of_firm1.size end def test_adding force_signal37_to_load_all_clients_of_firm natural = Client.new("name" => "Natural Company") companies(:first_firm).clients_of_firm << natural assert_equal 2, companies(:first_firm).clients_of_firm.size # checking via the collection assert_equal 2, companies(:first_firm).clients_of_firm(true).size # checking using the db assert_equal natural, companies(:first_firm).clients_of_firm.last end def test_adding_using_create first_firm = companies(:first_firm) assert_equal 2, first_firm.plain_clients.size natural = first_firm.plain_clients.create(:name => "Natural Company") assert_equal 3, first_firm.plain_clients.length assert_equal 3, first_firm.plain_clients.size end def test_adding_a_mismatch_class assert_raises(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << nil } assert_raises(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << 1 } assert_raises(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << Topic.find(1) } end def test_adding_a_collection force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.concat([Client.new("name" => "Natural Company"), Client.new("name" => "Apple")]) assert_equal 3, companies(:first_firm).clients_of_firm.size assert_equal 3, companies(:first_firm).clients_of_firm(true).size end def test_adding_before_save no_of_firms = Firm.count no_of_clients = Client.count new_firm = Firm.new("name" => "A New Firm, Inc") new_firm.clients_of_firm.push Client.new("name" => "Natural Company") new_firm.clients_of_firm << (c = Client.new("name" => "Apple")) assert new_firm.new_record? assert c.new_record? assert_equal 2, new_firm.clients_of_firm.size assert_equal no_of_firms, Firm.count # Firm was not saved to database. assert_equal no_of_clients, Client.count # Clients were not saved to database. assert new_firm.save assert !new_firm.new_record? assert !c.new_record? assert_equal new_firm, c.firm assert_equal no_of_firms+1, Firm.count # Firm was saved to database. assert_equal no_of_clients+2, Client.count # Clients were saved to database. assert_equal 2, new_firm.clients_of_firm.size assert_equal 2, new_firm.clients_of_firm(true).size end def test_invalid_adding firm = Firm.find(1) assert !(firm.clients_of_firm << c = Client.new) assert c.new_record? assert !firm.valid? assert !firm.save assert c.new_record? end def test_invalid_adding_before_save no_of_firms = Firm.count no_of_clients = Client.count new_firm = Firm.new("name" => "A New Firm, Inc") new_firm.clients_of_firm.concat([c = Client.new, Client.new("name" => "Apple")]) assert c.new_record? assert !c.valid? assert !new_firm.valid? assert !new_firm.save assert c.new_record? assert new_firm.new_record? end def test_build new_client = companies(:first_firm).clients_of_firm.build("name" => "Another Client") assert_equal "Another Client", new_client.name assert new_client.new_record? assert_equal new_client, companies(:first_firm).clients_of_firm.last assert companies(:first_firm).save assert !new_client.new_record? assert_equal 2, companies(:first_firm).clients_of_firm(true).size end def test_build_many new_clients = companies(:first_firm).clients_of_firm.build([{"name" => "Another Client"}, {"name" => "Another Client II"}]) assert_equal 2, new_clients.size assert companies(:first_firm).save assert_equal 3, companies(:first_firm).clients_of_firm(true).size end def test_build_without_loading_association first_topic = topics(:first) Reply.column_names assert_equal 1, first_topic.replies.length assert_no_queries do first_topic.replies.build(:title => "Not saved", :content => "Superstars") assert_equal 2, first_topic.replies.size end assert_equal 2, first_topic.replies.to_ary.size end def test_create_without_loading_association first_firm = companies(:first_firm) Firm.column_names Client.column_names assert_equal 1, first_firm.clients_of_firm.size first_firm.clients_of_firm.reset assert_queries(1) do first_firm.clients_of_firm.create(:name => "Superstars") end assert_equal 2, first_firm.clients_of_firm.size end def test_invalid_build new_client = companies(:first_firm).clients_of_firm.build assert new_client.new_record? assert !new_client.valid? assert_equal new_client, companies(:first_firm).clients_of_firm.last assert !companies(:first_firm).save assert new_client.new_record? assert_equal 1, companies(:first_firm).clients_of_firm(true).size end def test_create force_signal37_to_load_all_clients_of_firm new_client = companies(:first_firm).clients_of_firm.create("name" => "Another Client") assert !new_client.new_record? assert_equal new_client, companies(:first_firm).clients_of_firm.last assert_equal new_client, companies(:first_firm).clients_of_firm(true).last end def test_create_many companies(:first_firm).clients_of_firm.create([{"name" => "Another Client"}, {"name" => "Another Client II"}]) assert_equal 3, companies(:first_firm).clients_of_firm(true).size end def test_find_or_create number_of_clients = companies(:first_firm).clients.size the_client = companies(:first_firm).clients.find_or_create_by_name("Yet another client") assert_equal number_of_clients + 1, companies(:first_firm, :refresh).clients.size assert_equal the_client, companies(:first_firm).clients.find_or_create_by_name("Yet another client") assert_equal number_of_clients + 1, companies(:first_firm, :refresh).clients.size end def test_deleting force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.delete(companies(:first_firm).clients_of_firm.first) assert_equal 0, companies(:first_firm).clients_of_firm.size assert_equal 0, companies(:first_firm).clients_of_firm(true).size end def test_deleting_before_save new_firm = Firm.new("name" => "A New Firm, Inc.") new_client = new_firm.clients_of_firm.build("name" => "Another Client") assert_equal 1, new_firm.clients_of_firm.size new_firm.clients_of_firm.delete(new_client) assert_equal 0, new_firm.clients_of_firm.size end def test_deleting_a_collection force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.create("name" => "Another Client") assert_equal 2, companies(:first_firm).clients_of_firm.size companies(:first_firm).clients_of_firm.delete([companies(:first_firm).clients_of_firm[0], companies(:first_firm).clients_of_firm[1]]) assert_equal 0, companies(:first_firm).clients_of_firm.size assert_equal 0, companies(:first_firm).clients_of_firm(true).size end def test_delete_all force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.create("name" => "Another Client") assert_equal 2, companies(:first_firm).clients_of_firm.size companies(:first_firm).clients_of_firm.delete_all assert_equal 0, companies(:first_firm).clients_of_firm.size assert_equal 0, companies(:first_firm).clients_of_firm(true).size end def test_delete_all_with_not_yet_loaded_association_collection force_signal37_to_load_all_clients_of_firm companies(:first_firm).clients_of_firm.create("name" => "Another Client") assert_equal 2, companies(:first_firm).clients_of_firm.size companies(:first_firm).clients_of_firm.reset companies(:first_firm).clients_of_firm.delete_all assert_equal 0, companies(:first_firm).clients_of_firm.size assert_equal 0, companies(:first_firm).clients_of_firm(true).size end def test_clearing_an_association_collection firm = companies(:first_firm) client_id = firm.clients_of_firm.first.id assert_equal 1, firm.clients_of_firm.size firm.clients_of_firm.clear assert_equal 0, firm.clients_of_firm.size assert_equal 0, firm.clients_of_firm(true).size assert_equal [], Client.destroyed_client_ids[firm.id] # Should not be destroyed since the association is not dependent. assert_nothing_raised do assert Client.find(client_id).firm.nil? end end def test_clearing_a_dependent_association_collection firm = companies(:first_firm) client_id = firm.dependent_clients_of_firm.first.id assert_equal 1, firm.dependent_clients_of_firm.size # :dependent means destroy is called on each client firm.dependent_clients_of_firm.clear assert_equal 0, firm.dependent_clients_of_firm.size assert_equal 0, firm.dependent_clients_of_firm(true).size assert_equal [client_id], Client.destroyed_client_ids[firm.id] # Should be destroyed since the association is dependent. assert Client.find_by_id(client_id).nil? end def test_clearing_an_exclusively_dependent_association_collection firm = companies(:first_firm) client_id = firm.exclusively_dependent_clients_of_firm.first.id assert_equal 1, firm.exclusively_dependent_clients_of_firm.size assert_equal [], Client.destroyed_client_ids[firm.id] # :exclusively_dependent means each client is deleted directly from # the database without looping through them calling destroy. firm.exclusively_dependent_clients_of_firm.clear assert_equal 0, firm.exclusively_dependent_clients_of_firm.size assert_equal 0, firm.exclusively_dependent_clients_of_firm(true).size assert_equal [3], Client.destroyed_client_ids[firm.id] # Should be destroyed since the association is exclusively dependent. assert Client.find_by_id(client_id).nil? end def test_clearing_without_initial_access firm = companies(:first_firm) firm.clients_of_firm.clear assert_equal 0, firm.clients_of_firm.size assert_equal 0, firm.clients_of_firm(true).size end def test_deleting_a_item_which_is_not_in_the_collection force_signal37_to_load_all_clients_of_firm summit = Client.find_first("name = 'Summit'") companies(:first_firm).clients_of_firm.delete(summit) assert_equal 1, companies(:first_firm).clients_of_firm.size assert_equal 1, companies(:first_firm).clients_of_firm(true).size assert_equal 2, summit.client_of end def test_deleting_type_mismatch david = Developer.find(1) david.projects.reload assert_raises(ActiveRecord::AssociationTypeMismatch) { david.projects.delete(1) } end def test_deleting_self_type_mismatch david = Developer.find(1) david.projects.reload assert_raises(ActiveRecord::AssociationTypeMismatch) { david.projects.delete(Project.find(1).developers) } end def test_destroy_all force_signal37_to_load_all_clients_of_firm assert !companies(:first_firm).clients_of_firm.empty?, "37signals has clients after load" companies(:first_firm).clients_of_firm.destroy_all assert companies(:first_firm).clients_of_firm.empty?, "37signals has no clients after destroy all" assert companies(:first_firm).clients_of_firm(true).empty?, "37signals has no clients after destroy all and refresh" end def test_dependence firm = companies(:first_firm) assert_equal 2, firm.clients.size firm.destroy assert Client.find(:all, :conditions => "firm_id=#{firm.id}").empty? end def test_destroy_dependent_when_deleted_from_association firm = Firm.find(:first) assert_equal 2, firm.clients.size client = firm.clients.first firm.clients.delete(client) assert_raise(ActiveRecord::RecordNotFound) { Client.find(client.id) } assert_raise(ActiveRecord::RecordNotFound) { firm.clients.find(client.id) } assert_equal 1, firm.clients.size end def test_three_levels_of_dependence topic = Topic.create "title" => "neat and simple" reply = topic.replies.create "title" => "neat and simple", "content" => "still digging it" silly_reply = reply.replies.create "title" => "neat and simple", "content" => "ain't complaining" assert_nothing_raised { topic.destroy } end uses_transaction :test_dependence_with_transaction_support_on_failure def test_dependence_with_transaction_support_on_failure firm = companies(:first_firm) clients = firm.clients assert_equal 2, clients.length clients.last.instance_eval { def before_destroy() raise "Trigger rollback" end } firm.destroy rescue "do nothing" assert_equal 2, Client.find(:all, :conditions => "firm_id=#{firm.id}").size end def test_dependence_on_account num_accounts = Account.count companies(:first_firm).destroy assert_equal num_accounts - 1, Account.count end def test_depends_and_nullify num_accounts = Account.count num_companies = Company.count core = companies(:rails_core) assert_equal accounts(:rails_core_account), core.account assert_equal [companies(:leetsoft), companies(:jadedpixel)], core.companies core.destroy assert_nil accounts(:rails_core_account).reload.firm_id assert_nil companies(:leetsoft).reload.client_of assert_nil companies(:jadedpixel).reload.client_of assert_equal num_accounts, Account.count end def test_included_in_collection assert companies(:first_firm).clients.include?(Client.find(2)) end def test_adding_array_and_collection assert_nothing_raised { Firm.find(:first).clients + Firm.find(:all).last.clients } end def test_find_all_without_conditions firm = companies(:first_firm) assert_equal 2, firm.clients.find(:all).length end def test_replace_with_less firm = Firm.find(:first) firm.clients = [companies(:first_client)] assert firm.save, "Could not save firm" firm.reload assert_equal 1, firm.clients.length end def test_replace_with_new firm = Firm.find(:first) new_client = Client.new("name" => "New Client") firm.clients = [companies(:second_client),new_client] firm.save firm.reload assert_equal 2, firm.clients.length assert !firm.clients.include?(:first_client) end def test_replace_on_new_object firm = Firm.new("name" => "New Firm") firm.clients = [companies(:second_client), Client.new("name" => "New Client")] assert firm.save firm.reload assert_equal 2, firm.clients.length assert firm.clients.include?(Client.find_by_name("New Client")) end def test_assign_ids firm = Firm.new("name" => "Apple") firm.client_ids = [companies(:first_client).id, companies(:second_client).id] firm.save firm.reload assert_equal 2, firm.clients.length assert firm.clients.include?(companies(:second_client)) end end class BelongsToAssociationsTest < Test::Unit::TestCase fixtures :accounts, :companies, :developers, :projects, :topics, :developers_projects, :computers, :authors, :posts def test_belongs_to Client.find(3).firm.name assert_equal companies(:first_firm).name, Client.find(3).firm.name assert !Client.find(3).firm.nil?, "Microsoft should have a firm" end def test_proxy_assignment account = Account.find(1) assert_nothing_raised { account.firm = account.firm } end def test_triple_equality assert Client.find(3).firm === Firm assert Firm === Client.find(3).firm end def test_type_mismatch assert_raise(ActiveRecord::AssociationTypeMismatch) { Account.find(1).firm = 1 } assert_raise(ActiveRecord::AssociationTypeMismatch) { Account.find(1).firm = Project.find(1) } end def test_natural_assignment apple = Firm.create("name" => "Apple") citibank = Account.create("credit_limit" => 10) citibank.firm = apple assert_equal apple.id, citibank.firm_id end def test_creating_the_belonging_object citibank = Account.create("credit_limit" => 10) apple = citibank.create_firm("name" => "Apple") assert_equal apple, citibank.firm citibank.save citibank.reload assert_equal apple, citibank.firm end def test_building_the_belonging_object citibank = Account.create("credit_limit" => 10) apple = citibank.build_firm("name" => "Apple") citibank.save assert_equal apple.id, citibank.firm_id end def test_natural_assignment_to_nil client = Client.find(3) client.firm = nil client.save assert_nil client.firm(true) assert_nil client.client_of end def test_with_different_class_name assert_equal Company.find(1).name, Company.find(3).firm_with_other_name.name assert_not_nil Company.find(3).firm_with_other_name, "Microsoft should have a firm" end def test_with_condition assert_equal Company.find(1).name, Company.find(3).firm_with_condition.name assert_not_nil Company.find(3).firm_with_condition, "Microsoft should have a firm" end def test_belongs_to_counter debate = Topic.create("title" => "debate") assert_equal 0, debate.send(:read_attribute, "replies_count"), "No replies yet" trash = debate.replies.create("title" => "blah!", "content" => "world around!") assert_equal 1, Topic.find(debate.id).send(:read_attribute, "replies_count"), "First reply created" trash.destroy assert_equal 0, Topic.find(debate.id).send(:read_attribute, "replies_count"), "First reply deleted" end def test_belongs_to_counter_with_reassigning t1 = Topic.create("title" => "t1") t2 = Topic.create("title" => "t2") r1 = Reply.new("title" => "r1", "content" => "r1") r1.topic = t1 assert r1.save assert_equal 1, Topic.find(t1.id).replies.size assert_equal 0, Topic.find(t2.id).replies.size r1.topic = Topic.find(t2.id) assert r1.save assert_equal 0, Topic.find(t1.id).replies.size assert_equal 1, Topic.find(t2.id).replies.size r1.topic = nil assert_equal 0, Topic.find(t1.id).replies.size assert_equal 0, Topic.find(t2.id).replies.size r1.topic = t1 assert_equal 1, Topic.find(t1.id).replies.size assert_equal 0, Topic.find(t2.id).replies.size r1.destroy assert_equal 0, Topic.find(t1.id).replies.size assert_equal 0, Topic.find(t2.id).replies.size end def test_assignment_before_parent_saved client = Client.find(:first) apple = Firm.new("name" => "Apple") client.firm = apple assert_equal apple, client.firm assert apple.new_record? assert client.save assert apple.save assert !apple.new_record? assert_equal apple, client.firm assert_equal apple, client.firm(true) end def test_assignment_before_child_saved final_cut = Client.new("name" => "Final Cut") firm = Firm.find(1) final_cut.firm = firm assert final_cut.new_record? assert final_cut.save assert !final_cut.new_record? assert !firm.new_record? assert_equal firm, final_cut.firm assert_equal firm, final_cut.firm(true) end def test_assignment_before_either_saved final_cut = Client.new("name" => "Final Cut") apple = Firm.new("name" => "Apple") final_cut.firm = apple assert final_cut.new_record? assert apple.new_record? assert final_cut.save assert !final_cut.new_record? assert !apple.new_record? assert_equal apple, final_cut.firm assert_equal apple, final_cut.firm(true) end def test_new_record_with_foreign_key_but_no_object c = Client.new("firm_id" => 1) assert_equal Firm.find(:first), c.firm_with_basic_id end def test_forgetting_the_load_when_foreign_key_enters_late c = Client.new assert_nil c.firm_with_basic_id c.firm_id = 1 assert_equal Firm.find(:first), c.firm_with_basic_id end def test_field_name_same_as_foreign_key computer = Computer.find(1) assert_not_nil computer.developer, ":foreign key == attribute didn't lock up" # ' end def test_counter_cache topic = Topic.create :title => "Zoom-zoom-zoom" assert_equal 0, topic[:replies_count] reply = Reply.create(:title => "re: zoom", :content => "speedy quick!") reply.topic = topic assert_equal 1, topic.reload[:replies_count] assert_equal 1, topic.replies.size topic[:replies_count] = 15 assert_equal 15, topic.replies.size end def test_custom_counter_cache reply = Reply.create(:title => "re: zoom", :content => "speedy quick!") assert_equal 0, reply[:replies_count] silly = SillyReply.create(:title => "gaga", :content => "boo-boo") silly.reply = reply assert_equal 1, reply.reload[:replies_count] assert_equal 1, reply.replies.size reply[:replies_count] = 17 assert_equal 17, reply.replies.size end def test_store_two_association_with_one_save num_orders = Order.count num_customers = Customer.count order = Order.new customer1 = order.billing = Customer.new customer2 = order.shipping = Customer.new assert order.save assert_equal customer1, order.billing assert_equal customer2, order.shipping order.reload assert_equal customer1, order.billing assert_equal customer2, order.shipping assert_equal num_orders +1, Order.count assert_equal num_customers +2, Customer.count end def test_store_association_in_two_relations_with_one_save num_orders = Order.count num_customers = Customer.count order = Order.new customer = order.billing = order.shipping = Customer.new assert order.save assert_equal customer, order.billing assert_equal customer, order.shipping order.reload assert_equal customer, order.billing assert_equal customer, order.shipping assert_equal num_orders +1, Order.count assert_equal num_customers +1, Customer.count end def test_store_association_in_two_relations_with_one_save_in_existing_object num_orders = Order.count num_customers = Customer.count order = Order.create customer = order.billing = order.shipping = Customer.new assert order.save assert_equal customer, order.billing assert_equal customer, order.shipping order.reload assert_equal customer, order.billing assert_equal customer, order.shipping assert_equal num_orders +1, Order.count assert_equal num_customers +1, Customer.count end def test_store_association_in_two_relations_with_one_save_in_existing_object_with_values num_orders = Order.count num_customers = Customer.count order = Order.create customer = order.billing = order.shipping = Customer.new assert order.save assert_equal customer, order.billing assert_equal customer, order.shipping order.reload customer = order.billing = order.shipping = Customer.new assert order.save order.reload assert_equal customer, order.billing assert_equal customer, order.shipping assert_equal num_orders +1, Order.count assert_equal num_customers +2, Customer.count end def test_association_assignment_sticks post = Post.find(:first) author1, author2 = Author.find(:all, :limit => 2) assert_not_nil author1 assert_not_nil author2 # make sure the association is loaded post.author # set the association by id, directly post.author_id = author2.id # save and reload post.save! post.reload # the author id of the post should be the id we set assert_equal post.author_id, author2.id end end class ProjectWithAfterCreateHook < ActiveRecord::Base set_table_name 'projects' has_and_belongs_to_many :developers, :class_name => "DeveloperForProjectWithAfterCreateHook", :join_table => "developers_projects", :foreign_key => "project_id", :association_foreign_key => "developer_id" after_create :add_david def add_david david = DeveloperForProjectWithAfterCreateHook.find_by_name('David') david.projects << self end end class DeveloperForProjectWithAfterCreateHook < ActiveRecord::Base set_table_name 'developers' has_and_belongs_to_many :projects, :class_name => "ProjectWithAfterCreateHook", :join_table => "developers_projects", :association_foreign_key => "project_id", :foreign_key => "developer_id" end class HasAndBelongsToManyAssociationsTest < Test::Unit::TestCase fixtures :accounts, :companies, :categories, :posts, :categories_posts, :developers, :projects, :developers_projects def test_has_and_belongs_to_many david = Developer.find(1) assert !david.projects.empty? assert_equal 2, david.projects.size active_record = Project.find(1) assert !active_record.developers.empty? assert_equal 3, active_record.developers.size assert active_record.developers.include?(david) end def test_triple_equality assert !(Array === Developer.find(1).projects) assert Developer.find(1).projects === Array end def test_adding_single jamis = Developer.find(2) jamis.projects.reload # causing the collection to load action_controller = Project.find(2) assert_equal 1, jamis.projects.size assert_equal 1, action_controller.developers.size jamis.projects << action_controller assert_equal 2, jamis.projects.size assert_equal 2, jamis.projects(true).size assert_equal 2, action_controller.developers(true).size end def test_adding_type_mismatch jamis = Developer.find(2) assert_raise(ActiveRecord::AssociationTypeMismatch) { jamis.projects << nil } assert_raise(ActiveRecord::AssociationTypeMismatch) { jamis.projects << 1 } end def test_adding_from_the_project jamis = Developer.find(2) action_controller = Project.find(2) action_controller.developers.reload assert_equal 1, jamis.projects.size assert_equal 1, action_controller.developers.size action_controller.developers << jamis assert_equal 2, jamis.projects(true).size assert_equal 2, action_controller.developers.size assert_equal 2, action_controller.developers(true).size end def test_adding_from_the_project_fixed_timestamp jamis = Developer.find(2) action_controller = Project.find(2) action_controller.developers.reload assert_equal 1, jamis.projects.size assert_equal 1, action_controller.developers.size updated_at = jamis.updated_at action_controller.developers << jamis assert_equal updated_at, jamis.updated_at assert_equal 2, jamis.projects(true).size assert_equal 2, action_controller.developers.size assert_equal 2, action_controller.developers(true).size end def test_adding_multiple aredridel = Developer.new("name" => "Aredridel") aredridel.save aredridel.projects.reload aredridel.projects.push(Project.find(1), Project.find(2)) assert_equal 2, aredridel.projects.size assert_equal 2, aredridel.projects(true).size end def test_adding_a_collection aredridel = Developer.new("name" => "Aredridel") aredridel.save aredridel.projects.reload aredridel.projects.concat([Project.find(1), Project.find(2)]) assert_equal 2, aredridel.projects.size assert_equal 2, aredridel.projects(true).size end def test_adding_uses_default_values_on_join_table ac = projects(:action_controller) assert !developers(:jamis).projects.include?(ac) developers(:jamis).projects << ac assert developers(:jamis, :reload).projects.include?(ac) project = developers(:jamis).projects.detect { |p| p == ac } assert_equal 1, project.access_level.to_i end def test_adding_uses_explicit_values_on_join_table ac = projects(:action_controller) assert !developers(:jamis).projects.include?(ac) developers(:jamis).projects.push_with_attributes(ac, :access_level => 3) assert developers(:jamis, :reload).projects.include?(ac) project = developers(:jamis).projects.detect { |p| p == ac } assert_equal 3, project.access_level.to_i end def test_hatbm_attribute_access_and_respond_to project = developers(:jamis).projects[0] assert project.has_attribute?("name") assert project.has_attribute?("joined_on") assert project.has_attribute?("access_level") assert project.respond_to?("name") assert project.respond_to?("name=") assert project.respond_to?("name?") assert project.respond_to?("joined_on") assert project.respond_to?("joined_on=") assert project.respond_to?("joined_on?") assert project.respond_to?("access_level") assert project.respond_to?("access_level=") assert project.respond_to?("access_level?") end def test_habtm_adding_before_save no_of_devels = Developer.count no_of_projects = Project.count aredridel = Developer.new("name" => "Aredridel") aredridel.projects.concat([Project.find(1), p = Project.new("name" => "Projekt")]) assert aredridel.new_record? assert p.new_record? assert aredridel.save assert !aredridel.new_record? assert_equal no_of_devels+1, Developer.count assert_equal no_of_projects+1, Project.count assert_equal 2, aredridel.projects.size assert_equal 2, aredridel.projects(true).size end def test_habtm_adding_before_save_with_join_attributes no_of_devels = Developer.count no_of_projects = Project.count now = Date.today ken = Developer.new("name" => "Ken") ken.projects.push_with_attributes( Project.find(1), :joined_on => now ) p = Project.new("name" => "Foomatic") ken.projects.push_with_attributes( p, :joined_on => now ) assert ken.new_record? assert p.new_record? assert ken.save assert !ken.new_record? assert_equal no_of_devels+1, Developer.count assert_equal no_of_projects+1, Project.count assert_equal 2, ken.projects.size assert_equal 2, ken.projects(true).size kenReloaded = Developer.find_by_name 'Ken' kenReloaded.projects.each {|prj| assert_date_from_db(now, prj.joined_on)} end def test_habtm_saving_multiple_relationships new_project = Project.new("name" => "Grimetime") amount_of_developers = 4 developers = (0...amount_of_developers).collect {|i| Developer.create(:name => "JME #{i}") }.reverse new_project.developer_ids = [developers[0].id, developers[1].id] new_project.developers_with_callback_ids = [developers[2].id, developers[3].id] assert new_project.save new_project.reload assert_equal amount_of_developers, new_project.developers.size assert_equal developers, new_project.developers end def test_habtm_unique_order_preserved assert_equal [developers(:poor_jamis), developers(:jamis), developers(:david)], projects(:active_record).non_unique_developers assert_equal [developers(:poor_jamis), developers(:jamis), developers(:david)], projects(:active_record).developers end def test_build devel = Developer.find(1) proj = devel.projects.build("name" => "Projekt") assert_equal devel.projects.last, proj assert proj.new_record? devel.save assert !proj.new_record? assert_equal devel.projects.last, proj assert_equal Developer.find(1).projects.sort_by(&:id).last, proj # prove join table is updated end def test_build_by_new_record devel = Developer.new(:name => "Marcel", :salary => 75000) proj1 = devel.projects.build(:name => "Make bed") proj2 = devel.projects.build(:name => "Lie in it") assert_equal devel.projects.last, proj2 assert proj2.new_record? devel.save assert !devel.new_record? assert !proj2.new_record? assert_equal devel.projects.last, proj2 assert_equal Developer.find_by_name("Marcel").projects.last, proj2 # prove join table is updated end def test_create devel = Developer.find(1) proj = devel.projects.create("name" => "Projekt") assert_equal devel.projects.last, proj assert !proj.new_record? assert_equal Developer.find(1).projects.sort_by(&:id).last, proj # prove join table is updated end def test_create_by_new_record devel = Developer.new(:name => "Marcel", :salary => 75000) proj1 = devel.projects.create(:name => "Make bed") proj2 = devel.projects.create(:name => "Lie in it") assert_equal devel.projects.last, proj2 assert proj2.new_record? devel.save assert !devel.new_record? assert !proj2.new_record? assert_equal devel.projects.last, proj2 assert_equal Developer.find_by_name("Marcel").projects.last, proj2 # prove join table is updated end def test_uniq_after_the_fact developers(:jamis).projects << projects(:active_record) developers(:jamis).projects << projects(:active_record) assert_equal 3, developers(:jamis).projects.size assert_equal 1, developers(:jamis).projects.uniq.size end def test_uniq_before_the_fact projects(:active_record).developers << developers(:jamis) projects(:active_record).developers << developers(:david) assert_equal 3, projects(:active_record, :reload).developers.size end def test_deleting david = Developer.find(1) active_record = Project.find(1) david.projects.reload assert_equal 2, david.projects.size assert_equal 3, active_record.developers.size david.projects.delete(active_record) assert_equal 1, david.projects.size assert_equal 1, david.projects(true).size assert_equal 2, active_record.developers(true).size end def test_deleting_array david = Developer.find(1) david.projects.reload david.projects.delete(Project.find(:all)) assert_equal 0, david.projects.size assert_equal 0, david.projects(true).size end def test_deleting_with_sql david = Developer.find(1) active_record = Project.find(1) active_record.developers.reload assert_equal 3, active_record.developers_by_sql.size active_record.developers_by_sql.delete(david) assert_equal 2, active_record.developers_by_sql(true).size end def test_deleting_array_with_sql active_record = Project.find(1) active_record.developers.reload assert_equal 3, active_record.developers_by_sql.size active_record.developers_by_sql.delete(Developer.find(:all)) assert_equal 0, active_record.developers_by_sql(true).size end def test_deleting_all david = Developer.find(1) david.projects.reload david.projects.clear assert_equal 0, david.projects.size assert_equal 0, david.projects(true).size end def test_removing_associations_on_destroy david = DeveloperWithBeforeDestroyRaise.find(1) assert !david.projects.empty? assert_nothing_raised { david.destroy } assert david.projects.empty? assert DeveloperWithBeforeDestroyRaise.connection.select_all("SELECT * FROM developers_projects WHERE developer_id = 1").empty? end def test_additional_columns_from_join_table assert_date_from_db Date.new(2004, 10, 10), Developer.find(1).projects.first.joined_on end def test_destroy_all david = Developer.find(1) david.projects.reload assert !david.projects.empty? david.projects.destroy_all assert david.projects.empty? assert david.projects(true).empty? end def test_rich_association jamis = developers(:jamis) jamis.projects.push_with_attributes(projects(:action_controller), :joined_on => Date.today) assert_date_from_db Date.today, jamis.projects.select { |p| p.name == projects(:action_controller).name }.first.joined_on assert_date_from_db Date.today, developers(:jamis).projects.select { |p| p.name == projects(:action_controller).name }.first.joined_on end def test_associations_with_conditions assert_equal 3, projects(:active_record).developers.size assert_equal 1, projects(:active_record).developers_named_david.size assert_equal developers(:david), projects(:active_record).developers_named_david.find(developers(:david).id) assert_equal developers(:david), projects(:active_record).salaried_developers.find(developers(:david).id) projects(:active_record).developers_named_david.clear assert_equal 2, projects(:active_record, :reload).developers.size end def test_find_in_association # Using sql assert_equal developers(:david), projects(:active_record).developers.find(developers(:david).id), "SQL find" # Using ruby active_record = projects(:active_record) active_record.developers.reload assert_equal developers(:david), active_record.developers.find(developers(:david).id), "Ruby find" end def test_find_in_association_with_custom_finder_sql assert_equal developers(:david), projects(:active_record).developers_with_finder_sql.find(developers(:david).id), "SQL find" active_record = projects(:active_record) active_record.developers_with_finder_sql.reload assert_equal developers(:david), active_record.developers_with_finder_sql.find(developers(:david).id), "Ruby find" end def test_find_in_association_with_custom_finder_sql_and_string_id assert_equal developers(:david), projects(:active_record).developers_with_finder_sql.find(developers(:david).id.to_s), "SQL find" end def test_find_with_merged_options assert_equal 1, projects(:active_record).limited_developers.size assert_equal 1, projects(:active_record).limited_developers.find(:all).size assert_equal 3, projects(:active_record).limited_developers.find(:all, :limit => nil).size end def test_new_with_values_in_collection jamis = DeveloperForProjectWithAfterCreateHook.find_by_name('Jamis') david = DeveloperForProjectWithAfterCreateHook.find_by_name('David') project = ProjectWithAfterCreateHook.new(:name => "Cooking with Bertie") project.developers << jamis project.save! project.reload assert project.developers.include?(jamis) assert project.developers.include?(david) end def test_find_in_association_with_options developers = projects(:active_record).developers.find(:all) assert_equal 3, developers.size assert_equal developers(:poor_jamis), projects(:active_record).developers.find(:first, :conditions => "salary < 10000") assert_equal developers(:jamis), projects(:active_record).developers.find(:first, :order => "salary DESC") end def test_replace_with_less david = developers(:david) david.projects = [projects(:action_controller)] assert david.save assert_equal 1, david.projects.length end def test_replace_with_new david = developers(:david) david.projects = [projects(:action_controller), Project.new("name" => "ActionWebSearch")] david.save assert_equal 2, david.projects.length assert !david.projects.include?(projects(:active_record)) end def test_replace_on_new_object new_developer = Developer.new("name" => "Matz") new_developer.projects = [projects(:action_controller), Project.new("name" => "ActionWebSearch")] new_developer.save assert_equal 2, new_developer.projects.length end def test_consider_type developer = Developer.find(:first) special_project = SpecialProject.create("name" => "Special Project") other_project = developer.projects.first developer.special_projects << special_project developer.reload assert developer.projects.include?(special_project) assert developer.special_projects.include?(special_project) assert !developer.special_projects.include?(other_project) end def test_update_attributes_after_push_without_duplicate_join_table_rows developer = Developer.new("name" => "Kano") project = SpecialProject.create("name" => "Special Project") assert developer.save developer.projects << project developer.update_attribute("name", "Bruza") assert_equal 1, Developer.connection.select_value(<<-end_sql).to_i SELECT count(*) FROM developers_projects WHERE project_id = #{project.id} AND developer_id = #{developer.id} end_sql end def test_updating_attributes_on_non_rich_associations welcome = categories(:technology).posts.first welcome.title = "Something else" assert welcome.save! end def test_updating_attributes_on_rich_associations david = projects(:action_controller).developers.first david.name = "DHH" assert_raises(ActiveRecord::ReadOnlyRecord) { david.save! } end def test_updating_attributes_on_rich_associations_with_limited_find david = projects(:action_controller).developers.find(:all, :select => "developers.*").first david.name = "DHH" assert david.save! end def test_join_table_alias assert_equal 3, Developer.find(:all, :include => {:projects => :developers}, :conditions => 'developers_projects_join.joined_on IS NOT NULL').size end def test_join_with_group group = Developer.columns.inject([]) do |g, c| g << "developers.#{c.name}" g << "developers_projects_2.#{c.name}" end Project.columns.each { |c| group << "projects.#{c.name}" } assert_equal 3, Developer.find(:all, :include => {:projects => :developers}, :conditions => 'developers_projects_join.joined_on IS NOT NULL', :group => group.join(",")).size end end