tests

Signed-off-by: Stanislav Knot <sknot@redhat.com>
strimzi · Jun 2, 2021 · b834225 · b834225
1 parent 7c038fa
commit b834225
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/src/test/java/io/strimzi/kafka/bridge/http/OtherServicesIT.java b/src/test/java/io/strimzi/kafka/bridge/http/OtherServicesIT.java
@@ -173,4 +173,44 @@ void openApiTestWithForwardedPath(VertxTestContext context) {
                     context.completeNow();
                 });
     }
+
+    @Test
+    void openApiTestWithForwardedPrefix(VertxTestContext context) {
+        String forwardedPrefix = "/app/kafka-bridge";
+        baseService()
+                .getRequest("/openapi")
+                .putHeader("x-Forwarded-Prefix", forwardedPrefix)
+                .as(BodyCodec.jsonObject())
+                .send(ar -> {
+                    context.verify(() -> {
+                        assertThat(ar.succeeded(), is(true));
+                        HttpResponse<JsonObject> response = ar.result();
+                        assertThat(response.statusCode(), is(HttpResponseStatus.OK.code()));
+                        JsonObject bridgeResponse = response.body();
+                        assertThat(bridgeResponse.getString("basePath"), is(forwardedPrefix));
+                    });
+                    context.completeNow();
+                });
+    }
+
+    @Test
+    void openApiTestWithForwardedPathAndPrefix(VertxTestContext context) {
+        String forwardedPath = "/app/kafka-bridge-path";
+        String forwardedPrefix = "/app/kafka-bridge-prefix";
+        baseService()
+                .getRequest("/openapi")
+                .putHeader("x-Forwarded-Path", forwardedPath)
+                .putHeader("x-Forwarded-Prefix", forwardedPrefix)
+                .as(BodyCodec.jsonObject())
+                .send(ar -> {
+                    context.verify(() -> {
+                        assertThat(ar.succeeded(), is(true));
+                        HttpResponse<JsonObject> response = ar.result();
+                        assertThat(response.statusCode(), is(HttpResponseStatus.OK.code()));
+                        JsonObject bridgeResponse = response.body();
+                        assertThat(bridgeResponse.getString("basePath"), is(forwardedPath));
+                    });
+                    context.completeNow();
+                });
+    }
 }