Codebase list comidi-clojure / 3831b6f
Merge commit 'b116202' Chris Price 8 years ago
2 changed file(s) with 70 addition(s) and 2 deletion(s). Raw diff Collapse all Expand all
+28
-2
src/puppetlabs/comidi.clj less more
270270 [method pattern bindings body]
271271 `[~pattern {~method (handler-fn* ~bindings ~body)}])
272272
273 (defn wrap-routes*
274 "Help function, used by compojure-like wrap-routes function to wrap leaf handlers
275 in the bidi route with the middleware"
276 [loc middleware]
277 (let [node (zip/node loc)
278 loc (cond
279 (fn? node) (zip/replace loc (middleware node))
280 (map? node) (zip/replace
281 loc
282 (reduce-kv (fn [m k v] (assoc m k (middleware v))) {} node))
283 :else loc)]
284 (if (zip/end? loc)
285 loc
286 (wrap-routes* (zip/next loc) middleware))))
287
273288 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
274289 ;;;; Public - core functions
275290
301316 :match-context match-context))))))
302317
303318 (schema/defn ^:always-validate
304 routes :- bidi-schema/RoutePair
319 wrap-routes :- bidi-schema/RoutePair
320 "Wraps middleware around the handlers at every leaf in the route in a manner
321 analagous to compojure's wrap-routes function"
322 [routes :- bidi-schema/RoutePair
323 middleware :- (schema/pred fn?)]
324 (-> routes
325 zip/vector-zip
326 (wrap-routes* middleware)
327 zip/root))
328
329 (schema/defn ^:always-validate
330 routes :- bidi-schema/RoutePair
305331 "Combines multiple bidi routes into a single data structure; this is largely
306332 just a convenience function for grouping several routes together as a single
307333 object that can be passed around."
309335 ["" (vec routes)])
310336
311337 (schema/defn ^:always-validate
312 context :- bidi-schema/RoutePair
338 context :- bidi-schema/RoutePair
313339 "Combines multiple bidi routes together into a single data structure, but nests
314340 them all under the given url-prefix. This is similar to compojure's `context`
315341 macro, but does not accept a binding form. You can still destructure variables
+42
-0
test/puppetlabs/comidi_test.clj less more
287287 routes->handler
288288 wrap-with-fake-match-context)]
289289 (is (= "bar!" (handler {:uri "/bunk"}))))))
290
291 (deftest wrap-leaves-with-middleware-test
292 (let [inner-middleware (fn [handler]
293 (fn [request]
294 (update-in (handler request) [:body] #(str "inner-" %))))
295 bb-wrapper-middleware (fn [handler]
296 (fn [request]
297 (update-in (handler request) [:body] #(str "bb-wrapper-" %))))
298 outer-middleware (fn [handler]
299 (fn [request]
300 (update-in (handler request) [:body] #(str "outer-" %))))
301 aa-route (GET "/aa" request "aa!")
302 bb-route (ANY "/bb" request "bb!")
303 cc-route (ANY "/cc" request "cc!")
304 dd-route (DELETE "/dd" request "dd!")
305 ee-route (GET "/ee" request "ee!")
306 ff-route (ANY "/ff" request "ff!")
307 left-routes (context "/left" aa-route bb-route)
308 middle-routes (context "/middle" cc-route dd-route)
309 right-routes (context "/right" ee-route ff-route)
310 handler (-> (routes left-routes middle-routes right-routes) routes->handler)]
311 (testing "Routes without middleware applied"
312 (is (= (:body (handler {:uri "/left/aa" :request-method :get})) "aa!"))
313 (is (= (:body (handler {:uri "/left/bb" :request-method :post})) "bb!"))
314 (is (= (:body (handler {:uri "/middle/cc" :request-method :get})) "cc!"))
315 (is (= (:body (handler {:uri "/middle/dd" :request-method :delete})) "dd!"))
316 (is (= (:body (handler {:uri "/right/ee" :request-method :get})) "ee!"))
317 (is (= (:body (handler {:uri "/right/ff" :request-method :delete})) "ff!")))
318 (testing "Routes but now with middleware applied"
319 (let [wrapped-bb-route (-> bb-route (wrap-routes bb-wrapper-middleware))
320 left-routes (-> (context "/left" aa-route wrapped-bb-route)
321 (wrap-routes inner-middleware)
322 (wrap-routes outer-middleware))
323 middle-routes (context "/middle" cc-route dd-route)
324 right-routes (-> (context "/right" ee-route ff-route) (wrap-routes outer-middleware))
325 handler (-> (routes left-routes middle-routes right-routes) routes->handler)]
326 (is (= (:body (handler {:uri "/left/aa" :request-method :get})) "outer-inner-aa!"))
327 (is (= (:body (handler {:uri "/left/bb" :request-method :post})) "outer-inner-bb-wrapper-bb!"))
328 (is (= (:body (handler {:uri "/middle/cc" :request-method :get})) "cc!"))
329 (is (= (:body (handler {:uri "/middle/dd" :request-method :delete})) "dd!"))
330 (is (= (:body (handler {:uri "/right/ee" :request-method :get})) "outer-ee!"))
331 (is (= (:body (handler {:uri "/right/ff" :request-method :delete})) "outer-ff!"))))))