From c7e39691899b080a451a4c94a3632cada5ab0750 Mon Sep 17 00:00:00 2001
From: Jeremy Kemper <jeremy@bitsweat.net>
Date: Mon, 26 Nov 2007 22:46:11 +0000
Subject: Foxy fixtures: support single-table inheritance. Closes #10234.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@8219 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 activerecord/test/fixtures/db_definitions/schema.rb | 2 ++
 activerecord/test/fixtures/parrot.rb                | 8 ++++++++
 activerecord/test/fixtures/parrots.yml              | 6 ++++++
 activerecord/test/fixtures_test.rb                  | 7 ++++++-
 4 files changed, 22 insertions(+), 1 deletion(-)

(limited to 'activerecord/test')

diff --git a/activerecord/test/fixtures/db_definitions/schema.rb b/activerecord/test/fixtures/db_definitions/schema.rb
index f6bd0efa5d..a943b84f42 100644
--- a/activerecord/test/fixtures/db_definitions/schema.rb
+++ b/activerecord/test/fixtures/db_definitions/schema.rb
@@ -307,6 +307,8 @@ ActiveRecord::Schema.define do
 
   create_table :parrots, :force => true do |t|
     t.column :name, :string
+    t.column :parrot_sti_class, :string
+    t.column :killer_id, :integer
     t.column :created_at, :datetime
     t.column :created_on, :datetime
     t.column :updated_at, :datetime
diff --git a/activerecord/test/fixtures/parrot.rb b/activerecord/test/fixtures/parrot.rb
index 10850669e7..65191c1aa5 100644
--- a/activerecord/test/fixtures/parrot.rb
+++ b/activerecord/test/fixtures/parrot.rb
@@ -1,5 +1,13 @@
 class Parrot < ActiveRecord::Base
+  set_inheritance_column :parrot_sti_class
   has_and_belongs_to_many :pirates
   has_and_belongs_to_many :treasures
   has_many :loots, :as => :looter
 end
+
+class LiveParrot < Parrot
+end
+
+class DeadParrot < Parrot
+  belongs_to :killer, :class_name => 'Pirate'
+end
diff --git a/activerecord/test/fixtures/parrots.yml b/activerecord/test/fixtures/parrots.yml
index dd2c9548e7..8b73b8cdf6 100644
--- a/activerecord/test/fixtures/parrots.yml
+++ b/activerecord/test/fixtures/parrots.yml
@@ -1,21 +1,27 @@
 george:
   name: "Curious George"
   treasures: diamond, sapphire
+  parrot_sti_class: LiveParrot
 
 louis:
   name: "King Louis"
   treasures: [diamond, sapphire]
+  parrot_sti_class: LiveParrot
 
 frederick:
   name: $LABEL
+  parrot_sti_class: LiveParrot
 
 polly:
   id: 4
   name: $LABEL
+  killer: blackbeard
   treasures: sapphire, ruby
+  parrot_sti_class: DeadParrot
 
 DEFAULTS: &DEFAULTS
   treasures: sapphire, ruby
+  parrot_sti_class: LiveParrot
 
 davey:
   <<: *DEFAULTS
diff --git a/activerecord/test/fixtures_test.rb b/activerecord/test/fixtures_test.rb
index d51a4029bc..c1e79b807f 100755
--- a/activerecord/test/fixtures_test.rb
+++ b/activerecord/test/fixtures_test.rb
@@ -562,12 +562,17 @@ class FoxyFixturesTest < Test::Unit::TestCase
     assert_equal(pirates(:redbeard), treasures(:sapphire).looter)
     assert_equal(parrots(:louis), treasures(:ruby).looter)
   end
-  
+
   def test_only_generates_a_pk_if_necessary
     m = Matey.find(:first)
     m.pirate = pirates(:blackbeard)
     m.target = pirates(:redbeard)
   end
+
+  def test_supports_sti
+    assert_kind_of DeadParrot, parrots(:polly)
+    assert_equal pirates(:blackbeard), parrots(:polly).killer
+  end
 end
 
 class ActiveSupportSubclassWithFixturesTest < ActiveSupport::TestCase
-- 
cgit v1.2.3