X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FCryptUtilsTest.java;h=0c53461ea56aed8c5959e4b834ac90306994c8f4;hb=505be508ae7d3fb48122be548b310a238cfb91eb;hp=597650466ef1d1c024d9860b77c52a17317f5e1e;hpb=f3502fbd11d357e1d33e2eb6fc6261df3cf26d09;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/test_code/CryptUtilsTest.java b/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
index 5976504..0c53461 100644
--- a/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
+++ b/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
@@ -1,6 +1,11 @@
 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;
 
@@ -70,8 +75,8 @@ class CryptUtilsTest extends TestLauncher {
 			@Override
 			public void test() throws Exception {
 				String orig = "data";
-				String encrypted = crypt.encrypt64(orig, false);
-				String decrypted = crypt.decrypt64s(encrypted, false);
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
 
 				assertEquals(orig, decrypted);
 			}
@@ -92,8 +97,8 @@ class CryptUtilsTest extends TestLauncher {
 			@Override
 			public void test() throws Exception {
 				String orig = "";
-				String encrypted = crypt.encrypt64(orig, false);
-				String decrypted = crypt.decrypt64s(encrypted, false);
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
 
 				assertEquals(orig, decrypted);
 			}
@@ -114,8 +119,8 @@ class CryptUtilsTest extends TestLauncher {
 			@Override
 			public void test() throws Exception {
 				String orig = longData;
-				String encrypted = crypt.encrypt64(orig, false);
-				String decrypted = crypt.decrypt64s(encrypted, false);
+				String encrypted = crypt.encrypt64(orig);
+				String decrypted = crypt.decrypt64s(encrypted);
 
 				assertEquals(orig, decrypted);
 			}
@@ -127,9 +132,23 @@ class CryptUtilsTest extends TestLauncher {
 		super(title, args);
 		this.key = key;
 
-		addTest(new TestCase("TODO: Make some tests with the Streams") {
+		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]);
 			}
 		});
 	}