From 3333b9ee98a5f94a0b259d9d2250801ff3cc5e5f Mon Sep 17 00:00:00 2001 From: serega6531 Date: Tue, 10 Mar 2020 17:13:46 +0300 Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=20=D1=82=D0=B5=D1=81=D1=82=20=D1=80=D0=B0=D1=81=D0=BF?= =?UTF-8?q?=D0=B0=D0=BA=D0=BE=D0=B2=D0=BA=D0=B8=20gzip?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../packmate/StreamOptimizerTest.java | 20 +++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/src/test/java/ru/serega6531/packmate/StreamOptimizerTest.java b/src/test/java/ru/serega6531/packmate/StreamOptimizerTest.java index 4f83ae3..ecdbf5c 100644 --- a/src/test/java/ru/serega6531/packmate/StreamOptimizerTest.java +++ b/src/test/java/ru/serega6531/packmate/StreamOptimizerTest.java @@ -1,20 +1,36 @@ package ru.serega6531.packmate; +import org.apache.commons.lang3.ArrayUtils; import org.junit.jupiter.api.Test; import ru.serega6531.packmate.model.CtfService; import ru.serega6531.packmate.model.Packet; import ru.serega6531.packmate.service.StreamOptimizer; import java.util.ArrayList; +import java.util.Base64; import java.util.List; -import static org.junit.jupiter.api.Assertions.*; +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertTrue; class StreamOptimizerTest { @Test void testUnpackGzip() { - //TODO + String encoded = "H4sIAAAAAAAA/0tMTExKSgIA2KWG6gYAAAA="; + final byte[] gzipped = Base64.getDecoder().decode(encoded); + final byte[] content = ArrayUtils.addAll("HTTP/1.1 200 OK\r\nContent-Encoding: gzip\r\nContent-Length: 26\r\n\r\n".getBytes(), gzipped); + + CtfService service = new CtfService(); + service.setUngzipHttp(true); + + Packet p = createPacket(content, false); + List list = new ArrayList<>(); + list.add(p); + + new StreamOptimizer(service, list).optimizeStream(); + final String processed = new String(list.get(0).getContent()); + assertTrue(processed.contains("aaabbb")); } @Test