StreamUtilTest.java

package liquibase.util;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.StringReader;

import org.junit.Test;

public class StreamUtilTest {

	@Test
	public void testGetStreamContents() throws IOException {
		byte[] contents = "TEST2".getBytes();
		ByteArrayInputStream stream = new ByteArrayInputStream(contents);
		String result = StreamUtil.getStreamContents(stream);
		assertEquals("TEST2", result);
	}

	@Test
	public void testGetReaderContents() throws IOException {
		String contents = "TEST";
		StringReader reader = new StringReader(contents);
		String result = StreamUtil.getReaderContents(reader);
		assertEquals(contents, result);
	}

	@Test
	public void testWithBomNoEncodingGiven() throws IOException {
		String contents = "abc";
		ByteArrayInputStream bais = new ByteArrayInputStream(new byte[] {
				(byte) 0xEF, (byte) 0xBB, (byte) 0xBF, 0x61, 0x62, 0x63 });

		assertEquals(contents, StreamUtil.getStreamContents(bais));
	}

	@Test
	public void testWithBomCorrectEncodingGiven() throws IOException {
		String contents = "abc";
		ByteArrayInputStream bais = new ByteArrayInputStream(new byte[] {
				(byte) 0xEF, (byte) 0xBB, (byte) 0xBF, 0x61, 0x62, 0x63 });

		assertEquals(contents, StreamUtil.getStreamContents(bais, "UTF8"));
	}

	@Test
	public void testWithoutBomUtf8() throws IOException {
		String contents = "abc";
		ByteArrayInputStream bais = new ByteArrayInputStream(new byte[] { 0x61,
				0x62, 0x63 });

		assertEquals(contents, StreamUtil.getStreamContents(bais, "UTF8"));
	}

	@Test
	public void testWithoutBomLatin1() throws IOException {
		String contents = "abc";
		ByteArrayInputStream bais = new ByteArrayInputStream(new byte[] { 0x61,
				0x62, 0x63 });

		assertEquals(contents, StreamUtil.getStreamContents(bais, "Latin1"));
	}
	
	@Test
	public void testWithBomWrongEncodingGiven() throws IOException {
		ByteArrayInputStream bais = new ByteArrayInputStream(new byte[] {
				(byte) 0xEF, (byte) 0xBB, (byte) 0xBF, 0x61, 0x62, 0x63 });

		try {
			StreamUtil.getStreamContents(bais, "UTF-16BE");
			fail("Should have thrown an IllegalArgumentException");
		} catch (IllegalArgumentException e) {
			assertTrue(e.getMessage().contains("UTF-16BE"));
		}
	}
	
	@Test
	public void testContentLength() throws IOException {
                // These tests fails if windows environment where LF is replaced with CR LF
                // This test in quite silly.  Testing resource copy from src to target?
		InputStream in = getClass().getResourceAsStream("/liquibase/util/unicode-file.txt");
                long reallen=StreamUtil.getContentLength(in);
		assertTrue((reallen==50L) || (reallen==51L));
		
		in = getClass().getResourceAsStream("/liquibase/util/unicode-file.txt");
                long reallen2 = StreamUtil.getContentLength(new InputStreamReader(in, "UTF-8"));
		assertTrue((reallen2==39L) || (reallen2==40L));
	}
}