From: Niki Roo <niki@nikiroo.be>
Date: Tue, 23 Apr 2019 12:36:13 +0000 (+0200)
Subject: Test: add Stream CryptUtils
X-Git-Url: https://git.nikiroo.be/?a=commitdiff_plain;h=6fd1a061ff06f14552deaf0e20aa21cee07a396d;p=fanfix-jexer.git

Test: add Stream CryptUtils
---

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]);
 			}
 		});
 	}