viirya commented on a change in pull request #3336:
URL: https://github.com/apache/hadoop/pull/3336#discussion_r699650509
##########
File path:
hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/io/compress/TestCodec.java
##########
@@ -725,6 +726,172 @@ public void testGzipCompatibility() throws IOException {
assertArrayEquals(b, dflchk);
}
+ @Test
+ public void testGzipCompatibilityWithCompressor() throws IOException {
+ // don't use native libs
+ ZlibFactory.setNativeZlibLoaded(false);
+ Configuration conf = new Configuration();
+ CompressionCodec codec = ReflectionUtils.newInstance(GzipCodec.class,
conf);
+
+ for (int i = 0; i < 100; i++){
+ Compressor compressor = codec.createCompressor();
+ assertThat(compressor).withFailMessage("should be BuiltInGzipCompressor")
+ .isInstanceOf(BuiltInGzipCompressor.class);
+
+ Random r = new Random();
+ long seed = r.nextLong();
+ r.setSeed(seed);
+ LOG.info("seed: {}", seed);
+
+ int inputSize = r.nextInt(256 * 1024 + 1);
+ byte[] b = new byte[inputSize];
+ r.nextBytes(b);
+
+ compressor.setInput(b, 0, b.length);
+ compressor.finish();
+
+ byte[] output = new byte[inputSize + 1024];
+ int outputOff = 0;
+
+ while (!compressor.finished()) {
+ byte[] buf = new byte[r.nextInt(1024)];
+ int compressed = compressor.compress(buf, 0, buf.length);
+ System.arraycopy(buf, 0, output, outputOff, compressed);
+ outputOff += compressed;
+ }
+
+ DataInputBuffer gzbuf = new DataInputBuffer();
+ gzbuf.reset(output, outputOff);
+
+ Decompressor decom = codec.createDecompressor();
+ assertThat(decom).as("decompressor should not be null").isNotNull();
+ assertThat(decom).withFailMessage("should be BuiltInGzipDecompressor")
+ .isInstanceOf(BuiltInGzipDecompressor.class);
+ try (InputStream gzin = codec.createInputStream(gzbuf, decom);
+ DataOutputBuffer dflbuf = new DataOutputBuffer()) {
+ dflbuf.reset();
+ IOUtils.copyBytes(gzin, dflbuf, 4096);
+ final byte[] dflchk = Arrays.copyOf(dflbuf.getData(),
dflbuf.getLength());
+ assertThat(b).as("check decompressed output").isEqualTo(dflchk);
+ }
+ }
+ }
+
+ @Test
+ public void testGzipCompatibilityWithCompressorAndGZIPOutputStream() throws
IOException {
+ // don't use native libs
+ ZlibFactory.setNativeZlibLoaded(false);
+ Configuration conf = new Configuration();
+ CompressionCodec codec = ReflectionUtils.newInstance(GzipCodec.class,
conf);
+
+ for (int i = 0; i < 100; i++){
+ Compressor compressor = codec.createCompressor();
+ assertThat(compressor).withFailMessage("should be BuiltInGzipCompressor")
+ .isInstanceOf(BuiltInGzipCompressor.class);
+
+ Random r = new Random();
Review comment:
Ok, moved.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
