1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.shiro.crypto;
20  
21  import org.apache.shiro.codec.CodecSupport;
22  import org.apache.shiro.util.ByteSource;
23  import org.junit.Test;
24  
25  import java.io.ByteArrayInputStream;
26  import java.io.ByteArrayOutputStream;
27  import java.io.InputStream;
28  import java.util.Arrays;
29  
30  import static junit.framework.Assert.assertTrue;
31  
32  /**
33   * Test class for the AesCipherService class.
34   *
35   * @since 1.0
36   */
37  public class AesCipherServiceTest {
38  
39      private static final String[] PLAINTEXTS = new String[]{
40              "Hello, this is a test.",
41              "Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."
42      };
43  
44      @Test
45      public void testBlockOperations() {
46          AesCipherService aes = new AesCipherService();
47  
48          byte[] key = aes.generateNewKey().getEncoded();
49  
50          for (String plain : PLAINTEXTS) {
51              byte[] plaintext = CodecSupport.toBytes(plain);
52              ByteSource ciphertext = aes.encrypt(plaintext, key);
53              ByteSource decrypted = aes.decrypt(ciphertext.getBytes(), key);
54              assertTrue(Arrays.equals(plaintext, decrypted.getBytes()));
55          }
56      }
57  
58      @Test
59      public void testStreamingOperations() {
60  
61          AesCipherService cipher = new AesCipherService();
62          byte[] key = cipher.generateNewKey().getEncoded();
63  
64          for (String plain : PLAINTEXTS) {
65              byte[] plaintext = CodecSupport.toBytes(plain);
66              InputStream plainIn = new ByteArrayInputStream(plaintext);
67              ByteArrayOutputStream cipherOut = new ByteArrayOutputStream();
68              cipher.encrypt(plainIn, cipherOut, key);
69  
70              byte[] ciphertext = cipherOut.toByteArray();
71              InputStream cipherIn = new ByteArrayInputStream(ciphertext);
72              ByteArrayOutputStream plainOut = new ByteArrayOutputStream();
73              cipher.decrypt(cipherIn, plainOut, key);
74  
75              byte[] decrypted = plainOut.toByteArray();
76              assertTrue(Arrays.equals(plaintext, decrypted));
77          }
78      }
79  }