cadi/core/src/test/java/org/onap/aaf/cadi/test/JU_Symm.java

   1 /*******************************************************************************
   2  * ============LICENSE_START====================================================
   3  * * org.onap.aaf
   4  * * ===========================================================================
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * * ===========================================================================
   7  * * Licensed under the Apache License, Version 2.0 (the "License");
   8  * * you may not use this file except in compliance with the License.
   9  * * You may obtain a copy of the License at
  10  * *
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0
  12  * *
  13  *  * Unless required by applicable law or agreed to in writing, software
  14  * * distributed under the License is distributed on an "AS IS" BASIS,
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * * See the License for the specific language governing permissions and
  17  * * limitations under the License.
  18  * * ============LICENSE_END====================================================
  19  * *
  20  * *
  21  ******************************************************************************/
  22 package org.onap.aaf.cadi.test;
  23
  24 import static org.hamcrest.CoreMatchers.*;
  25 import static org.junit.Assert.*;
  26 import static org.mockito.Mockito.*;
  27 import java.lang.reflect.*;
  28 import org.junit.*;
  29
  30 import java.io.ByteArrayInputStream;
  31 import java.io.ByteArrayOutputStream;
  32 import java.io.PrintStream;
  33 import java.util.Arrays;
  34
  35 import org.onap.aaf.cadi.CadiException;
  36 import org.onap.aaf.cadi.PropAccess;
  37 import org.onap.aaf.cadi.Symm;
  38
  39 public class JU_Symm {
  40         private Symm defaultSymm;
  41
  42         private ByteArrayOutputStream outStream;
  43
  44         @Before
  45         public void setup() throws Exception {
  46                 defaultSymm = new Symm(
  47                                 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/".toCharArray()
  48                                 ,76, "Use default!" ,true);
  49                 outStream = new ByteArrayOutputStream();
  50                 System.setOut(new PrintStream(outStream));
  51         }
  52
  53         @After
  54         public void tearDown() {
  55                 System.setOut(System.out);
  56         }
  57
  58         @Test
  59         public void constructorTest() throws Exception {
  60                 Symm myCustomSymm = new Symm(
  61                         "ACEGIKMOQSUWYacegikmoqsuwy02468+/".toCharArray(), 76, "Default", true);
  62                 Field convert_field = Symm.class.getDeclaredField("convert");
  63                 convert_field.setAccessible(true);
  64
  65                 Class<?> Unordered_class = Class.forName("org.onap.aaf.cadi.Symm$Unordered");
  66                 assertThat(convert_field.get(myCustomSymm), instanceOf(Unordered_class));
  67         }
  68
  69         @SuppressWarnings("unused")
  70         @Test
  71         public void copyTest() throws Exception {
  72                 Symm copy = Symm.base64.copy(76);
  73         }
  74
  75         @SuppressWarnings("deprecation")
  76         @Test
  77         public void deprecatedTest() {
  78                 assertEquals(Symm.base64(), Symm.base64);
  79                 assertEquals(Symm.base64noSplit(), Symm.base64noSplit);
  80                 assertEquals(Symm.base64url(), Symm.base64url);
  81                 assertEquals(Symm.baseCrypt(), Symm.encrypt);
  82         }
  83
  84         @Test
  85         public void encodeDecodeStringTest() throws Exception {
  86                 String orig = "hello";
  87                 String b64encrypted = Symm.base64.encode(orig);
  88                 assertEquals(Symm.base64.decode(b64encrypted), orig);
  89
  90                 String defaultEnrypted = defaultSymm.encode(orig);
  91                 assertEquals(defaultSymm.decode(defaultEnrypted), orig);
  92         }
  93
  94         @Test
  95         public void encodeDecodeByteArrayTest() throws Exception {
  96                 String orig = "hello";
  97                 byte[] b64encrypted = Symm.base64.encode(orig.getBytes());
  98                 assertEquals(new String(Symm.base64.decode(b64encrypted)), orig);
  99
 100                 byte[] empty = null;
 101                 assertTrue(Arrays.equals(Symm.base64.encode(empty), new byte[0]));
 102         }
 103
 104         @Test
 105         public void encodeDecodeStringToStreamTest() throws Exception {
 106                 String orig = "I'm a password, really";
 107                 String b64encrypted;
 108                 String output;
 109
 110                 ByteArrayOutputStream baosEncrypt = new ByteArrayOutputStream();
 111                 Symm.base64.encode(orig, baosEncrypt);
 112                 b64encrypted = new String(baosEncrypt.toByteArray());
 113
 114                 ByteArrayOutputStream baosDecrypt = new ByteArrayOutputStream();
 115                 Symm.base64.decode(b64encrypted, baosDecrypt);
 116                 output = new String(baosDecrypt.toByteArray());
 117
 118                 assertEquals(orig, output);
 119         }
 120
 121         @Test
 122         public void encryptDecryptStreamWithPrefixTest() throws Exception {
 123                 String orig = "I'm a password, really";
 124                 byte[] b64encrypted;
 125                 String output;
 126
 127                 byte[] prefix = "enc:".getBytes();
 128
 129                 ByteArrayInputStream baisEncrypt = new ByteArrayInputStream(orig.getBytes());
 130                 ByteArrayOutputStream baosEncrypt = new ByteArrayOutputStream();
 131                 Symm.base64.encode(baisEncrypt, baosEncrypt, prefix);
 132
 133                 b64encrypted = baosEncrypt.toByteArray();
 134
 135                 ByteArrayInputStream baisDecrypt = new ByteArrayInputStream(b64encrypted);
 136                 ByteArrayOutputStream baosDecrypt = new ByteArrayOutputStream();
 137                 Symm.base64.decode(baisDecrypt, baosDecrypt, prefix.length);
 138
 139                 output = new String(baosDecrypt.toByteArray());
 140                 assertEquals(orig, output);
 141         }
 142
 143         @Test
 144         public void randomGenTest() {
 145                 // Ian - There really isn't a great way to test for randomness...
 146                 String prev = null;
 147                 for (int i = 0; i < 10; i++) {
 148                         String current = Symm.randomGen(100);
 149                         if (current.equals(prev)) {
 150                                 fail("I don't know how, but you generated the exact same random string twice in a row");
 151                         }
 152                         prev = current;
 153                 }
 154                 assertTrue(true);
 155         }
 156
 157         @Test
 158         public void obtainTest() throws Exception {
 159                 Symm symm = Symm.base64.obtain();
 160
 161                 String orig ="Another Password, please";
 162                 String encrypted = symm.enpass(orig);
 163                 String decrypted = symm.depass(encrypted);
 164                 assertEquals(orig, decrypted);
 165         }
 166
 167         @Test
 168         public void InputStreamObtainTest() throws Exception {
 169                 byte[] keygen = Symm.keygen();
 170
 171                 Symm symm = Symm.obtain(new ByteArrayInputStream(keygen));
 172
 173                 String orig ="Another Password, please";
 174                 String encrypted = symm.enpass(orig);
 175                 String decrypted = symm.depass(encrypted);
 176                 assertEquals(orig, decrypted);
 177         }
 178
 179         @Test
 180         public void StringObtainTest() throws Exception {
 181                 byte[] keygen = Symm.keygen();
 182
 183                 Symm symm = Symm.obtain(new String(keygen));
 184
 185                 String orig ="Another Password, please";
 186                 String encrypted = symm.enpass(orig);
 187                 String decrypted = symm.depass(encrypted);
 188                 assertEquals(orig, decrypted);
 189         }
 190
 191         @Test
 192         public void AccessObtainTest() throws Exception {
 193                 PropAccess pa = new PropAccess("cadi_keyfile=src/test/resources/keyfile");
 194                 Symm symm = Symm.obtain(pa);
 195                 String orig ="Another Password, please";
 196                 String encrypted = symm.enpass(orig);
 197                 String decrypted = symm.depass(encrypted);
 198                 assertEquals(orig, decrypted);
 199
 200                 try {
 201                         PropAccess badPa = mock(PropAccess.class);
 202                         when(badPa.getProperty("cadi_keyfile", null)).thenReturn("not_a_real_file.txt");
 203                         symm = Symm.obtain(badPa);
 204                         fail("Should have thrown an exception");
 205                 } catch (CadiException e) {
 206                         assertTrue(e.getMessage().contains("ERROR: "));
 207                         assertTrue(e.getMessage().contains("not_a_real_file.txt"));
 208                         assertTrue(e.getMessage().contains(" does not exist!"));
 209                 }
 210         }
 211
 212 }