lib/rack/body_proxy.rb in rack-1.6.13 vs lib/rack/body_proxy.rb in rack-2.0.0.alpha

@@ -1,14 +1,19 @@
 module Rack
   class BodyProxy
     def initialize(body, &block)
-      @body, @block, @closed = body, block, false
+      @body = body
+      @block = block
+      @closed = false
     end
 
-    def respond_to?(*args)
-      return false if args.first.to_s =~ /^to_ary$/
-      super or @body.respond_to?(*args)
+    def respond_to?(method_name, include_all=false)
+      case method_name
+      when :to_ary, 'to_ary'
+        return false
+      end
+      super or @body.respond_to?(method_name, include_all)
     end
 
     def close
       return if @closed
       @closed = true
@@ -25,15 +30,15 @@
 
     # N.B. This method is a special case to address the bug described by #434.
     # We are applying this special case for #each only. Future bugs of this
     # class will be handled by requesting users to patch their ruby
     # implementation, to save adding too many methods in this class.
-    def each(*args, &block)
-      @body.each(*args, &block)
+    def each
+      @body.each { |body| yield body }
     end
 
-    def method_missing(*args, &block)
-      super if args.first.to_s =~ /^to_ary$/
-      @body.__send__(*args, &block)
+    def method_missing(method_name, *args, &block)
+      super if :to_ary == method_name
+      @body.__send__(method_name, *args, &block)
     end
   end
 end