1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
(ns ring.middleware.proxy-headers-test
(:use clojure.test
ring.middleware.proxy-headers
[ring.mock.request :only [request header]]
[ring.util.response :only [response]]))
(deftest test-wrap-forwarded-remote-addr
(let [handler (wrap-forwarded-remote-addr (comp response :remote-addr))]
(testing "without x-forwarded-for"
(let [req (assoc (request :get "/") :remote-addr "1.2.3.4")
resp (handler req)]
(is (= (:body resp) "1.2.3.4"))))
(testing "with x-forwarded-for"
(let [req (-> (request :get "/")
(assoc :remote-addr "127.0.0.1")
(header "x-forwarded-for" "1.2.3.4"))
resp (handler req)]
(is (= (:body resp) "1.2.3.4"))))
(testing "with multiple proxies"
(let [req (-> (request :get "/")
(assoc :remote-addr "127.0.0.1")
(header "x-forwarded-for" "10.0.1.9, 192.168.4.98, 1.2.3.4"))
resp (handler req)]
(is (= (:body resp) "1.2.3.4"))))))
(deftest test-wrap-forwarded-remote-addr-cps
(let [handler (wrap-forwarded-remote-addr
(fn [request respond _] (respond (response (:remote-addr request)))))]
(testing "without x-forwarded-for"
(let [req (assoc (request :get "/") :remote-addr "1.2.3.4")
resp (promise)
ex (promise)]
(handler req resp ex)
(is (not (realized? ex)))
(is (= (:body @resp) "1.2.3.4"))))
(testing "with x-forwarded-for"
(let [req (-> (request :get "/")
(assoc :remote-addr "127.0.0.1")
(header "x-forwarded-for" "1.2.3.4"))
resp (promise)
ex (promise)]
(handler req resp ex)
(is (not (realized? ex)))
(is (= (:body @resp) "1.2.3.4"))))))
|
