X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FReplaceInputStreamTest.java;h=efab8c73d62e5f79779a41b6f8c4c889ad8a8f74;hb=68232a49b2f63e195dfaf1d907bd20c48cf2b394;hp=d6a741fd7f85b1d5b34c036fdafb3e4a76101a47;hpb=6ef54b36380b9d123bbdb6b4056cf64b7489225e;p=nikiroo-utils.git
diff --git a/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java b/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java
index d6a741f..efab8c7 100644
--- a/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java
+++ b/src/be/nikiroo/utils/test_code/ReplaceInputStreamTest.java
@@ -4,89 +4,201 @@ import java.io.ByteArrayInputStream;
import java.io.InputStream;
import be.nikiroo.utils.IOUtils;
-import be.nikiroo.utils.ReplaceInputStream;
+import be.nikiroo.utils.streams.ReplaceInputStream;
import be.nikiroo.utils.test.TestCase;
import be.nikiroo.utils.test.TestLauncher;
-public class ReplaceInputStreamTest extends TestLauncher {
+class ReplaceInputStreamTest extends TestLauncher {
public ReplaceInputStreamTest(String[] args) {
super("ReplaceInputStream test", args);
- addTest(new TestCase("Simple InputStream, empty replace") {
+ addTest(new TestCase("Empty replace") {
@Override
public void test() throws Exception {
- byte[] expected = new byte[] { 42, 12, 0, 127 };
+ byte[] data = new byte[] { 42, 12, 0, 127 };
ReplaceInputStream in = new ReplaceInputStream(
- new ByteArrayInputStream(expected), new byte[0],
+ new ByteArrayInputStream(data), new byte[0],
new byte[0]);
- checkArrays(this, "FIRST", in, expected);
+
+ checkArrays(this, "FIRST", in, data);
}
});
- addTest(new TestCase("Simple InputStream, simple replace") {
+ addTest(new TestCase("Simple replace") {
@Override
public void test() throws Exception {
- byte[] expected = new byte[] { 42, 12, 0, 127 };
+ byte[] data = new byte[] { 42, 12, 0, 127 };
ReplaceInputStream in = new ReplaceInputStream(
- new ByteArrayInputStream(expected), new byte[] { 0 },
+ new ByteArrayInputStream(data), new byte[] { 0 },
new byte[] { 10 });
checkArrays(this, "FIRST", in, new byte[] { 42, 12, 10, 127 });
}
});
- addTest(new TestCase("Simple byte array reading, 3/4 replace") {
+ addTest(new TestCase("3/4 replace") {
@Override
public void test() throws Exception {
- byte[] expected = new byte[] { 42, 12, 0, 127 };
+ byte[] data = new byte[] { 42, 12, 0, 127 };
ReplaceInputStream in = new ReplaceInputStream(
- new ByteArrayInputStream(expected), new byte[] { 12, 0,
- 127 }, new byte[] { 10, 10, 10 });
+ 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("Simple byte array reading, longer replace") {
+ addTest(new TestCase("Longer replace") {
@Override
public void test() throws Exception {
- byte[] expected = new byte[] { 42, 12, 0, 127 };
+ byte[] data = new byte[] { 42, 12, 0, 127 };
ReplaceInputStream in = new ReplaceInputStream(
- new ByteArrayInputStream(expected), new byte[] { 0 },
+ new ByteArrayInputStream(data), new byte[] { 0 },
new byte[] { 10, 10, 10 });
- // TODO NOT OK!!
- System.out.println();
- for (int i = 0; i < expected.length; i++) {
- System.out.println("expected[" + i + "] = " + expected[i]);
- }
- byte[] actual = IOUtils.readSmallStream(in).getBytes("UTF-8");
- for (int i = 0; i < actual.length; i++) {
- System.out.println("actual[" + i + "] = " + actual[i]);
- }
- System.exit(1);
- //
-
checkArrays(this, "FIRST", in, new byte[] { 42, 12, 10, 10, 10,
127 });
}
});
- addTest(new TestCase("Simple byte array reading, shorter replace") {
+ addTest(new TestCase("Shorter replace") {
@Override
public void test() throws Exception {
- byte[] expected = new byte[] { 42, 12, 0, 127 };
+ byte[] data = new byte[] { 42, 12, 0, 127 };
ReplaceInputStream in = new ReplaceInputStream(
- new ByteArrayInputStream(expected), new byte[] { 42,
- 12, 0 }, new byte[] { 10 });
+ 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", "blue");
+
+ checkArrays(this, "FIRST", in, "I like blue".getBytes("UTF-8"));
+
+ data = "I like blue hammers".getBytes("UTF-8");
+ in = new ReplaceInputStream(new ByteArrayInputStream(data),
+ "blue", "red");
+
+ checkArrays(this, "SECOND", in, "I like red hammers".getBytes("UTF-8"));
+ }
+ });
+
+ addTest(new TestCase("Multiple replaces") {
+ @Override
+ public void test() throws Exception {
+ byte[] data = "I like red cabage".getBytes("UTF-8");
+ ReplaceInputStream in = new ReplaceInputStream(
+ new ByteArrayInputStream(data), //
+ new String[] { "red", "like" }, //
+ new String[] { "green", "very very much like" } //
+ );
+
+ String result = new String(IOUtils.toByteArray(in), "UTF-8");
+ assertEquals("I very very much like green cabage", result);
+ }
+ });
+
+ addTest(new TestCase("Multiple replaces") {
+ @Override
+ public void test() throws Exception {
+ String str= ("" //
+ + "\n" //
+ + "\n" //
+ + "\n" //
+ + "\n" //
+ + "\t\n" //
+ + "\t\n" //
+ + "\t${title}\n" //
+ + "\t\n" //
+ + "\t\n" //
+ + "\n" //
+ + "\n" //
+ + "\t\n" //
+ + "${banner}${content}\t
\n" //
+ + "\n" //
+ + "" //
+ );
+ byte[] data = str.getBytes("UTF-8");
+
+ String title = "Fanfix";
+ String banner = "Super banner v3
";
+ String content = "";
+
+ InputStream in = new ReplaceInputStream(
+ new ByteArrayInputStream(data), //
+ new String[] { "${title}", "${banner}", "${content}" }, //
+ new String[] { title, banner, content } //
+ );
+
+ String result = new String(IOUtils.toByteArray(in), "UTF-8");
+ assertEquals(str //
+ .replace("${title}", title) //
+ .replace("${banner}", banner) //
+ .replace("${content}", content) //
+ , result);
+ }
+ });
+
+
}
static void checkArrays(TestCase test, String prefix, InputStream in,
byte[] expected) throws Exception {
byte[] actual = IOUtils.toByteArray(in);
+
+// System.out.println("\nActual:");
+// for(byte byt : actual) {
+// System.out.print(byt+" ");
+// }
+// System.out.println("\nExpected:");
+// for(byte byt : expected) {
+// System.out.print(byt+" ");
+// }
+
test.assertEquals("The " + prefix
+ " resulting array has not the correct number of items",
expected.length, actual.length);