Add 'src/be/nikiroo/utils/' from commit '46add0670fdee4bd936a13fe2448c5e20a7ffd0a'

[nikiroo-utils.git] / src / be / nikiroo / utils / test_code / CryptUtilsTest.java

diff --git a/src/be/nikiroo/utils/test_code/CryptUtilsTest.java b/src/be/nikiroo/utils/test_code/CryptUtilsTest.java
index 597650466ef1d1c024d9860b77c52a17317f5e1e..0c53461ea56aed8c5959e4b834ac90306994c8f4 100644 (file)
--- 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]);
                        }
                });
        }