mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
5c50df27cf
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@132 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
562 lines
19 KiB
Ruby
Executable file
562 lines
19 KiB
Ruby
Executable file
require 'abstract_unit'
|
|
require 'fixtures/developer'
|
|
require 'fixtures/project'
|
|
# require File.dirname(__FILE__) + '/../dev-utils/eval_debugger'
|
|
require 'fixtures/company'
|
|
require 'fixtures/topic'
|
|
require 'fixtures/reply'
|
|
|
|
# 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 ActiveRecord::ActiveRecordError
|
|
bad_collection_keys = true
|
|
end
|
|
raise "ActiveRecord should have barked on bad collection keys" unless bad_collection_keys
|
|
|
|
|
|
class AssociationsTest < Test::Unit::TestCase
|
|
def setup
|
|
create_fixtures "accounts", "companies", "developers", "projects", "developers_projects"
|
|
@signals37 = Firm.find(1)
|
|
end
|
|
|
|
def test_force_reload
|
|
firm = Firm.new
|
|
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("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 HasOneAssociationsTest < Test::Unit::TestCase
|
|
def setup
|
|
create_fixtures "accounts", "companies", "developers", "projects", "developers_projects"
|
|
@signals37 = Firm.find(1)
|
|
end
|
|
|
|
def test_has_one
|
|
assert_equal @signals37.account, Account.find(1)
|
|
assert_equal Account.find(1).credit_limit, @signals37.account.credit_limit
|
|
assert @signals37.has_account?, "37signals should have an account"
|
|
assert Account.find(1).firm?(@signals37), "37signals account should be able to backtrack"
|
|
assert Account.find(1).has_firm?, "37signals account should be able to backtrack"
|
|
|
|
assert !Account.find(2).has_firm?, "Unknown isn't linked"
|
|
assert !Account.find(2).firm?(@signals37), "Unknown isn't linked"
|
|
end
|
|
|
|
def test_type_mismatch
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { @signals37.account = 1 }
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { @signals37.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 = @signals37.account.id
|
|
@signals37.account = nil
|
|
@signals37.save
|
|
assert_nil @signals37.account
|
|
assert_nil Account.find(old_account_id).firm_id
|
|
end
|
|
|
|
def test_build
|
|
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_create
|
|
firm = Firm.new("name" => "GlobalMegaCorp")
|
|
firm.save
|
|
assert_equal firm.create_account("credit_limit" => 1000), firm.account
|
|
end
|
|
|
|
def test_dependence
|
|
firm = Firm.find(1)
|
|
assert !firm.account.nil?
|
|
firm.destroy
|
|
assert_equal 1, Account.find_all.length
|
|
end
|
|
|
|
def test_dependence_with_missing_association
|
|
Account.destroy_all
|
|
firm = Firm.find(1)
|
|
assert !firm.has_account?
|
|
firm.destroy
|
|
end
|
|
end
|
|
|
|
|
|
class HasManyAssociationsTest < Test::Unit::TestCase
|
|
def setup
|
|
create_fixtures "accounts", "companies", "developers", "projects", "developers_projects", "topics"
|
|
@signals37 = Firm.find(1)
|
|
end
|
|
|
|
def force_signal37_to_load_all_clients_of_firm
|
|
@signals37.clients_of_firm.each {|f| }
|
|
end
|
|
|
|
def test_finding
|
|
assert_equal 2, Firm.find_first.clients.length
|
|
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_find_all
|
|
assert_equal 2, Firm.find_first.clients.find_all("type = 'Client'").length
|
|
assert_equal 1, Firm.find_first.clients.find_all("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"])
|
|
end
|
|
|
|
def test_find_in_collection
|
|
assert_equal Client.find(2).name, @signals37.clients.find(2).name
|
|
assert_equal Client.find(2).name, @signals37.clients.find {|c| c.name == @signals37.clients.find(2).name }.name
|
|
assert_raises(ActiveRecord::RecordNotFound) { @signals37.clients.find(6) }
|
|
end
|
|
|
|
def test_adding
|
|
force_signal37_to_load_all_clients_of_firm
|
|
natural = Client.new("name" => "Natural Company")
|
|
@signals37.clients_of_firm << natural
|
|
assert_equal 2, @signals37.clients_of_firm.size # checking via the collection
|
|
assert_equal 2, @signals37.clients_of_firm(true).size # checking using the db
|
|
assert_equal natural, @signals37.clients_of_firm.last
|
|
end
|
|
|
|
def test_adding_a_mismatch_class
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { @signals37.clients_of_firm << nil }
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { @signals37.clients_of_firm << 1 }
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { @signals37.clients_of_firm << Topic.find(1) }
|
|
end
|
|
|
|
def test_adding_a_collection
|
|
force_signal37_to_load_all_clients_of_firm
|
|
@signals37.clients_of_firm.concat([Client.new("name" => "Natural Company"), Client.new("name" => "Apple")])
|
|
assert_equal 3, @signals37.clients_of_firm.size
|
|
assert_equal 3, @signals37.clients_of_firm(true).size
|
|
end
|
|
|
|
def test_build
|
|
new_client = @signals37.clients_of_firm.build("name" => "Another Client")
|
|
assert_equal "Another Client", new_client.name
|
|
assert new_client.save
|
|
assert_equal 2, @signals37.clients_of_firm(true).size
|
|
end
|
|
|
|
def test_create
|
|
force_signal37_to_load_all_clients_of_firm
|
|
new_client = @signals37.clients_of_firm.create("name" => "Another Client")
|
|
assert_equal new_client, @signals37.clients_of_firm.last
|
|
assert_equal new_client, @signals37.clients_of_firm(true).last
|
|
end
|
|
|
|
def test_deleting
|
|
force_signal37_to_load_all_clients_of_firm
|
|
@signals37.clients_of_firm.delete(@signals37.clients_of_firm.first)
|
|
assert_equal 0, @signals37.clients_of_firm.size
|
|
assert_equal 0, @signals37.clients_of_firm(true).size
|
|
end
|
|
|
|
def test_deleting_a_collection
|
|
force_signal37_to_load_all_clients_of_firm
|
|
@signals37.clients_of_firm.create("name" => "Another Client")
|
|
assert_equal 2, @signals37.clients_of_firm.size
|
|
#@signals37.clients_of_firm.clear
|
|
@signals37.clients_of_firm.delete([@signals37.clients_of_firm[0], @signals37.clients_of_firm[1]])
|
|
assert_equal 0, @signals37.clients_of_firm.size
|
|
assert_equal 0, @signals37.clients_of_firm(true).size
|
|
end
|
|
|
|
def test_deleting_a_association_collection
|
|
force_signal37_to_load_all_clients_of_firm
|
|
@signals37.clients_of_firm.create("name" => "Another Client")
|
|
assert_equal 2, @signals37.clients_of_firm.size
|
|
@signals37.clients_of_firm.clear
|
|
assert_equal 0, @signals37.clients_of_firm.size
|
|
assert_equal 0, @signals37.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'")
|
|
@signals37.clients_of_firm.delete(summit)
|
|
assert_equal 1, @signals37.clients_of_firm.size
|
|
assert_equal 1, @signals37.clients_of_firm(true).size
|
|
assert_equal 2, summit.client_of
|
|
end
|
|
|
|
def test_deleting_type_mismatch
|
|
david = Developer.find(1)
|
|
david.projects.id
|
|
assert_raises(ActiveRecord::AssociationTypeMismatch) { david.projects.delete(1) }
|
|
end
|
|
|
|
def test_deleting_self_type_mismatch
|
|
david = Developer.find(1)
|
|
david.projects.id
|
|
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 !@signals37.clients_of_firm.empty?, "37signals has clients after load"
|
|
@signals37.clients_of_firm.destroy_all
|
|
assert @signals37.clients_of_firm.empty?, "37signals has no clients after destroy all"
|
|
assert @signals37.clients_of_firm(true).empty?, "37signals has no clients after destroy all and refresh"
|
|
end
|
|
|
|
def test_dependence
|
|
assert_equal 2, Client.find_all.length
|
|
Firm.find_first.destroy
|
|
assert_equal 0, Client.find_all.length
|
|
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.silly_replies.create "title" => "neat and simple", "content" => "ain't complaining"
|
|
|
|
assert_nothing_raised { topic.destroy }
|
|
end
|
|
|
|
def test_dependence_with_transaction_support_on_failure
|
|
assert_equal 2, Client.find_all.length
|
|
firm = Firm.find_first
|
|
clients = firm.clients
|
|
clients.last.instance_eval { def before_destroy() raise "Trigger rollback" end }
|
|
|
|
firm.destroy rescue "do nothing"
|
|
|
|
assert_equal 2, Client.find_all.length
|
|
end
|
|
|
|
def test_dependence_on_account
|
|
assert_equal 2, Account.find_all.length
|
|
@signals37.destroy
|
|
assert_equal 1, Account.find_all.length
|
|
end
|
|
|
|
def test_included_in_collection
|
|
assert @signals37.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
|
|
end
|
|
|
|
class BelongsToAssociationsTest < Test::Unit::TestCase
|
|
def setup
|
|
create_fixtures "accounts", "companies", "developers", "projects", "developers_projects", "topics"
|
|
@signals37 = Firm.find(1)
|
|
end
|
|
|
|
def test_belongs_to
|
|
Client.find(3).firm.name
|
|
assert_equal @signals37.name, Client.find(3).firm.name
|
|
assert !Client.find(3).firm.nil?, "Microsoft should have a 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_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 xtest_counter_cache
|
|
apple = Firm.create("name" => "Apple")
|
|
final_cut = apple.clients.create("name" => "Final Cut")
|
|
|
|
apple.clients.to_s
|
|
assert_equal 1, apple.clients.size, "Created one client"
|
|
|
|
apple.companies_count = 2
|
|
apple.save
|
|
|
|
apple = Firm.find_first("name = 'Apple'")
|
|
assert_equal 2, apple.clients.size, "Should use the new cached number"
|
|
|
|
apple.clients.to_s
|
|
assert_equal 1, apple.clients.size, "Should not use the cached number, but go to the database"
|
|
end
|
|
end
|
|
|
|
|
|
class HasAndBelongsToManyAssociationsTest < Test::Unit::TestCase
|
|
def setup
|
|
@accounts, @companies, @developers, @projects, @developers_projects =
|
|
create_fixtures "accounts", "companies", "developers", "projects", "developers_projects"
|
|
|
|
@signals37 = Firm.find(1)
|
|
end
|
|
|
|
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 2, active_record.developers.size
|
|
assert_equal david.name, active_record.developers.first.name
|
|
end
|
|
|
|
def test_adding_single
|
|
jamis = Developer.find(2)
|
|
jamis.projects.id # 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.id
|
|
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_multiple
|
|
aridridel = Developer.new("name" => "Aridridel")
|
|
aridridel.save
|
|
aridridel.projects.id
|
|
aridridel.projects.push(Project.find(1), Project.find(2))
|
|
assert_equal 2, aridridel.projects.size
|
|
assert_equal 2, aridridel.projects(true).size
|
|
end
|
|
|
|
def test_adding_a_collection
|
|
aridridel = Developer.new("name" => "Aridridel")
|
|
aridridel.save
|
|
aridridel.projects.id
|
|
aridridel.projects.concat([Project.find(1), Project.find(2)])
|
|
assert_equal 2, aridridel.projects.size
|
|
assert_equal 2, aridridel.projects(true).size
|
|
end
|
|
|
|
def test_uniq_after_the_fact
|
|
@developers["jamis"].find.projects << @projects["active_record"].find
|
|
@developers["jamis"].find.projects << @projects["active_record"].find
|
|
assert_equal 3, @developers["jamis"].find.projects.size
|
|
assert_equal 1, @developers["jamis"].find.projects.uniq.size
|
|
end
|
|
|
|
def test_uniq_before_the_fact
|
|
@projects["active_record"].find.developers << @developers["jamis"].find
|
|
@projects["active_record"].find.developers << @developers["david"].find
|
|
assert_equal 2, @projects["active_record"].find.developers.size
|
|
end
|
|
|
|
def test_deleting
|
|
david = Developer.find(1)
|
|
active_record = Project.find(1)
|
|
david.projects.id
|
|
assert_equal 2, david.projects.size
|
|
assert_equal 2, active_record.developers.size
|
|
|
|
david.projects.delete(active_record)
|
|
|
|
assert_equal 1, david.projects.size
|
|
assert_equal 1, david.projects(true).size
|
|
assert_equal 1, active_record.developers(true).size
|
|
end
|
|
|
|
def test_deleting_array
|
|
david = Developer.find(1)
|
|
david.projects.id
|
|
david.projects.delete(Project.find_all)
|
|
assert_equal 0, david.projects.size
|
|
assert_equal 0, david.projects(true).size
|
|
end
|
|
|
|
def test_deleting_all
|
|
david = Developer.find(1)
|
|
david.projects.id
|
|
david.projects.clear
|
|
assert_equal 0, david.projects.size
|
|
assert_equal 0, david.projects(true).size
|
|
end
|
|
|
|
def test_removing_associations_on_destroy
|
|
Developer.find(1).destroy
|
|
assert Developer.connection.select_all("SELECT * FROM developers_projects WHERE developer_id = '1'").empty?
|
|
end
|
|
|
|
def test_additional_columns_from_join_table
|
|
assert_equal Date.new(2004, 10, 10).to_s, Developer.find(1).projects.first.joined_on.to_s
|
|
end
|
|
|
|
def test_destroy_all
|
|
david = Developer.find(1)
|
|
david.projects.id
|
|
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"].find
|
|
@jamis.projects.push_with_attributes(@projects["action_controller"].find, :joined_on => Date.today)
|
|
assert_equal Date.today.to_s, @jamis.projects.select { |p| p.name == @projects["action_controller"]["name"] }.first.joined_on.to_s
|
|
assert_equal Date.today.to_s, @developers["jamis"].find.projects.select { |p| p.name == @projects["action_controller"]["name"] }.first.joined_on.to_s
|
|
end
|
|
|
|
def test_associations_with_conditions
|
|
assert_equal 2, @projects["active_record"].find.developers.size
|
|
assert_equal 1, @projects["active_record"].find.developers_named_david.size
|
|
|
|
@projects["active_record"].find.developers_named_david.clear
|
|
assert_equal 1, @projects["active_record"].find.developers.size
|
|
end
|
|
|
|
def test_find_in_association
|
|
# Using sql
|
|
assert_equal @developers["david"].find, @projects["active_record"].find.developers.find(@developers["david"]["id"]), "SQL find"
|
|
|
|
# Using ruby
|
|
@active_record = @projects["active_record"].find
|
|
@active_record.developers.reload
|
|
assert_equal @developers["david"].find, @active_record.developers.find(@developers["david"]["id"]), "Ruby find"
|
|
end
|
|
end
|