Fix output error

johnbintz · Dec 8, 2014 · edfbe49 · edfbe49
1 parent e6260d6
commit edfbe49
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/rack/livereload/body_processor.rb b/lib/rack/livereload/body_processor.rb
@@ -69,7 +69,7 @@ def process!(env)
         head_elements.each do |head|
           head.inner_html = "#{template.result(binding)}\n #{head.inner_html}"
         end
-        @string_body = xmlfragment.to_html
+        @string_body = xmlfragment.to_s
 
         @content_length = @string_body.bytesize()
         @livereload_added = true if head_elements.length > 0

diff --git a/spec/rack/livereload/body_processor_spec.rb b/spec/rack/livereload/body_processor_spec.rb
@@ -130,7 +130,7 @@
       let(:page_html) { "<script type='text/template'><head></head></script> <!-- <head></head> --> <xmp><head></head></xmp>" }
 
       it 'should not add the livereload js' do
-        processed_body.should == "<script type=\"text/template\"><head></head></script> <!-- <head></head> --> <xmp><head></head></xmp>"
+        processed_body.should == "<script type=\"text/template\">\n  <head/>\n</script> <!-- <head></head> --> <xmp>\n  <head/>\n</xmp>"
       end
     end