Allow blocks when specifying Rack middleware to use

lib/lotus/action/rack.rb

@@ -94,8 +94,8 @@ def rack_builder
         #       end
         #     end
         #   end
-        def use(middleware, *args)
-          rack_builder.use middleware, *args
+        def use(middleware, *args, &block)
+          rack_builder.use middleware, *args, &block
         end
       end
 

test/fixtures.rb

@@ -170,6 +170,18 @@ def call(params)
   end
 end
 
+class ZMiddleware
+  def initialize(app, &message)
+    @app = app
+    @message = message
+  end
+
+  def call(env)
+    code, headers, body = @app.call(env)
+    [code, headers.merge!('Z-Middleware' => @message.call), body]
+  end
+end
+
 class YMiddleware
   def initialize(app)
     @app = app
@@ -210,6 +222,17 @@ def call(params)
       self.body = 'Hello from UseAction::Show'
     end
   end
+
+  class Edit
+    include Lotus::Action
+    use ZMiddleware do
+      'OK'
+    end
+
+    def call(params)
+      self.body = 'Hello from UseAction::Edit'
+    end
+  end
 end
 
 module NoUseAction

test/integration/use_test.rb

@@ -15,6 +15,7 @@ def response
       router = Lotus::Router.new do
         get '/', to: 'use_action#index'
         get '/show', to: 'use_action#show'
+        get '/edit', to: 'use_action#edit'
       end
 
       UseActionApplication = Rack::Builder.new do
@@ -34,6 +35,14 @@ def response
       response.headers.fetch('Y-Middleware').must_equal 'OK'
       response.headers['X-Middleware'].must_be_nil
       response.body.must_equal 'Hello from UseAction::Show'
+
+      get '/edit'
+
+      response.status.must_equal 200
+      response.headers.fetch('Z-Middleware').must_equal 'OK'
+      response.headers['X-Middleware'].must_be_nil
+      response.headers['Y-Middleware'].must_be_nil
+      response.body.must_equal 'Hello from UseAction::Edit'
     end
   end