Added this_context

auser · Mar 19, 2009 · 1692ccb · 1692ccb
1 parent df11571
commit 1692ccb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lib/parenting/base.rb b/lib/parenting/base.rb
@@ -13,16 +13,16 @@ def run_in_context(&block)
       @parent = parent
 
       context_stack.push self
-      instance_eval &block if block
+      this_context.instance_eval &block if block
       context_stack.pop
       head   
     end
 
     def head
       context_stack.first
     end
-    def context
-      context_stack.last
+    def this_context
+      @this_context ||= context_stack.last
     end
     def parent
       @parent ||= current_context[-1] == self ? current_context[-2] : current_context[-1]

diff --git a/test/test_parenting.rb b/test/test_parenting.rb
@@ -83,5 +83,11 @@ class QuickieTest < Test::Unit::TestCase
       $d.parent.should == @a.b.c
       $d.parent.parent.parent.should == @a
     end
+    it "should have the latest context set as the last item in the stack" do
+      @a.b.c.this_context.nil?.should == false
+      @a.b.this_context.should == @a.b
+      @a.b.c.this_context.should == @a.b.c
+      @a.b.c.d.this_context.should == @a.b.c.d
+    end
   end
 end