phorsfall-tidy_rack 0.1.0 → 0.2.0

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
-:minor: 1
+:minor: 2
 :patch: 0
 :major: 0

data/lib/tidy_rack.rb

@@ -7,7 +7,7 @@ class TidyRack
   
   def call(env)
     status, headers, response = @app.call(env)
-    if headers['Content-Type'].include?('text/html')
+    if /text\/html/o.match(headers['Content-Type'])
       response = tidy(response)
       headers['Content-Length'] = response.size.to_s
     end

data/test/tidy_rack_test.rb

@@ -12,17 +12,24 @@ class TidyRackTest < Test::Unit::TestCase
     eof
   end
   
-  should "include the HTML Tidy generated meta tag" do
+  should "tidy html responses" do
     app = lambda { |env| [200, { 'Content-Type' => 'text/html' }, html] }
     status, headers, response = TidyRack.new(app).call({})
     doc = Hpricot(response)
     assert doc.at('meta[@content*="HTML Tidy"]')
   end
   
-  should "only tidy html responses" do
+  should "not tidy plain text responses" do
     app = lambda { |env| [200, { 'Content-Type' => 'text/plain' }, html] }
     status, headers, response = TidyRack.new(app).call({})
     doc = Hpricot(response)
     assert_nil doc.at('meta[@content*="HTML Tidy"]')
   end
+  
+  should "not die when content-type is nil" do
+    app = lambda { |env| [200, {}, html] }
+    status, headers, response = TidyRack.new(app).call({})
+    doc = Hpricot(response)
+    assert_nil doc.at('meta[@content*="HTML Tidy"]')
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: phorsfall-tidy_rack
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Paul Horsfall