Fix threshold when content type is not set before rendering.

Content-Type is implicitly set in response.prepare!, but that is called after setting the body to avoid setting a content length.
oggy · Apr 28, 2010 · 3281177 · 3281177
1 parent 59920f1
commit 3281177
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,7 @@
+== LATEST
+
+ * Fix threshold when content type is not set before rendering.
+
 == 0.0.6 2010-04-28
 
  * Fix #render_to_string.

diff --git a/lib/template_streaming.rb b/lib/template_streaming.rb
@@ -120,7 +120,8 @@ def with_template_streaming_condition(*args)
     # anything will be rendered.
     #
     def progressive_rendering_threshold
-      response.header['Content-type'] =~ %r'\Atext/html' or
+      content_type = response.header['Content-type']
+      content_type.nil? || content_type =~ %r'\Atext/html' or
         return 0
 
       case request.env['HTTP_USER_AGENT']