| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.commons.compress.compressors.lz4; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.IOException; |
| import java.util.Arrays; |
| import java.util.Collection; |
| import org.apache.commons.compress.AbstractTestCase; |
| import org.apache.commons.compress.utils.IOUtils; |
| import org.junit.Assert; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| import org.junit.runners.Parameterized.Parameters; |
| |
| @RunWith(Parameterized.class) |
| public final class FramedLZ4CompressorRoundtripTest extends AbstractTestCase { |
| |
| @Parameters(name = "using {0}") |
| public static Collection<Object[]> factory() { |
| return Arrays.asList(new Object[][] { |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.K64) }, |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.K256) }, |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.M1) }, |
| new Object[] { FramedLZ4CompressorOutputStream.Parameters.DEFAULT }, |
| // default without content checksum |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.M4, |
| false, false, false) }, |
| // default with block checksum |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.M4, |
| true, true, false) }, |
| // small blocksize (so we get enough blocks) and enabled block dependency, otherwise defaults |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.K64, |
| true, false, true) }, |
| // default, tuned for speed |
| new Object[] { new FramedLZ4CompressorOutputStream.Parameters(FramedLZ4CompressorOutputStream.BlockSize.M4, |
| true, false, false, BlockLZ4CompressorOutputStream.createParameterBuilder() |
| .tunedForSpeed().build()) }, |
| }); |
| } |
| |
| private final FramedLZ4CompressorOutputStream.Parameters params; |
| |
| public FramedLZ4CompressorRoundtripTest(FramedLZ4CompressorOutputStream.Parameters params) { |
| this.params = params; |
| } |
| |
| private void roundTripTest(String testFile) throws IOException { |
| File input = getFile(testFile); |
| long start = System.currentTimeMillis(); |
| final File outputSz = new File(dir, input.getName() + ".framed.lz4"); |
| byte[] expected; |
| try (FileInputStream is = new FileInputStream(input)) { |
| expected = IOUtils.toByteArray(is); |
| } |
| ByteArrayOutputStream bos = new ByteArrayOutputStream(); |
| try (FramedLZ4CompressorOutputStream los = new FramedLZ4CompressorOutputStream(bos, |
| params)) { |
| IOUtils.copy(new ByteArrayInputStream(expected), los); |
| } |
| System.err.println(input.getName() + " written, uncompressed bytes: " + input.length() |
| + ", compressed bytes: " + outputSz.length() + " after " + (System.currentTimeMillis() - start) + "ms"); |
| start = System.currentTimeMillis(); |
| try (FramedLZ4CompressorInputStream sis = new FramedLZ4CompressorInputStream( |
| new ByteArrayInputStream(bos.toByteArray()))) { |
| byte[] actual = IOUtils.toByteArray(sis); |
| Assert.assertArrayEquals(expected, actual); |
| } |
| |
| System.err.println(outputSz.getName() + " read after " + (System.currentTimeMillis() - start) + "ms"); |
| } |
| |
| // should yield decent compression |
| @Test |
| public void blaTarRoundtrip() throws IOException { |
| roundTripTest("bla.tar"); |
| } |
| |
| // yields no compression at all |
| @Test |
| public void gzippedLoremIpsumRoundtrip() throws IOException { |
| roundTripTest("lorem-ipsum.txt.gz"); |
| } |
| |
| @Test |
| public void biggerFileRoundtrip() throws IOException { |
| roundTripTest("COMPRESS-256.7z"); |
| } |
| |
| } |