fix for ActiveSupport >= 3.0.8 SafeBuffer

juggy · Jul 2, 2011 · 2ebee3d · 2ebee3d
1 parent 5f66f1c
commit 2ebee3d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/pdfkit/pdfkit.rb b/lib/pdfkit/pdfkit.rb
@@ -103,7 +103,7 @@ def append_stylesheets
 
       stylesheets.each do |stylesheet|
         if @source.to_s.match(/<\/head>/)
-          @source.to_s.gsub!(/(<\/head>)/, style_tag_for(stylesheet)+'\1')
+          @source = Source.new(@source.to_s.gsub(/(<\/head>)/, style_tag_for(stylesheet)+'\1'))
         else
           @source.to_s.insert(0, style_tag_for(stylesheet))
         end

diff --git a/spec/pdfkit_spec.rb b/spec/pdfkit_spec.rb
@@ -184,7 +184,7 @@
       css = File.join(SPEC_ROOT,'fixtures','example.css')
       pdfkit.stylesheets << css
       pdfkit.to_pdf
-      pdfkit.source.to_s.should include("<style>#{File.read(css)}</style><html>")
+      pdfkit.source.to_s.should include("<style>#{File.read(css)}</style></head>")
     end
 
   end