/*
* Copyright 2016 Huawei Technologies Co., Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.openo.baseservice.encrypt.cbb.impl;
import java.security.NoSuchAlgorithmException;
import java.security.spec.InvalidKeySpecException;
import javax.crypto.Cipher;
import javax.crypto.SecretKeyFactory;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.openo.baseservice.encrypt.cbb.CipherCreator;
import org.openo.baseservice.encrypt.cbb.inf.AbstractCipher;
import junit.framework.Assert;
import mockit.Mocked;
import mockit.NonStrictExpectations;
/**
*
*
*
*
* @author
* @version 02-Jun-2016
*/
public class AesCipherTest {
/**
*
*
* @throws java.lang.Exception
* @since
*/
@BeforeClass
public static void setUpBeforeClass() throws Exception {
CipherCreator.instance().setFactory(new AesCipherFactory());
}
/**
*
*
* @throws java.lang.Exception
* @since
*/
@AfterClass
public static void tearDownAfterClass() throws Exception {
}
/**
*
*
* @throws java.lang.Exception
* @since
*/
@Before
public void setUp() throws Exception {
}
/**
*
*
* @throws java.lang.Exception
* @since
*/
@After
public void tearDown() throws Exception {
}
/**
* Test method for
* {@link org.openo.baseservice.encrypt.cbb.impl.AesCipher#encrypt(java.lang.String)}.
*/
@Test
public void testEncrypt() {
final AbstractCipher cipherManager = CipherCreator.instance().create();
final String encrypted = cipherManager.encrypt("test-encrypt");
final String decrypted = cipherManager.decrypt(encrypted);
Assert.assertEquals("test-encrypt", decrypted);
}
@Test
public void testEncryptException() throws Exception {
new NonStrictExpectations() {
@Mocked
Cipher cipher;
{
cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
result = new InvalidKeySpecException();
}
};
final AbstractCipher cipherManager = CipherCreator.instance().create();
final String encrypted = cipherManager.encrypt("test-encrypt");
Assert.assertEquals(null, encrypted);
}
/**
* Test method for
* {@link org.openo.baseservice.encrypt.cbb.impl.AesCipher#decrypt(java.lang.String)}.
*/
@Test
public void testDecrypt() {
final AbstractCipher cipherManager = CipherCreator.instance().create();
final String encrypted = cipherManager.encrypt("test-encrypt");
final String decrypted = cipherManager.decrypt(encrypted);
Assert.assertEquals("test-encrypt", decrypted);
}
@Test
public void testDecryptNull() {
final AbstractCipher cipherManager = CipherCreator.instance().create();
String decrypted = cipherManager.decrypt(null);
Assert.assertEquals(null, decrypted);
decrypted = cipherManager.decrypt("");
Assert.assertEquals(null, decrypted);
}
/**
* Test method for
* {@link
* org.openo.baseservice.encrypt.cbb.impl.AesCipher#CipherManagerImpl(java.lang.String)}
* .
*/
@Test
public void testCipherManagerImplString() {
final AbstractCipher cipherManager = CipherCreator.instance().create("secret-key");
final String encrypted = cipherManager.encrypt("test-encrypt");
final String decrypted = cipherManager.decrypt(encrypted);
Assert.assertEquals("test-encrypt", decrypted);
}
/**
*
*
* @since
*/
@Test
public void testCipherManagerImplStringDiffKey() {
final String encrypted = CipherCreator.instance().create("secret-key").encrypt("test-encrypt");
final String decrypted = CipherCreator.instance().create("wrong-key").decrypt(encrypted);
Assert.assertNotSame("test-encrypt", decrypted);
final String decrypt = CipherCreator.instance().create("secret-key").decrypt(encrypted);
Assert.assertEquals("test-encrypt", decrypt);
}
@Test
public void testCreateSecretKeyNoSuchAlgorithmException() throws Exception {
new NonStrictExpectations() {
@Mocked
SecretKeyFactory keyFactory;
{
keyFactory = SecretKeyFactory.getInstance("PBKDF2WithHmacSHA1");
result = new NoSuchAlgorithmException();
}
};
final AbstractCipher cipherManager = CipherCreator.instance().create("secret-key");
final String encrypted = cipherManager.encrypt("test-encrypt");
Assert.assertEquals(encrypted, null);
}
@Test
public void testCreateSecretKeyInvalidKeySpecException() throws Exception {
new NonStrictExpectations() {
@Mocked
SecretKeyFactory keyFactory;
{
keyFactory = SecretKeyFactory.getInstance("PBKDF2WithHmacSHA1");
result = new InvalidKeySpecException();
}
};
final AbstractCipher cipherManager = CipherCreator.instance().create("secret-key");
final String decrypted = cipherManager.decrypt("test-encrypt");
Assert.assertEquals(decrypted, null);
}
}