X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;ds=sidebyside;f=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FCryptUtilsTest.java;fp=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FCryptUtilsTest.java;h=0c53461ea56aed8c5959e4b834ac90306994c8f4;hb=d46b7b96f94e88a776bcd2dfd756549ffb300cc9;hp=0000000000000000000000000000000000000000;hpb=c9994f27667bc421bcd448d39e55774fddf5c431;p=fanfix.git

diff --git a/src/be/nikiroo/utils/test_code/CryptUtilsTest.java b/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
new file mode 100644
index 0000000..0c53461
--- /dev/null
+++ b/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
@@ -0,0 +1,155 @@
+package be.nikiroo.utils.test_code;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+
+import be.nikiroo.utils.CryptUtils;
+import be.nikiroo.utils.IOUtils;
+import be.nikiroo.utils.test.TestCase;
+import be.nikiroo.utils.test.TestLauncher;
+
+class CryptUtilsTest extends TestLauncher {
+	private String key;
+	private CryptUtils crypt;
+
+	public CryptUtilsTest(String[] args) {
+		super("CryptUtils test", args);
+
+		String longKey = "some long string with more than 128 bits (=32 bytes) of data";
+
+		addSeries(new CryptUtilsTest(args, "Manual input wuth NULL key", null,
+				1));
+		addSeries(new CryptUtilsTest(args, "Streams with NULL key", null, true));
+
+		addSeries(new CryptUtilsTest(args, "Manual input with emptykey", "", 1));
+		addSeries(new CryptUtilsTest(args, "Streams with empty key", "", true));
+
+		addSeries(new CryptUtilsTest(args, "Manual input with long key",
+				longKey, 1));
+		addSeries(new CryptUtilsTest(args, "Streams with long key", longKey,
+				true));
+	}
+
+	@Override
+	protected void addTest(final TestCase test) {
+		super.addTest(new TestCase(test.getName()) {
+			@Override
+			public void test() throws Exception {
+				test.test();
+			}
+
+			@Override
+			public void setUp() throws Exception {
+				crypt = new CryptUtils(key);
+				test.setUp();
+			}
+
+			@Override
+			public void tearDown() throws Exception {
+				test.tearDown();
+				crypt = null;
+			}
+		});
+	}
+
+	private CryptUtilsTest(String[] args, String title, String key,
+			@SuppressWarnings("unused") int dummy) {
+		super(title, args);
+		this.key = key;
+
+		final String longData = "Le premier jour, Le Grand Barbu dans le cloud fit la lumière, et il vit que c'était bien. Ou quelque chose comme ça. Je préfère la Science-Fiction en général, je trouve ça plus sain :/";
+
+		addTest(new TestCase("Short") {
+			@Override
+			public void test() throws Exception {
+				String orig = "data";
+				byte[] encrypted = crypt.encrypt(orig);
+				String decrypted = crypt.decrypts(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+
+		addTest(new TestCase("Short, base64") {
+			@Override
+			public void test() throws Exception {
+				String orig = "data";
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+
+		addTest(new TestCase("Empty") {
+			@Override
+			public void test() throws Exception {
+				String orig = "";
+				byte[] encrypted = crypt.encrypt(orig);
+				String decrypted = crypt.decrypts(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+
+		addTest(new TestCase("Empty, base64") {
+			@Override
+			public void test() throws Exception {
+				String orig = "";
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+
+		addTest(new TestCase("Long") {
+			@Override
+			public void test() throws Exception {
+				String orig = longData;
+				byte[] encrypted = crypt.encrypt(orig);
+				String decrypted = crypt.decrypts(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+
+		addTest(new TestCase("Long, base64") {
+			@Override
+			public void test() throws Exception {
+				String orig = longData;
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
+
+				assertEquals(orig, decrypted);
+			}
+		});
+	}
+
+	private CryptUtilsTest(String[] args, String title, String key,
+			@SuppressWarnings("unused") boolean dummy) {
+		super(title, args);
+		this.key = key;
+
+		addTest(new TestCase("Simple test") {
+			@Override
+			public void test() throws Exception {
+				InputStream in = new ByteArrayInputStream(new byte[] { 42, 127,
+						12 });
+				crypt.encrypt(in);
+				ByteArrayOutputStream out = new ByteArrayOutputStream();
+				IOUtils.write(in, out);
+				byte[] result = out.toByteArray();
+
+				assertEquals(
+						"We wrote 3 bytes, we expected 3 bytes back but got: "
+								+ result.length, result.length, result.length);
+
+				assertEquals(42, result[0]);
+				assertEquals(127, result[1]);
+				assertEquals(12, result[2]);
+			}
+		});
+	}
+}