@Test
public void testReadAndWrite() throws Exception {
byte[] fluff = constructFluff(132000);
byte[] comp = LZFEncoder.encode(fluff);
ByteArrayOutputStream bytes = new ByteArrayOutputStream(fluff.length);
LZFInputStream in = new LZFInputStream(new ByteArrayInputStream(comp));
in.readAndWrite(bytes);
in.close();
byte[] actual = bytes.toByteArray();
Assert.assertEquals(actual.length, fluff.length);
Assert.assertEquals(actual, fluff);
}
@Test
public void testAvailable() throws IOException {
ByteArrayInputStream bis = new ByteArrayInputStream(compressedBytes);
LZFInputStream is = new LZFInputStream(bis);
Assert.assertSame(is.getUnderlyingInputStream(), bis);
Assert.assertEquals(0, is.available());
// read one byte; should decode bunch more, make available
is.read();
int total = 1; // since we read one byte already
Assert.assertEquals(is.available(), 65534);
// and after we skip through all of it, end with -1 for EOF
long count;
while ((count = is.skip(16384L)) > 0L) {
total += (int) count;
}
// nothing more available; but we haven't yet closed so:
Assert.assertEquals(is.available(), 0);
// and then we close it:
is.close();
Assert.assertEquals(is.available(), 0);
Assert.assertEquals(total, compressableInputLength);
}
// Mostly for [Issue#19]
@Test
public void testLongSkips() throws Exception {
// 64k per block, 200k gives 3 full, one small
byte[] fluff = constructFluff(200000);
byte[] comp = LZFEncoder.encode(fluff);
// we get about 200k, maybe byte or two more, so:
final int LENGTH = fluff.length;
LZFInputStream in = new LZFInputStream(new ByteArrayInputStream(comp));
// read one byte for fun
Assert.assertEquals(in.read(), fluff[0] & 0xFF);
// then skip all but one
long amt = in.skip(LENGTH - 2);
Assert.assertEquals(amt, (long) (LENGTH - 2));
Assert.assertEquals(in.read(), fluff[LENGTH - 1] & 0xFF);
Assert.assertEquals(in.read(), -1);
in.close();
}
/**
* Test that creates a longer piece of content, compresses it, and reads back in arbitrary small
* reads.
*/
private void doTestIncremental(boolean fullReads) throws IOException {
// first need to compress something...
String[] words = new String[] {"what", "ever", "some", "other", "words", "too"};
StringBuilder sb = new StringBuilder(258000);
Random rnd = new Random(123);
while (sb.length() < 256000) {
int i = (rnd.nextInt() & 31);
if (i < words.length) {
sb.append(words[i]);
} else {
sb.append(i);
}
}
byte[] uncomp = sb.toString().getBytes("UTF-8");
ByteArrayOutputStream bytes = new ByteArrayOutputStream();
LZFOutputStream lzOut = new LZFOutputStream(bytes);
lzOut.write(uncomp);
lzOut.close();
byte[] comp = bytes.toByteArray();
// read back, in chunks
bytes = new ByteArrayOutputStream(uncomp.length);
byte[] buffer = new byte[500];
LZFInputStream lzIn = new LZFInputStream(new ByteArrayInputStream(comp), fullReads);
int pos = 0;
while (true) {
int len = 1 + ((rnd.nextInt() & 0x7FFFFFFF) % buffer.length);
int offset = buffer.length - len;
int count = lzIn.read(buffer, offset, len);
if (count < 0) {
break;
}
if (count > len) {
Assert.fail(
"Requested "
+ len
+ " bytes (offset "
+ offset
+ ", array length "
+ buffer.length
+ "), got "
+ count);
}
pos += count;
// with full reads, ought to get full results
if (count != len) {
if (fullReads) {
// Except at the end, with last incomplete chunk
if (pos != uncomp.length) {
Assert.fail(
"Got partial read (when requested full read!), position "
+ pos
+ " (of full "
+ uncomp.length
+ ")");
}
}
}
bytes.write(buffer, offset, count);
}
byte[] result = bytes.toByteArray();
Assert.assertEquals(result.length, uncomp.length);
Assert.assertEquals(result, uncomp);
}
