diff --git a/lib/rack/mock.rb b/lib/rack/mock.rb index 5b2512ca0..33b3a073c 100644 --- a/lib/rack/mock.rb +++ b/lib/rack/mock.rb @@ -212,8 +212,15 @@ def body # end buffer = String.new - super.each do |chunk| - buffer << chunk + body = super + body = body.body if body.respond_to?(:body) + + if body.respond_to?(:each) + body.each do |chunk| + buffer << chunk + end + else + buffer = body.to_s end return buffer