add single-table inheritance tests

Brian Christian
1 parent c395dd2f
Showing 4 changed files with 32 additions and 0 deletions Show diff stats
test/acts_as_followable_test.rb
test/dummy30/app/models/band/punk.rb
test/dummy30/app/models/band/punk/pop_punk.rb
test/factories/bands.rb
@@ -20,6 +20,8 @@ class ActsAsFollowableTest &lt; ActiveSupport::TestCase
       @jon = FactoryGirl.create(:jon)
       @oasis = FactoryGirl.create(:oasis)
       @metallica = FactoryGirl.create(:metallica)
+      @green_day = FactoryGirl.create(:green_day)
+      @blink_182 = FactoryGirl.create(:blink_182)
       @sam.follow(@jon)
     end
@@ -225,6 +227,20 @@ class ActsAsFollowableTest &lt; ActiveSupport::TestCase
       end
     end
+    context "followers_with_sti" do
+      setup do
+        @sam.follow(@green_day)
+        @sam.follow(@blink_182)
+      end
+
+      should "return the followers for given type" do
+        assert_equal @sam.follows_by_type('Band').first.followable, @green_day.becomes(Band)
+        assert_equal @sam.follows_by_type('Band').second.followable, @blink_182.becomes(Band)
+        assert @green_day.followers_by_type('User').include?(@sam)
+        assert @blink_182.followers_by_type('User').include?(@sam)
+      end
+    end
+
     context "method_missing" do
       setup do
         @sam.follow(@oasis)
@@ -0,0 +1,4 @@
+class Band::Punk < Band
+  validates_presence_of :name
+  acts_as_followable
+end
@@ -0,0 +1,4 @@
+class Band::Punk::PopPunk < Band::Punk
+  validates_presence_of :name
+  acts_as_followable
+end
@@ -6,4 +6,12 @@ FactoryGirl.define do
   factory :metallica, :class => Band do |b|
     b.name 'Metallica'
   end
+  
+  factory :green_day, :class => Band::Punk do |b|
+    b.name 'Green Day'
+  end
+
+  factory :blink_182, :class => Band::Punk::PopPunk do |b|
+    b.name 'Blink 182'
+  end
 end
	@@ -20,6 +20,8 @@ class ActsAsFollowableTest < ActiveSupport::TestCase		@@ -20,6 +20,8 @@ class ActsAsFollowableTest < ActiveSupport::TestCase
20	@jon = FactoryGirl.create(:jon)	20	@jon = FactoryGirl.create(:jon)
21	@oasis = FactoryGirl.create(:oasis)	21	@oasis = FactoryGirl.create(:oasis)
22	@metallica = FactoryGirl.create(:metallica)	22	@metallica = FactoryGirl.create(:metallica)
		23	+ @green_day = FactoryGirl.create(:green_day)
		24	+ @blink_182 = FactoryGirl.create(:blink_182)
23	@sam.follow(@jon)	25	@sam.follow(@jon)
24	end	26	end
25		27
	@@ -225,6 +227,20 @@ class ActsAsFollowableTest < ActiveSupport::TestCase		@@ -225,6 +227,20 @@ class ActsAsFollowableTest < ActiveSupport::TestCase
225	end	227	end
226	end	228	end
227		229
		230	+ context "followers_with_sti" do
		231	+ setup do
		232	+ @sam.follow(@green_day)
		233	+ @sam.follow(@blink_182)
		234	+ end
		235	+
		236	+ should "return the followers for given type" do
		237	+ assert_equal @sam.follows_by_type('Band').first.followable, @green_day.becomes(Band)
		238	+ assert_equal @sam.follows_by_type('Band').second.followable, @blink_182.becomes(Band)
		239	+ assert @green_day.followers_by_type('User').include?(@sam)
		240	+ assert @blink_182.followers_by_type('User').include?(@sam)
		241	+ end
		242	+ end
		243	+
228	context "method_missing" do	244	context "method_missing" do
229	setup do	245	setup do
230	@sam.follow(@oasis)	246	@sam.follow(@oasis)
@@ -0,0 +1,4 @@		@@ -0,0 +1,4 @@
	1	+class Band::Punk < Band
	2	+ validates_presence_of :name
	3	+ acts_as_followable
	4	+end
@@ -0,0 +1,4 @@		@@ -0,0 +1,4 @@
	1	+class Band::Punk::PopPunk < Band::Punk
	2	+ validates_presence_of :name
	3	+ acts_as_followable
	4	+end
	@@ -6,4 +6,12 @@ FactoryGirl.define do		@@ -6,4 +6,12 @@ FactoryGirl.define do
6	factory :metallica, :class => Band do \|b\|	6	factory :metallica, :class => Band do \|b\|
7	b.name 'Metallica'	7	b.name 'Metallica'
8	end	8	end
		9	+
		10	+ factory :green_day, :class => Band::Punk do \|b\|
		11	+ b.name 'Green Day'
		12	+ end
		13	+
		14	+ factory :blink_182, :class => Band::Punk::PopPunk do \|b\|
		15	+ b.name 'Blink 182'
		16	+ end
9	end	17	end