/**
* Copyright (C) 2009 eXo Platform SAS.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.exoplatform.commons.utils;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.concurrent.atomic.AtomicBoolean;
import org.exoplatform.component.test.AbstractGateInTest;
/**
* @author <a href="mailto:julien.viet@exoplatform.com">Julien Viet</a>
* @version $Revision$
*/
public class TestBufferingOutputStream extends AbstractGateInTest {
private ByteArrayOutputStream bytes = new ByteArrayOutputStream();
private BufferingOutputStream out = new BufferingOutputStream(bytes, 5);
@Override
protected void setUp() throws Exception {
bytes = new ByteArrayOutputStream();
out = new BufferingOutputStream(bytes, 5);
}
@Override
protected void tearDown() throws Exception {
bytes = null;
out = null;
}
public void testCtorIAE() {
try {
new BufferingOutputStream(bytes, 0);
} catch (IllegalArgumentException e) {
}
try {
new BufferingOutputStream(bytes, -1);
} catch (IllegalArgumentException e) {
}
}
public void testCtorNPE() {
try {
new BufferingOutputStream(null, 1);
} catch (NullPointerException e) {
}
}
public void testStreamClose() throws Exception {
final AtomicBoolean closed = new AtomicBoolean(false);
BufferingOutputStream out = new BufferingOutputStream(new ByteArrayOutputStream() {
@Override
public void close() throws IOException {
closed.set(true);
super.close();
}
}, 1);
out.close();
assertTrue(closed.get());
try {
out.write(0);
} catch (IOException ignore) {
}
try {
out.write(new byte[] { 0 });
} catch (IOException ignore) {
}
try {
out.flush();
} catch (IOException ignore) {
}
try {
out.close();
} catch (IOException ignore) {
}
}
public void testFlush() throws Exception {
out.write(0);
assertBytes();
out.flush();
assertBytes(0);
out.write(new byte[] { 1, 2, 3, 4 });
assertBytes();
out.close();
assertBytes(1, 2, 3, 4);
}
public void testWriteByte() throws Exception {
out.write(0);
assertBytes();
out.close();
assertBytes(0);
}
public void testAlmostFill() throws Exception {
out.write(new byte[] { 0, 1, 2, 3 });
assertBytes();
out.close();
assertBytes(0, 1, 2, 3);
}
public void testFill() throws Exception {
out.write(new byte[] { 0, 1, 2, 3, 4 });
assertBytes(0, 1, 2, 3, 4);
out.close();
assertBytes();
}
public void testBufferOverflowWithByte() throws Exception {
out.write(new byte[] { 0, 1, 2, 3 });
assertBytes();
out.write(4);
assertBytes();
out.write(5);
assertBytes(0, 1, 2, 3, 4);
out.close();
assertBytes(5);
}
public void testBufferOverflowWithArray() throws Exception {
out.write(new byte[] { 0, 1, 2, 3 });
assertBytes();
out.write(new byte[] { 4 });
assertBytes(0, 1, 2, 3);
out.close();
assertBytes(4);
}
public void testBufferOverflowWithLongArray() throws Exception {
out.write(new byte[] { 0, 1, 2, 3 });
assertBytes();
out.write(new byte[] { 4, 5, 6, 7, 8, 9 });
assertBytes(0, 1, 2, 3, 4, 5, 6, 7, 8);
out.close();
assertBytes(9);
}
public void testBufferOverflowWithVeryLongArray() throws Exception {
out.write(new byte[] { 0, 1, 2, 3 });
assertBytes();
out.write(new byte[] { 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14 });
assertBytes(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13);
out.close();
assertBytes(14);
}
private void assertBytes(int... expectedBytes) {
int len = expectedBytes.length;
assertEquals(len, bytes.size());
if (len > 0) {
byte[] actualBytes = bytes.toByteArray();
for (int i = 0; i < len; i++) {
int expectedByte = expectedBytes[i];
byte actualByte = actualBytes[i];
assertEquals("Was expecting byte at index " + i + " to be equals to " + expectedByte + " instead of "
+ actualByte, expectedByte, actualByte);
}
bytes.reset();
}
}
}