package be.nikiroo.utils.test_code; import java.io.ByteArrayInputStream; import java.io.InputStream; import be.nikiroo.utils.IOUtils; import be.nikiroo.utils.streams.ReplaceInputStream; import be.nikiroo.utils.test.TestCase; import be.nikiroo.utils.test.TestLauncher; class ReplaceInputStreamTest extends TestLauncher { public ReplaceInputStreamTest(String[] args) { super("ReplaceInputStream test", args); addTest(new TestCase("Empty replace") { @Override public void test() throws Exception { byte[] data = new byte[] { 42, 12, 0, 127 }; ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), new byte[0], new byte[0]); checkArrays(this, "FIRST", in, data); } }); addTest(new TestCase("Simple replace") { @Override public void test() throws Exception { byte[] data = new byte[] { 42, 12, 0, 127 }; ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), new byte[] { 0 }, new byte[] { 10 }); checkArrays(this, "FIRST", in, new byte[] { 42, 12, 10, 127 }); } }); addTest(new TestCase("3/4 replace") { @Override public void test() throws Exception { byte[] data = new byte[] { 42, 12, 0, 127 }; ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), new byte[] { 12, 0, 127 }, new byte[] { 10, 10, 10 }); checkArrays(this, "FIRST", in, new byte[] { 42, 10, 10, 10 }); } }); addTest(new TestCase("Lnger replace") { @Override public void test() throws Exception { byte[] data = new byte[] { 42, 12, 0, 127 }; ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), new byte[] { 0 }, new byte[] { 10, 10, 10 }); checkArrays(this, "FIRST", in, new byte[] { 42, 12, 10, 10, 10, 127 }); } }); addTest(new TestCase("Shorter replace") { @Override public void test() throws Exception { byte[] data = new byte[] { 42, 12, 0, 127 }; ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), new byte[] { 42, 12, 0 }, new byte[] { 10 }); checkArrays(this, "FIRST", in, new byte[] { 10, 127 }); } }); addTest(new TestCase("String replace") { @Override public void test() throws Exception { byte[] data = "I like red".getBytes("UTF-8"); ReplaceInputStream in = new ReplaceInputStream( new ByteArrayInputStream(data), "red".getBytes("UTF-8"), "blue".getBytes("UTF-8")); checkArrays(this, "FIRST", in, "I like blue".getBytes("UTF-8")); data = "I like blue".getBytes("UTF-8"); in = new ReplaceInputStream(new ByteArrayInputStream(data), "blue".getBytes("UTF-8"), "red".getBytes("UTF-8")); checkArrays(this, "FIRST", in, "I like red".getBytes("UTF-8")); } }); } static void checkArrays(TestCase test, String prefix, InputStream in, byte[] expected) throws Exception { byte[] actual = IOUtils.toByteArray(in); test.assertEquals("The " + prefix + " resulting array has not the correct number of items", expected.length, actual.length); for (int i = 0; i < actual.length; i++) { test.assertEquals("Item " + i + " (0-based) is not the same", expected[i], actual[i]); } } }