Initial

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@4 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

1 files changed, 125 insertions, 0 deletions

diff --git a/activerecord/test/inheritance_test.rb b/activerecord/test/inheritance_test.rb
new file mode 100755
index 0000000000..6f8175801d
--- /dev/null
+++ b/activerecord/test/inheritance_test.rb
@@ -0,0 +1,125 @@
+require 'abstract_unit'
+require 'fixtures/company'
+
+
+class InheritanceTest < Test::Unit::TestCase
+  def setup
+    @company_fixtures = create_fixtures "companies"
+  end
+
+  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
+
+  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 3, Company.find_all.length
+    assert_equal 1, Firm.find_all.length
+    assert_equal 2, 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 1, 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
+end
\ No newline at end of file