0
@@ -16,6 +16,14 @@ load(File.dirname(__FILE__) + "/schema.rb")
0
 
0
 ActiveRecord::Base.send :extend, Technoweenie::ActiveRecordContext
0
 
0
-class Foo < ActiveRecord::Base
0
-  set_table_name 'foo'
0
+class Topic < ActiveRecord::Base
0
+end
0
+
0
+class Post < ActiveRecord::Base; end
0
+class NormalPost < Post
0
+  belongs_to :topic
0
+end
0
+
0
+class PolymorphPost < Post
0
+  belongs_to :topic, :polymorphic => true
0
 end
0
\ No newline at end of file

0
@@ -2,39 +2,61 @@ require File.join(File.dirname(__FILE__), 'abstract_unit')
0
 
0
 class ActiveRecordContextTest < Test::Unit::TestCase
0
   def setup
0
-    Foo.destroy_all
0
-    @records = {}
0
-    2.times { |i| f = Foo.create!(:bar => "test#{i}"); @records[f.id] = f }
0
-    assert_nil Foo.context_cache
0
+    Post.destroy_all
0
+    @posts = []
0
+    @topic = Topic.create! :title => 'test'
0
+    @posts << NormalPost.create!(:body => 'normal body', :topic => @topic)
0
+    @posts << PolymorphPost.create!(:body => 'polymorph body', :topic => @topic)
0
+    assert_nil Post.context_cache
0
   end
0
 
0
   def test_should_initialize_context_cache_hash
0
-    Foo.with_context do
0
-      assert_kind_of Hash, Foo.context_cache
0
-      assert_equal 0, Foo.context_cache.size
0
+    Post.with_context do
0
+      assert_kind_of Hash, Post.context_cache
0
+      assert_equal 0, Post.context_cache.size
0
     end
0
-    assert_nil Foo.context_cache
0
+    assert_nil Post.context_cache
0
   end
0
 
0
   def test_should_store_records_in_cache
0
-    Foo.with_context do
0
-      records = Foo.find(:all)
0
-      assert_equal records.size, Foo.context_cache[Foo].size
0
-      assert_equal @records[1], Foo.find_in_context(1)
0
-      assert_equal @records[2], Foo.find_in_context(2)
0
+    Post.with_context do
0
+      records = Post.find(:all)
0
+      assert_equal records.size, Post.context_cache[Post].size
0
+      assert_equal @posts[0], Post.find_in_context(1)
0
+      assert_equal @posts[1], Post.find_in_context(2)
0
     end
0
   end
0
   
0
   def test_should_find_records_in_context
0
-    Foo.with_context do
0
-      records = Foo.find(:all)
0
-      Foo.destroy_all
0
-      assert_equal @records[1], Foo.find(1)
0
-      assert_equal @records[2], Foo.find(2)
0
+    Post.with_context do
0
+      records = Post.find(:all)
0
+      Post.destroy_all
0
+      assert_equal @posts[0], Post.find(1)
0
+      assert_equal @posts[1], Post.find(2)
0
     end
0
     
0
     assert_raise ActiveRecord::RecordNotFound do
0
-      Foo.find 1
0
+      Post.find 1
0
     end
0
   end
0
+  
0
+  def test_should_find_belongs_to_record
0
+    Post.with_context do
0
+      Topic.find :all ; Topic.delete_all
0
+      assert_equal @topic, @posts[0].topic(true)
0
+    end
0
+    
0
+    assert_equal @topic, @posts[0].topic
0
+    assert_nil @posts[0].topic(true)
0
+  end
0
+  
0
+  def test_should_find_belongs_to_polymorphic_record
0
+    Post.with_context do
0
+      Topic.find :all ; Topic.delete_all
0
+      assert_equal @topic, @posts[1].topic(true)
0
+    end
0
+    
0
+    assert_equal @topic, @posts[1].topic
0
+    assert_nil @posts[1].topic(true)
0
+  end
0
 end

0
@@ -1,5 +1,12 @@
0
 ActiveRecord::Schema.define(:version => 0) do
0
-  create_table :foo, :force => true do |t|
0
-    t.column :bar, :string
0
+  create_table :topics, :force => true do |t|
0
+    t.column :title, :string
0
+  end
0
+
0
+  create_table :posts, :force => true do |t|
0
+    t.column :topic_id, :integer
0
+    t.column :topic_type, :string
0
+    t.column :type, :string
0
+    t.column :body, :string
0
   end
0
 end
0
\ No newline at end of file