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, false); String decrypted = crypt.decrypt64s(encrypted, false); 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, false); String decrypted = crypt.decrypt64s(encrypted, false); 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, false); String decrypted = crypt.decrypt64s(encrypted, false); 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.encryptInputStream(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]); } }); } }