EncryptionTest.java

/*
 * Copyright (C) 2018 LinkedIn Corp. All rights reserved.
 *
 * 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.
 */
package crypto;

import azkaban.crypto.Crypto;
import azkaban.crypto.ICrypto;
import azkaban.crypto.Version;
import junit.framework.Assert;
import org.apache.commons.lang.StringUtils;
import org.junit.Test;

public class EncryptionTest {

  @Test
  public void testEncryption() {
    final String plainText = "test";
    final String passphrase = "test1234";

    final ICrypto crypto = new Crypto();

    for (final Version ver : Version.values()) {
      final String cipheredText = crypto.encrypt(plainText, passphrase, ver);
      Assert.assertEquals(plainText, crypto.decrypt(cipheredText, passphrase));
    }
  }

  @Test
  public void testInvalidParams() {
    final ICrypto crypto = new Crypto();
    final String[] args = {"", null, "test"};
    for (final Version ver : Version.values()) {
      for (final String plaintext : args) {
        for (final String passphrase : args) {
          try {
            if (!StringUtils.isEmpty(plaintext) && !StringUtils.isEmpty(passphrase)) {
              final String cipheredText = crypto.encrypt(plaintext, passphrase, ver);
              Assert.assertEquals(plaintext, crypto.decrypt(cipheredText, passphrase));
            } else {
              crypto.encrypt(plaintext, passphrase, ver);
              Assert.fail("Encyption should have failed with invalid parameters. plaintext: "
                  + plaintext + " , passphrase: " + passphrase);
            }
          } catch (final Exception e) {
            Assert.assertTrue(e instanceof IllegalArgumentException);
          }

        }
      }
    }
  }

}