1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 package org.apache.hc.client5.http.entity;
29
30 import java.io.ByteArrayInputStream;
31 import java.io.ByteArrayOutputStream;
32 import java.io.IOException;
33 import java.io.OutputStream;
34 import java.nio.charset.StandardCharsets;
35
36 import org.apache.hc.core5.http.ContentType;
37 import org.apache.hc.core5.http.HttpEntity;
38 import org.apache.hc.core5.http.io.entity.ByteArrayEntity;
39 import org.apache.hc.core5.http.io.entity.EntityUtils;
40 import org.apache.hc.core5.http.io.entity.InputStreamEntity;
41 import org.apache.hc.core5.http.io.entity.StringEntity;
42 import org.junit.jupiter.api.Assertions;
43 import org.junit.jupiter.api.Test;
44 import org.mockito.ArgumentMatchers;
45 import org.mockito.Mockito;
46
47 public class TestGZip {
48
49 @Test
50 public void testBasic() throws Exception {
51 final String s = "some kind of text";
52 final StringEntity e = new StringEntity(s, ContentType.TEXT_PLAIN, false);
53 try (final GzipCompressingEntity gzipe = new GzipCompressingEntity(e)) {
54 Assertions.assertTrue(gzipe.isChunked());
55 Assertions.assertEquals(-1, gzipe.getContentLength());
56 Assertions.assertNotNull(gzipe.getContentEncoding());
57 Assertions.assertEquals("gzip", gzipe.getContentEncoding());
58 }
59 }
60
61 @Test
62 public void testCompressionDecompression() throws Exception {
63 final StringEntity in = new StringEntity("some kind of text", ContentType.TEXT_PLAIN);
64 try (final GzipCompressingEntity gzipe = new GzipCompressingEntity(in)) {
65 final ByteArrayOutputStream buf = new ByteArrayOutputStream();
66 gzipe.writeTo(buf);
67 final ByteArrayEntity out = new ByteArrayEntity(buf.toByteArray(), ContentType.APPLICATION_OCTET_STREAM);
68 final GzipDecompressingEntity gunzipe = new GzipDecompressingEntity(out);
69 Assertions.assertEquals("some kind of text", EntityUtils.toString(gunzipe, StandardCharsets.US_ASCII));
70 }
71 }
72
73 @Test
74 public void testCompressionIOExceptionLeavesOutputStreamOpen() throws Exception {
75 final HttpEntity in = Mockito.mock(HttpEntity.class);
76 Mockito.doThrow(new IOException("Ooopsie")).when(in).writeTo(ArgumentMatchers.any());
77 try (final GzipCompressingEntity gzipe = new GzipCompressingEntity(in)) {
78 final OutputStream out = Mockito.mock(OutputStream.class);
79 try {
80 gzipe.writeTo(out);
81 } catch (final IOException ex) {
82 Mockito.verify(out, Mockito.never()).close();
83 }
84 }
85 }
86
87 @Test
88 public void testDecompressionWithMultipleGZipStream() throws Exception {
89 final int[] data = new int[] {
90 0x1f, 0x8b, 0x08, 0x08, 0x03, 0xf1, 0x55, 0x5a, 0x00, 0x03, 0x74, 0x65, 0x73, 0x74, 0x31, 0x00,
91 0x2b, 0x2e, 0x29, 0x4a, 0x4d, 0xcc, 0xd5, 0x35, 0xe4, 0x02, 0x00, 0x03, 0x61, 0xf0, 0x5f, 0x09,
92 0x00, 0x00, 0x00, 0x1f, 0x8b, 0x08, 0x08, 0x08, 0xf1, 0x55, 0x5a, 0x00, 0x03, 0x74, 0x65, 0x73,
93 0x74, 0x32, 0x00, 0x2b, 0x2e, 0x29, 0x4a, 0x4d, 0xcc, 0xd5, 0x35, 0xe2, 0x02, 0x00, 0xc0, 0x32,
94 0xdd, 0x74, 0x09, 0x00, 0x00, 0x00
95 };
96 final byte[] bytes = new byte[data.length];
97 for (int i = 0; i < data.length; i++) {
98 bytes[i] = (byte) (data[i] & 0xff);
99 }
100
101 try (final GzipDecompressingEntity entity = new GzipDecompressingEntity(new InputStreamEntity(new ByteArrayInputStream(bytes), ContentType.APPLICATION_OCTET_STREAM))) {
102 Assertions.assertEquals("stream-1\nstream-2\n", EntityUtils.toString(entity, StandardCharsets.US_ASCII));
103 }
104 }
105
106 }