native compression tests

rashtao · rashtao · commit e3dccb04e63a · 2024-03-22T10:14:28.000+01:00
diff --git a/.github/workflows/native.yml b/.github/workflows/native.yml
@@ -14,12 +14,14 @@ jobs:
       fail-fast: false
       matrix:
         docker-img:
-          - docker.io/arangodb/enterprise:3.11.7
+          - docker.io/arangodb/enterprise-preview:devel-nightly
         topology:
-          - single
+          - cluster
         module:
           - driver
           - integration-tests
+        compression:
+          - true
 
     steps:
       - uses: actions/checkout@v2
@@ -36,6 +38,7 @@ jobs:
           ARANGO_LICENSE_KEY: ${{ secrets.ARANGO_LICENSE_KEY }}
           STARTER_MODE: ${{matrix.topology}}
           DOCKER_IMAGE: ${{matrix.docker-img}}
+          COMPRESSION: ${{matrix.compression}}
       - name: Info
         run: mvn -version
       - name: Install
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -218,12 +218,13 @@ jobs:
       fail-fast: false
       matrix:
         docker-img:
-          - docker.io/arangodb/arangodb:3.11.7
+          - docker.io/arangodb/arangodb-preview:devel-nightly
         topology:
           - single
-          - cluster
         java-version:
-          - 17
+          - 21
+        compression:
+          - true
 
     steps:
       - uses: actions/checkout@v2
@@ -238,6 +239,7 @@ jobs:
         env:
           STARTER_MODE: ${{matrix.topology}}
           DOCKER_IMAGE: ${{matrix.docker-img}}
+          COMPRESSION: ${{matrix.compression}}
       - name: Info
         run: mvn -version
       - name: Install
diff --git a/driver/src/test/java/com/arangodb/CompressionTest.java b/driver/src/test/java/com/arangodb/CompressionTest.java
@@ -0,0 +1,59 @@
+package com.arangodb;
+
+import com.arangodb.config.ArangoConfigProperties;
+import com.fasterxml.jackson.databind.JsonNode;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.EnumSource;
+
+import java.util.Locale;
+import java.util.UUID;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+/**
+ * Non-exhaustive tests of content encoding, executed during integration and native tests.
+ * A complete test is at resilience-tests/src/test/java/resilience/compression/CompressionTest.java
+ *
+ * @author Michele Rastelli
+ */
+class CompressionTest extends BaseJunit5 {
+
+    @ParameterizedTest
+    @EnumSource(Protocol.class)
+    void gzip(Protocol protocol) {
+        doTest(protocol, Compression.GZIP);
+    }
+
+    @ParameterizedTest
+    @EnumSource(Protocol.class)
+    void deflate(Protocol protocol) {
+        doTest(protocol, Compression.DEFLATE);
+    }
+
+    void doTest(Protocol protocol, Compression compression) {
+        assumeTrue(isAtLeastVersion(3, 12));
+        assumeTrue(protocol != Protocol.VST);
+
+        ArangoDB adb = new ArangoDB.Builder()
+                .loadProperties(ArangoConfigProperties.fromFile())
+                .protocol(protocol)
+                .compression(compression)
+                .compressionThreshold(0)
+                .compressionLevel(3)
+                .build();
+
+        Response<JsonNode> resp = adb.execute(Request.builder()
+                .method(Request.Method.POST)
+                .path("/_admin/echo")
+                .body(UUID.randomUUID().toString())
+                .build(), JsonNode.class);
+
+        String encoding = compression.toString().toLowerCase(Locale.ROOT);
+        String reqAcceptEncoding = resp.getBody().get("headers").get("accept-encoding").textValue();
+        assertThat(reqAcceptEncoding).contains(encoding);
+
+        adb.shutdown();
+    }
+
+}
