X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FReplaceInputStreamTest.java;fp=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FReplaceInputStreamTest.java;h=e6e211215430485cfd0938b5fcb89f6fc5ee4a5e;hb=5584adbbbf5444c0039fed2b35dc7d5bb57b71b1;hp=0000000000000000000000000000000000000000;hpb=ad207feb2815e429ae32484bc6930990099f8ea4;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java b/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java
new file mode 100644
index 0000000..e6e2112
--- /dev/null
+++ b/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java
@@ -0,0 +1,106 @@
+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]);
+		}
+	}
+}