Force response_body method return String.

sunlightlabs · Dec 17, 2013 · 3c75919 · 3c75919
1 parent 1f45844
commit 3c75919
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/lib/rack/honeypot.rb b/lib/rack/honeypot.rb
@@ -36,15 +36,9 @@ def call(env)
     private
 
     def response_body(response)
-      if response.respond_to?(:body)
-        response.body
-      elsif response.is_a?(String)
-        response
-      elsif response.respond_to?(:each)
-        body = ""
-        response.each { |part| body << part }
-        body
-      end
+      body = response.respond_to?(:body) ? response.body : response
+      body = body.inject("") { |i, a| i << a } if body.respond_to?(:each)
+      body.to_s
     end
 
     def spambot_submission?(form_hash)