Add method_missing to Raindrops::Middleware::Proxy
authorBen Somers <somers.ben@gmail.com>
Fri, 18 May 2012 01:46:58 +0000 (17 18:46 -0700)
committerEric Wong <normalperson@yhbt.net>
Fri, 18 May 2012 04:17:37 +0000 (18 04:17 +0000)
This enables it to behave more like a Rack
BodyProxy would, delegating methods to its body
object when it does not implement them itself.
(Also includes  a minor grammar fix to a comment.)

[ew: minor comment/whitespace fix]

Signed-off-by: Eric Wong <normalperson@yhbt.net>
lib/raindrops/middleware/proxy.rb
test/test_middleware.rb

index 53e14b5..51be3eb 100644 (file)
@@ -1,7 +1,7 @@
 # -*- encoding: binary -*-
 # :stopdoc:
-# This class is by Raindrops::Middleware to proxy application response
-# bodies.  There should be no need to use it directly.
+# This class is used by Raindrops::Middleware to proxy application
+# response bodies.  There should be no need to use it directly.
 class Raindrops::Middleware::Proxy
   def initialize(body, stats)
     @body, @stats = body, stats
@@ -31,4 +31,11 @@ class Raindrops::Middleware::Proxy
     m = m.to_sym
     :close == m || @body.respond_to?(m)
   end
+
+  # Rack::BodyProxy objects use +method_missing+ to delegate methods
+  # to their bodies
+  def method_missing(*args, &block)
+    @body.send(*args, &block)
+  end
+
 end
index eedf04a..56ce346 100644 (file)
@@ -120,5 +120,9 @@ class TestMiddleware < Test::Unit::TestCase
     def orig_body.to_path; "/dev/null"; end
     assert body.respond_to?(:to_path)
     assert_equal "/dev/null", body.to_path
+
+    def orig_body.body; "this is a body"; end
+    assert body.respond_to?(:body)
+    assert_equal "this is a body", body.body
   end
 end