test/test_middleware.rb in prop-0.7.0 vs test/test_middleware.rb in prop-0.7.1

@@ -30,8 +30,18 @@
         response = @middleware.call(@env)
 
         assert_equal 429, response[0]
         assert_equal ["Boom!"], response[2]
       end
+
+      context "with a custom error handler" do
+        setup do
+          @middleware = Prop::Middleware.new(@app, :error_handler => Proc.new { |error| "Oops" })
+        end
+
+        should "allow setting a custom error handler" do
+          assert_equal "Oops", @middleware.call(@env)
+        end
+      end
     end
   end
 end
\ No newline at end of file