@Test public void testOneByte() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final byte test = (byte) new TestRng(getName()).nextInt(); try { b.write(test); b.close(); assertEquals(1, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(1, r.length); assertEquals(test, r[0]); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(1, r.length); assertEquals(test, r[0]); } } finally { b.destroy(); } }
@Test public void testInCoreLimit_SwitchOnCopy() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final byte[] test = new TestRng(getName()).nextBytes(TemporaryBuffer.DEFAULT_IN_CORE_LIMIT * 2); try { final ByteArrayInputStream in = new ByteArrayInputStream( test, TemporaryBuffer.DEFAULT_IN_CORE_LIMIT, test.length - TemporaryBuffer.DEFAULT_IN_CORE_LIMIT); b.write(test, 0, TemporaryBuffer.DEFAULT_IN_CORE_LIMIT); b.copy(in); b.close(); assertEquals(test.length, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(test.length, r.length); assertArrayEquals(test, r); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(test.length, r.length); assertArrayEquals(test, r); } } finally { b.destroy(); } }
@Test public void testLarge_SingleWrite() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final byte[] test = new TestRng(getName()).nextBytes(TemporaryBuffer.DEFAULT_IN_CORE_LIMIT * 3); try { b.write(test); b.close(); assertEquals(test.length, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(test.length, r.length); assertArrayEquals(test, r); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(test.length, r.length); assertArrayEquals(test, r); } } finally { b.destroy(); } }
@Test public void testOneBlockAndHalf_Copy() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final byte[] test = new TestRng(getName()).nextBytes(TemporaryBuffer.Block.SZ * 3 / 2); try { final ByteArrayInputStream in = new ByteArrayInputStream(test); b.write(in.read()); b.copy(in); b.close(); assertEquals(test.length, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(test.length, r.length); assertArrayEquals(test, r); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(test.length, r.length); assertArrayEquals(test, r); } } finally { b.destroy(); } }
@Test public void testOneBlock_BulkWrite() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final byte[] test = new TestRng(getName()).nextBytes(TemporaryBuffer.Block.SZ); try { b.write(test, 0, 2); b.write(test, 2, 4); b.write(test, 6, test.length - 6 - 2); b.write(test, test.length - 2, 2); b.close(); assertEquals(test.length, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(test.length, r.length); assertArrayEquals(test, r); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(test.length, r.length); assertArrayEquals(test, r); } } finally { b.destroy(); } }
@Test public void testDestroyWhileOpen() throws IOException { @SuppressWarnings("resource" /* java 7 */) final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); try { b.write(new TestRng(getName()).nextBytes(TemporaryBuffer.DEFAULT_IN_CORE_LIMIT * 2)); } finally { b.destroy(); } }
@Test public void testEmpty() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); try { b.close(); assertEquals(0, b.length()); final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(0, r.length); } finally { b.destroy(); } }
@Test public void testRandomWrites() throws IOException { final TemporaryBuffer b = new TemporaryBuffer.LocalFile(null); final TestRng rng = new TestRng(getName()); final int max = TemporaryBuffer.DEFAULT_IN_CORE_LIMIT * 2; final byte[] expect = new byte[max]; try { int written = 0; boolean onebyte = true; while (written < max) { if (onebyte) { final byte v = (byte) rng.nextInt(); b.write(v); expect[written++] = v; } else { final int len = Math.min(rng.nextInt() & 127, max - written); final byte[] tmp = rng.nextBytes(len); b.write(tmp, 0, len); System.arraycopy(tmp, 0, expect, written, len); written += len; } onebyte = !onebyte; } assertEquals(expect.length, written); b.close(); assertEquals(expect.length, b.length()); { final byte[] r = b.toByteArray(); assertNotNull(r); assertEquals(expect.length, r.length); assertArrayEquals(expect, r); } { final ByteArrayOutputStream o = new ByteArrayOutputStream(); b.writeTo(o, null); o.close(); final byte[] r = o.toByteArray(); assertEquals(expect.length, r.length); assertArrayEquals(expect, r); } } finally { b.destroy(); } }