RailsWidget#capture should return a RawString

lib/erector/rails/extensions/rails_widget.rb

@@ -34,7 +34,7 @@ def output
     end
 
     def capture(&block)
-      parent ? parent.capture(&block) : super
+      parent ? raw(parent.capture(&block).to_s) : super
     end
 
     # This is here to force #parent.capture to return the output

spec/rails_root/spec/rails_widget_spec.rb

@@ -7,14 +7,32 @@
   end
 
   describe "#capture" do
+    it "returns a RawString" do
+      captured = nil
+      Erector::RailsWidget.inline do
+        captured = capture {}
+      end.to_s(:parent => @view).should == ""
+      captured.should be_a_kind_of Erector::RawString
+    end
+
+    it "captures helper output" do
+      captured = nil
+      Erector::RailsWidget.inline do
+        captured = capture do
+          helpers.concat "capture me!"
+        end
+      end.to_s(:parent => @view, :helpers => @view).should == ""
+      captured.should == "capture me!"
+    end
+
     it "captures with an erector block" do
       captured = nil
-      Erector::RailsWidget.inline(:parent => @view) do
-        captured = parent.capture do
-          h1 'capture me!'
+      Erector::RailsWidget.inline do
+        captured = capture do
+          text 'capture me!'
         end
-      end.to_s.should == ""
-      captured.should == "<h1>capture me!</h1>"
+      end.to_s(:parent => @view).should == ""
+      captured.should == "capture me!"
     end
   end
 end