@Test
public void testWriteArrayManySlabs() throws Throwable {
CapacityByteArrayOutputStream capacityByteArrayOutputStream =
new CapacityByteArrayOutputStream(10);
int it = 500;
int v = 23;
for (int j = 0; j < it; j++) {
for (int i = 0; i < v; i++) {
byte[] toWrite = {(byte) (i * 3), (byte) (i * 3 + 1), (byte) (i * 3 + 2)};
capacityByteArrayOutputStream.write(toWrite);
assertEquals((i + 1) * 3 + v * 3 * j, capacityByteArrayOutputStream.size());
}
}
byte[] byteArray = BytesInput.from(capacityByteArrayOutputStream).toByteArray();
assertEquals(v * 3 * it, byteArray.length);
for (int i = 0; i < v * 3 * it; i++) {
assertEquals(i % (v * 3), byteArray[i]);
}
// verifying we have not created 500 * 23 / 10 slabs
assertTrue(
"slab count: " + capacityByteArrayOutputStream.getSlabCount(),
capacityByteArrayOutputStream.getSlabCount() <= 20);
capacityByteArrayOutputStream.reset();
writeArraysOf3(capacityByteArrayOutputStream, v);
validate(capacityByteArrayOutputStream, v * 3);
// verifying we use less slabs now
assertTrue(
"slab count: " + capacityByteArrayOutputStream.getSlabCount(),
capacityByteArrayOutputStream.getSlabCount() <= 2);
}
private void validate(
CapacityByteArrayOutputStream capacityByteArrayOutputStream, final int expectedSize)
throws IOException {
final byte[] byteArray = BytesInput.from(capacityByteArrayOutputStream).toByteArray();
assertEquals(expectedSize, byteArray.length);
for (int i = 0; i < expectedSize; i++) {
assertEquals(i, byteArray[i]);
}
}
@Test
public void testWriteArrayBiggerThanSlab() throws Throwable {
CapacityByteArrayOutputStream capacityByteArrayOutputStream =
new CapacityByteArrayOutputStream(10);
int v = 23;
writeArraysOf3(capacityByteArrayOutputStream, v);
int n = v * 3;
byte[] toWrite = { // bigger than 2 slabs of size of 10
(byte) n,
(byte) (n + 1),
(byte) (n + 2),
(byte) (n + 3),
(byte) (n + 4),
(byte) (n + 5),
(byte) (n + 6),
(byte) (n + 7),
(byte) (n + 8),
(byte) (n + 9),
(byte) (n + 10),
(byte) (n + 11),
(byte) (n + 12),
(byte) (n + 13),
(byte) (n + 14),
(byte) (n + 15),
(byte) (n + 16),
(byte) (n + 17),
(byte) (n + 18),
(byte) (n + 19),
(byte) (n + 20)
};
capacityByteArrayOutputStream.write(toWrite);
n = n + toWrite.length;
assertEquals(n, capacityByteArrayOutputStream.size());
validate(capacityByteArrayOutputStream, n);
capacityByteArrayOutputStream.reset();
// check it works after reset too
capacityByteArrayOutputStream.write(toWrite);
assertEquals(toWrite.length, capacityByteArrayOutputStream.size());
byte[] byteArray = BytesInput.from(capacityByteArrayOutputStream).toByteArray();
assertEquals(toWrite.length, byteArray.length);
for (int i = 0; i < toWrite.length; i++) {
assertEquals(toWrite[i], byteArray[i]);
}
}
@Test
public void testReset() throws Throwable {
CapacityByteArrayOutputStream capacityByteArrayOutputStream =
new CapacityByteArrayOutputStream(10);
for (int i = 0; i < 54; i++) {
capacityByteArrayOutputStream.write(i);
assertEquals(i + 1, capacityByteArrayOutputStream.size());
}
capacityByteArrayOutputStream.reset();
for (int i = 0; i < 54; i++) {
capacityByteArrayOutputStream.write(54 + i);
assertEquals(i + 1, capacityByteArrayOutputStream.size());
}
final byte[] byteArray = BytesInput.from(capacityByteArrayOutputStream).toByteArray();
assertEquals(54, byteArray.length);
for (int i = 0; i < 54; i++) {
assertEquals(i + " in " + Arrays.toString(byteArray), 54 + i, byteArray[i]);
}
}
