X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Ftest_code%2FCryptUtilsTest.java;h=61e18fbb49198ec36240d361a9d16d13a44af7c3;hb=6fd1a061ff06f14552deaf0e20aa21cee07a396d;hp=597650466ef1d1c024d9860b77c52a17317f5e1e;hpb=0747c3c2d2c65e1e063884d6c074f24db93a0c33;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..61e18fb 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;
 
@@ -127,9 +132,21 @@ 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.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]);
 			}
 		});
 	}