0
@@ -22,8 +22,12 @@ module Rack
0
       instance_eval(&block)
0
     end
0
 
0
-    def use(middleware, *args)
0
-      @ins << lambda { |app| middleware.new(app, *args) }
0
+    def use(middleware, *args, &block)
0
+      @ins << if block_given?
0
+        lambda { |app| middleware.new(app, *args, &block) }
0
+      else
0
+        lambda { |app| middleware.new(app, *args) }
0
+      end
0
     end
0
 
0
     def run(app)

0
@@ -25,4 +25,26 @@ context "Rack::Builder" do
0
     Rack::MockRequest.new(app).get("/").should.be.server_error
0
     Rack::MockRequest.new(app).get("/").should.be.server_error
0
   end
0
+
0
+  specify "supports blocks on use" do
0
+    app = Rack::Builder.new do
0
+      use Rack::ShowExceptions
0
+      use Rack::Auth::Basic do |username, password|
0
+        'secret' == password
0
+      end
0
+
0
+      run lambda { |env| [200, {}, 'Hi Boss'] }
0
+    end
0
+
0
+    response = Rack::MockRequest.new(app).get("/")
0
+    response.should.be.client_error
0
+    response.status.should.equal 401
0
+
0
+    # with auth...
0
+    response = Rack::MockRequest.new(app).get("/", 
0
+        'HTTP_AUTHORIZATION' => 'Basic ' + ["joe:secret"].pack("m*"))
0
+    response.status.should.equal 200
0
+    response.body.to_s.should.equal 'Hi Boss'
0
+  end
0
+
0
 end