refactor: ZstdDecompressOutputStream - Test more compression levels i…

…n the test
avast · Dec 21, 2023 · 0221766 · 0221766
1 parent 1a5ce97
commit 0221766
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/...src/test/scala/com/avast/clients/storage/compression/ZstdDecompressOutputStreamTest.scala b/...src/test/scala/com/avast/clients/storage/compression/ZstdDecompressOutputStreamTest.scala
@@ -22,15 +22,22 @@ class ZstdDecompressOutputStreamTest extends FunSuite {
 
   test("decompress zstd stream") {
     val chunkSize = 4 * 1024
-    val testCases = Seq(0, 1, chunkSize, 10 * 1024 * 1024)
 
-    for (testCase <- testCases) {
-      val original_data = generateData(testCase)
+    val lengths = Seq(0, 1, chunkSize, 10 * 1024 * 1024)
+    val levels = Seq(Zstd.minCompressionLevel(), Zstd.defaultCompressionLevel(), 9, Zstd.maxCompressionLevel())
+
+    val testCases = for {
+      size <- lengths
+      level <- levels
+    } yield (size, level)
+
+    for ((size, level) <- testCases) {
+      val original_data = generateData(size)
       val original_sha256 = computeSha256(original_data)
 
-      println(s"Original data size: ${original_data.length}")
-      val compressed_data = Zstd.compress(original_data, 9)
-      println(s"Compressed data size: ${compressed_data.length}")
+      println(s"Original data size: ${original_data.length}, level: $level")
+      val compressed_data = Zstd.compress(original_data, level)
+      println(s"Compressed data size: ${compressed_data.length}, level: $level")
 
       val sourceStream = ByteBuffer.wrap(compressed_data)
       val targetStream = new ByteArrayOutputStream()