Add async arity to not-found route

Author: weavejester

src/compojure/route.clj

@@ -56,7 +56,10 @@
   supplied response body. The response body may be anything accepted by the
   [[response/render]] function."
   [body]
-  (fn [request]
-    (-> (response/render body request)
-        (status 404)
-        (cond-> (= (:request-method request) :head) (assoc :body nil)))))
+  (fn handler
+    ([request]
+     (-> (response/render body request)
+         (status 404)
+         (cond-> (= (:request-method request) :head) (assoc :body nil))))
+    ([request respond raise]
+     (respond (handler request)))))

test/compojure/route_test.clj

@@ -12,7 +12,15 @@
     (let [response ((route/not-found {:status 200 :body "bar"})
                     (mock/request :get "/"))]
       (is (= (:status response) 404))
-      (is (= (:body response) "bar")))))
+      (is (= (:body response) "bar"))))
+  (testing "async arity"
+    (let [handler   (route/not-found "baz")
+          response  (promise)
+          exception (promise)]
+      (handler (mock/request :get "/") response exception)
+      (is (not (realized? exception)))
+      (is (= (:status @response) 404))
+      (is (= (:body @response) "baz")))))
 
 (deftest resources-route
   (let [route    (route/resources "/foo" {:root "test_files"})