Extends tests to better test the deflater level

TooTallNate · Oct 13, 2024 · 4f4aed5 · 4f4aed5
1 parent fe7635b
commit 4f4aed5
Showing 1 changed file with 107 additions and 0 deletions.
diff --git a/src/test/java/org/java_websocket/extensions/PerMessageDeflateExtensionTest.java b/src/test/java/org/java_websocket/extensions/PerMessageDeflateExtensionTest.java
@@ -1,12 +1,14 @@
 package org.java_websocket.extensions;
 
+import static java.util.zip.GZIPInputStream.GZIP_MAGIC;
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
 
 import java.nio.ByteBuffer;
+import java.util.Arrays;
 import java.util.zip.Deflater;
 import java.util.zip.Inflater;
 import org.java_websocket.exceptions.InvalidDataException;
@@ -51,6 +53,111 @@ public void testDecodeFrameIfRSVIsNotSet() throws InvalidDataException {
     assertFalse(frame.isRSV1());
   }
 
+  @Test
+  public void testDecodeFrameNoCompression() throws InvalidDataException {
+    PerMessageDeflateExtension deflateExtension = new PerMessageDeflateExtension();
+    deflateExtension.setDeflaterLevel(Deflater.NO_COMPRESSION);
+    deflateExtension.setThreshold(0);
+    String str = "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text";
+    byte[] message = str.getBytes();
+    TextFrame frame = new TextFrame();
+    frame.setPayload(ByteBuffer.wrap(message));
+    deflateExtension.encodeFrame(frame);
+    byte[] payloadArray = frame.getPayloadData().array();
+    assertArrayEquals(message, Arrays.copyOfRange(payloadArray, 5,payloadArray.length-5));
+    assertTrue(frame.isRSV1());
+    deflateExtension.decodeFrame(frame);
+    assertArrayEquals(message, frame.getPayloadData().array());
+  }
+
+  @Test
+  public void testDecodeFrameBestSpeedCompression() throws InvalidDataException {
+    PerMessageDeflateExtension deflateExtension = new PerMessageDeflateExtension();
+    deflateExtension.setDeflaterLevel(Deflater.BEST_SPEED);
+    deflateExtension.setThreshold(0);
+    String str = "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text";
+    byte[] message = str.getBytes();
+    TextFrame frame = new TextFrame();
+    frame.setPayload(ByteBuffer.wrap(message));
+
+    Deflater localDeflater = new Deflater(Deflater.BEST_SPEED,true);
+    localDeflater.setInput(ByteBuffer.wrap(message).array());
+    byte[] buffer = new byte[1024];
+    int bytesCompressed = localDeflater.deflate(buffer, 0, buffer.length, Deflater.SYNC_FLUSH);
+
+    deflateExtension.encodeFrame(frame);
+    byte[] payloadArray = frame.getPayloadData().array();
+    assertArrayEquals(Arrays.copyOfRange(buffer,0, bytesCompressed), Arrays.copyOfRange(payloadArray,0,payloadArray.length));
+    assertTrue(frame.isRSV1());
+    deflateExtension.decodeFrame(frame);
+    assertArrayEquals(message, frame.getPayloadData().array());
+  }
+
+  @Test
+  public void testDecodeFrameBestCompression() throws InvalidDataException {
+    PerMessageDeflateExtension deflateExtension = new PerMessageDeflateExtension();
+    deflateExtension.setDeflaterLevel(Deflater.BEST_COMPRESSION);
+    deflateExtension.setThreshold(0);
+    String str = "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text";
+    byte[] message = str.getBytes();
+    TextFrame frame = new TextFrame();
+    frame.setPayload(ByteBuffer.wrap(message));
+
+    Deflater localDeflater = new Deflater(Deflater.BEST_COMPRESSION,true);
+    localDeflater.setInput(ByteBuffer.wrap(message).array());
+    byte[] buffer = new byte[1024];
+    int bytesCompressed = localDeflater.deflate(buffer, 0, buffer.length, Deflater.SYNC_FLUSH);
+
+    deflateExtension.encodeFrame(frame);
+    byte[] payloadArray = frame.getPayloadData().array();
+    assertArrayEquals(Arrays.copyOfRange(buffer,0, bytesCompressed), Arrays.copyOfRange(payloadArray,0,payloadArray.length));
+    assertTrue(frame.isRSV1());
+    deflateExtension.decodeFrame(frame);
+    assertArrayEquals(message, frame.getPayloadData().array());
+  }
+
+  @Test
+  public void testDecodeFrameSwitchCompression() throws InvalidDataException {
+    PerMessageDeflateExtension deflateExtension = new PerMessageDeflateExtension();
+    deflateExtension.setDeflaterLevel(Deflater.NO_COMPRESSION);
+    deflateExtension.setThreshold(0);
+    String str = "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text"
+            + "This is a highly compressable text";
+    byte[] message = str.getBytes();
+    TextFrame frame = new TextFrame();
+    frame.setPayload(ByteBuffer.wrap(message));
+
+    Deflater localDeflater = new Deflater(Deflater.BEST_COMPRESSION,true);
+    localDeflater.setInput(ByteBuffer.wrap(message).array());
+    byte[] buffer = new byte[1024];
+    int bytesCompressed = localDeflater.deflate(buffer, 0, buffer.length, Deflater.SYNC_FLUSH);
+
+    // Change the deflater level after the creation and switch to a new deflater level
+    // Compression strategy should be applied instantly since we call .deflate manually
+    deflateExtension.setDeflaterLevel(Deflater.BEST_COMPRESSION);
+    deflateExtension.encodeFrame(frame);
+    byte[] payloadArray = frame.getPayloadData().array();
+    assertArrayEquals(Arrays.copyOfRange(buffer,0, bytesCompressed), Arrays.copyOfRange(payloadArray,0,payloadArray.length));
+    assertTrue(frame.isRSV1());
+    deflateExtension.decodeFrame(frame);
+    assertArrayEquals(message, frame.getPayloadData().array());
+  }
+
   @Test
   public void testEncodeFrame() {
     PerMessageDeflateExtension deflateExtension = new PerMessageDeflateExtension();