1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activerecord/test/inheritance_test.rb

135 lines
4.1 KiB
Ruby
Executable file

require 'abstract_unit'
require 'fixtures/company'
require 'fixtures/project'
class InheritanceTest < Test::Unit::TestCase
fixtures :companies, :projects
def test_a_bad_type_column
Company.connection.insert "INSERT INTO companies (id, type, name) VALUES(100, 'bad_class!', 'Not happening')"
assert_raises(ActiveRecord::SubclassNotFound) { Company.find(100) }
end
def test_inheritance_find
assert Company.find(1).kind_of?(Firm), "37signals should be a firm"
assert Firm.find(1).kind_of?(Firm), "37signals should be a firm"
assert Company.find(2).kind_of?(Client), "Summit should be a client"
assert Client.find(2).kind_of?(Client), "Summit should be a client"
end
def test_alt_inheritance_find
switch_to_alt_inheritance_column
test_inheritance_find
end
def test_inheritance_find_all
companies = Company.find_all(nil, "id")
assert companies[0].kind_of?(Firm), "37signals should be a firm"
assert companies[1].kind_of?(Client), "Summit should be a client"
end
def test_alt_inheritance_find_all
switch_to_alt_inheritance_column
test_inheritance_find_all
end
def test_inheritance_save
firm = Firm.new
firm.name = "Next Angle"
firm.save
next_angle = Company.find(firm.id)
assert next_angle.kind_of?(Firm), "Next Angle should be a firm"
end
def test_alt_inheritance_save
switch_to_alt_inheritance_column
test_inheritance_save
end
def test_inheritance_condition
assert_equal 5, Company.find_all.length
assert_equal 2, Firm.find_all.length
assert_equal 3, Client.find_all.length
end
def test_alt_inheritance_condition
switch_to_alt_inheritance_column
test_inheritance_condition
end
def test_finding_incorrect_type_data
assert_raises(ActiveRecord::RecordNotFound) { Firm.find(2) }
assert_nothing_raised { Firm.find(1) }
end
def test_alt_finding_incorrect_type_data
switch_to_alt_inheritance_column
test_finding_incorrect_type_data
end
def test_update_all_within_inheritance
Client.update_all "name = 'I am a client'"
assert_equal "I am a client", Client.find_all.first.name
assert_equal "37signals", Firm.find_all.first.name
end
def test_alt_update_all_within_inheritance
switch_to_alt_inheritance_column
test_update_all_within_inheritance
end
def test_destroy_all_within_inheritance
Client.destroy_all
assert_equal 0, Client.find_all.length
assert_equal 2, Firm.find_all.length
end
def test_alt_destroy_all_within_inheritance
switch_to_alt_inheritance_column
test_destroy_all_within_inheritance
end
def test_find_first_within_inheritance
assert_kind_of Firm, Company.find_first("name = '37signals'")
assert_kind_of Firm, Firm.find_first("name = '37signals'")
assert_nil Client.find_first("name = '37signals'")
end
def test_alt_find_first_within_inheritance
switch_to_alt_inheritance_column
test_find_first_within_inheritance
end
def test_complex_inheritance
very_special_client = VerySpecialClient.create("name" => "veryspecial")
assert_equal very_special_client, VerySpecialClient.find_first("name = 'veryspecial'")
assert_equal very_special_client, SpecialClient.find_first("name = 'veryspecial'")
assert_equal very_special_client, Company.find_first("name = 'veryspecial'")
assert_equal very_special_client, Client.find_first("name = 'veryspecial'")
assert_equal 1, Client.find_all("name = 'Summit'").size
assert_equal very_special_client, Client.find(very_special_client.id)
end
def test_alt_complex_inheritance
switch_to_alt_inheritance_column
test_complex_inheritance
end
def test_inheritance_without_mapping
assert_kind_of SpecialProject, SpecialProject.find(1)
assert_nothing_raised { SpecialProject.create("name" => "And breaaaaathe!") }
end
private
def switch_to_alt_inheritance_column
# we don't want misleading test results, so get rid of the values in the type column
Company.find_all(nil, "id").each do |c|
c['type'] = nil
c.save
end
def Company.inheritance_column() "ruby_type" end
end
end