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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
(ns puppetlabs.trapperkeeper.services.webrouting.webrouting-service-proxy-test
(:require [clojure.test :refer :all]
[puppetlabs.trapperkeeper.services.webrouting.webrouting-service :refer :all]
[puppetlabs.trapperkeeper.services.webserver.jetty9-service :refer [jetty9-service]]
[puppetlabs.trapperkeeper.testutils.webrouting.common :refer :all]
[puppetlabs.trapperkeeper.app :refer [get-service]]
[puppetlabs.trapperkeeper.testutils.bootstrap :refer [with-app-with-config]]
[puppetlabs.trapperkeeper.services :as tk-services]
[schema.test :as schema-test]
[puppetlabs.trapperkeeper.testutils.webserver :as testutils]))
(use-fixtures :once
schema-test/validate-schemas
testutils/assert-clean-shutdown)
(defprotocol DummyService1
(dummy1 [this]))
(defprotocol DummyService2
(dummy2 [this]))
(tk-services/defservice dummy-service1
DummyService1
[]
(dummy1 [this]
"This is a dummy function. Please ignore."))
(tk-services/defservice dummy-service2
DummyService2
[]
(dummy2 [this]
"This is a dummy function. Please ignore."))
(defmacro with-target-and-proxy-servers
[{:keys [target proxy proxy-config proxy-opts]} & body]
`(with-app-with-config proxy-target-app#
[jetty9-service
webrouting-service
dummy-service1]
{:webserver ~target
:web-router-service {:puppetlabs.trapperkeeper.services.webrouting.webrouting-service-proxy-test/dummy-service1 "/hello"}}
(let [target-webserver# (get-service proxy-target-app# :WebroutingService)
svc# (get-service proxy-target-app# :DummyService1)]
(add-ring-handler
target-webserver#
svc#
(fn [req#]
(if (= "/hello/world" (:uri req#))
{:status 200 :body (str "Hello, World!"
((:headers req#) "x-fancy-proxy-header"))}
{:status 404 :body "D'oh"}))))
(with-app-with-config proxy-app#
[jetty9-service
webrouting-service
dummy-service2]
{:webserver ~proxy
:web-router-service {:puppetlabs.trapperkeeper.services.webrouting.webrouting-service-proxy-test/dummy-service2
{:bar "/hello-proxy"
:foo "/goodbye-proxy"}}}
(let [proxy-webserver# (get-service proxy-app# :WebroutingService)
svc# (get-service proxy-app# :DummyService2)]
(if ~proxy-opts
(add-proxy-route proxy-webserver# svc# ~proxy-config ~proxy-opts)
(add-proxy-route proxy-webserver# svc# ~proxy-config {:route-id :bar})))
~@body)))
(deftest proxy-test-web-routing
(testing "proxy support with web routing"
(with-target-and-proxy-servers
{:target {:host "0.0.0.0"
:port 9000}
:proxy {:host "0.0.0.0"
:port 10000}
:proxy-config {:host "localhost"
:port 9000
:path "/hello"}}
(let [response (http-get "http://localhost:9000/hello/world")]
(is (= (:status response) 200))
(is (= (:body response) "Hello, World!")))
(let [response (http-get "http://localhost:10000/hello-proxy/world")]
(is (= (:status response) 200))
(is (= (:body response) "Hello, World!")))))
(testing "basic https proxy support with multiple web routes"
(with-target-and-proxy-servers
{:target {:host "0.0.0.0"
:port 9000}
:proxy {:host "0.0.0.0"
:port 10000}
:proxy-config {:host "localhost"
:port 9000
:path "/hello"}
:proxy-opts {:route-id :foo}}
(let [response (http-get "http://localhost:9000/hello/world")]
(is (= (:status response) 200))
(is (= (:body response) "Hello, World!")))
(let [response (http-get "http://localhost:10000/goodbye-proxy/world")]
(is (= (:status response) 200))
(is (= (:body response) "Hello, World!"))))))
|
