test compression

Author: rashtao

resilience-tests/src/test/java/resilience/ActiveFailoverTest.java

@@ -49,6 +49,7 @@ void beforeEach() throws IOException {
         for (Endpoint ph : endpoints) {
             ph.getProxy().enable();
         }
+        logs.reset();
     }
 
     protected static List<Endpoint> getEndpoints() {

resilience-tests/src/test/java/resilience/ClusterTest.java

@@ -54,6 +54,7 @@ void beforeEach() throws IOException {
         for (Endpoint endpoint : endpoints) {
             endpoint.getProxy().enable();
         }
+        logs.reset();
     }
 
     protected static List<Endpoint> getEndpoints() {

resilience-tests/src/test/java/resilience/SingleServerTest.java

@@ -37,6 +37,7 @@ static void afterAll() throws IOException {
     @BeforeEach
     void beforeEach() {
         getEndpoint().enable();
+        logs.reset();
     }
 
     protected static Endpoint getEndpoint() {

resilience-tests/src/test/java/resilience/compression/CompressionTest.java

@@ -0,0 +1,72 @@
+package resilience.compression;
+
+import ch.qos.logback.classic.spi.ILoggingEvent;
+import com.arangodb.ArangoDB;
+import com.arangodb.Compression;
+import com.arangodb.Protocol;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.EnumSource;
+import resilience.ClusterTest;
+import resilience.SingleServerTest;
+
+import java.util.Collections;
+import java.util.List;
+import java.util.Locale;
+import java.util.UUID;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+/**
+ * @author Michele Rastelli
+ */
+class CompressionTest extends ClusterTest {
+
+    @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(protocol != Protocol.VST);
+        assumeTrue(protocol != Protocol.HTTP_VPACK);
+        assumeTrue(protocol != Protocol.HTTP_JSON);
+
+        ArangoDB adb = dbBuilder()
+                .protocol(protocol)
+                .compression(compression)
+                .compressionThreshold(0)
+                .build();
+
+        List<String> data = IntStream.range(0, 500)
+                .mapToObj(i -> UUID.randomUUID().toString())
+                .collect(Collectors.toList());
+
+        adb.db().query("FOR i IN @data RETURN i", String.class,
+                Collections.singletonMap("data", data)).asListRemaining();
+
+        adb.shutdown();
+
+        String compressionLC = compression.toString().toLowerCase(Locale.ROOT);
+
+        // request
+        assertThat(logs.getLogs())
+                .map(ILoggingEvent::getFormattedMessage)
+                .anyMatch(l -> l.contains("content-encoding: " + compressionLC) && l.contains("accept-encoding: " + compressionLC));
+
+        // response
+        assertThat(logs.getLogs())
+                .map(ILoggingEvent::getFormattedMessage)
+                .anyMatch(l -> l.contains("content-encoding: " + compressionLC) && l.contains("server: ArangoDB"));
+    }
+
+}