add brotli to IntegrationTest (#1449)

Netflix · Jan 31, 2023 · 378d105 · 378d105
1 parent fded9b4
commit 378d105
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 0 deletions.
diff --git a/gradle.properties b/gradle.properties
@@ -2,5 +2,6 @@ versions_groovy=3.0.13
 versions_ribbon=2.4.4
 versions_netty=4.1.87.Final
 versions_netty_io_uring=0.0.16.Final
+versions_brotli4j=1.9.0
 release.scope=patch
 release.version=2.3.1-SNAPSHOT
diff --git a/zuul-integration-test/build.gradle b/zuul-integration-test/build.gradle
@@ -16,6 +16,11 @@ dependencies {
     testImplementation libraries.truth, libraries.jupiterApi, libraries.jupiterParams, libraries.jupiterEngine, libraries.jupiterMockito, libraries.okhttp
     testRuntimeOnly 'org.apache.logging.log4j:log4j-core:2.19.0'
     testRuntimeOnly 'org.apache.logging.log4j:log4j-slf4j-impl:2.19.0'
+    testRuntimeOnly "com.aayushatharva.brotli4j:brotli4j:$versions_brotli4j"
+    testRuntimeOnly "com.aayushatharva.brotli4j:native-osx-aarch64:$versions_brotli4j"
+    testRuntimeOnly "com.aayushatharva.brotli4j:native-osx-x86_64:$versions_brotli4j"
+    testRuntimeOnly "com.aayushatharva.brotli4j:native-linux-x86_64:$versions_brotli4j"
+    testRuntimeOnly "com.aayushatharva.brotli4j:native-linux-aarch64:$versions_brotli4j"
 }
 
 tasks.withType(Test).all {

diff --git a/zuul-integration-test/src/test/java/com/netflix/zuul/integration/IntegrationTest.java b/zuul-integration-test/src/test/java/com/netflix/zuul/integration/IntegrationTest.java
@@ -27,6 +27,7 @@
 import com.netflix.zuul.integration.server.Bootstrap;
 import com.netflix.zuul.integration.server.HeaderNames;
 import com.netflix.zuul.integration.server.TestUtil;
+import io.netty.handler.codec.compression.Brotli;
 import io.netty.util.ResourceLeakDetector;
 import okhttp3.HttpUrl;
 import okhttp3.OkHttpClient;
@@ -389,6 +390,41 @@ void gzipOnly() throws Exception {
         connection.disconnect();
     }
 
+    @Test
+    void brotliOnly() throws Throwable {
+        Brotli.ensureAvailability();
+        final String expectedResponseBody = TestUtil.COMPRESSIBLE_CONTENT;
+        final WireMock wireMock = wmRuntimeInfo.getWireMock();
+        wireMock.register(
+                get(anyUrl())
+                        .willReturn(
+                                aResponse()
+                                        .withStatus(200)
+                                        .withBody(expectedResponseBody)
+                                        .withHeader("Content-Type", TestUtil.COMPRESSIBLE_CONTENT_TYPE)));
+
+        URL url = new URL(zuulBaseUri);
+        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
+        connection.setRequestMethod("GET");
+        connection.setAllowUserInteraction(false);
+        connection.setRequestProperty("Accept-Encoding", "br");
+        InputStream inputStream = connection.getInputStream();
+        assertEquals(200, connection.getResponseCode());
+        assertEquals("text/plain", connection.getHeaderField("Content-Type"));
+        assertEquals("br", connection.getHeaderField("Content-Encoding"));
+        byte[] compressedData = IOUtils.toByteArray(inputStream);
+        assertTrue(compressedData.length > 0);
+        /* TODO : call brotli4j DirectDecompress
+        DirectDecompress decompressResult = DirectDecompress.decompress(ByteBufUtil.getBytes(contentBuf));
+        assertEquals(DecoderJNI.Status.DONE, decompressResult.getResultStatus());
+        assertEquals("blah blah blah",
+                new String(decompressResult.getDecompressedData(), TestUtil.CHARSET));
+                */
+
+        inputStream.close();
+        connection.disconnect();
+    }
+
     @Test
     void noCompression() throws Exception {
         final String expectedResponseBody = TestUtil.COMPRESSIBLE_CONTENT;