spec: update to take care of :partial state

2 files changed, 4 insertions, 2 deletions

diff --git a/spec/node_spec.rb b/spec/node_spec.rb
index 8b2d40a..b667338 100644
--- a/spec/node_spec.rb
+++ b/spec/node_spec.rb
@@ -119,11 +119,11 @@ describe Zorglub do
         end
         #
         it "partial should render correctly" do
-            Node0.partial(:do_partial, 1, 2).should == "view_content"
+            Node0.partial(:do_partial, 1, 2).should == 'partial_content'
         end
         #
         it "method level view should work" do
-            Node0.partial(:other_view).should == "view_content"
+            Node0.partial(:other_view).should == 'partial_content'
         end
         #
         it "redirect should work" do
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 3a7e5fb..35b8e2a 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -19,6 +19,8 @@ RENDER_PROC = Proc.new { |path,obj|
         "layout_start #{obj.content} layout_end"
     when :view
         "view_content"
+    when :partial
+        'partial_content'
     else
         raise Exception.new
     end