Delegate lock proxy to_path

2 files changed, 34 insertions, 0 deletions

diff --git a/lib/rack/lock.rb b/lib/rack/lock.rb
index bfc86dd8..b5326a4f 100644
--- a/lib/rack/lock.rb
+++ b/lib/rack/lock.rb
@@ -12,6 +12,14 @@ module Rack
       ensure
         mutex.unlock
       end
+
+      def to_path
+        target.to_path
+      end
+
+      def respond_to?(sym)
+        sym.to_sym == :close || target.respond_to?(sym)
+      end
     end
 
     FLAG = 'rack.multithread'.freeze
diff --git a/test/spec_lock.rb b/test/spec_lock.rb
index 8386c5f2..64849096 100644
--- a/test/spec_lock.rb
+++ b/test/spec_lock.rb
@@ -39,6 +39,32 @@ describe Rack::Lock do
       response.each { |x| list << x }
       list.should.equal %w{ hi mom }
     end
+
+    should 'delegate to_path' do
+      lock = Lock.new
+      env  = Rack::MockRequest.env_for("/")
+
+      res = ['Hello World']
+      def res.to_path ; "/tmp/hello.txt" ; end
+
+      app = Rack::Lock.new(lambda { |inner_env| [200, {}, res] }, lock)
+      body = app.call(env)[2]
+
+      body.should.respond_to :to_path
+      body.to_path.should.equal "/tmp/hello.txt"
+    end
+
+    should 'not delegate to_path if body does not implement it' do
+      lock = Lock.new
+      env  = Rack::MockRequest.env_for("/")
+
+      res = ['Hello World']
+
+      app = Rack::Lock.new(lambda { |inner_env| [200, {}, res] }, lock)
+      body = app.call(env)[2]
+
+      body.should.not.respond_to :to_path
+    end
   end
 
   should 'call super on close' do