core/src/test/java/org/jclouds/encryption/EncryptionServiceTest.java

/**
 *
 * Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
 *
 * ====================================================================
 * 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.jclouds.encryption;

import static org.testng.Assert.assertEquals;

import java.io.UnsupportedEncodingException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;
import java.util.concurrent.Callable;
import java.util.concurrent.CompletionService;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorCompletionService;

import org.jclouds.PerformanceTest;
import org.jclouds.encryption.EncryptionService;
import org.jclouds.encryption.internal.Base64;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import com.google.inject.Guice;
import com.google.inject.Injector;

/**
 * This tests the performance of Digest commands.
 * 
 * @author Adrian Cole
 */
@Test(groups = "performance", sequential = true, testName = "jclouds.encryptionService")
public class EncryptionServiceTest extends PerformanceTest {

   protected EncryptionService encryptionService;
   @BeforeTest
   protected void createEncryptionService() {
      Injector i = Guice.createInjector();
      encryptionService = i.getInstance(EncryptionService.class);
   }
   
   public final static Object[][] base64KeyMessageDigest = {
      { Base64.decode("CwsLCwsLCwsLCwsLCwsLCwsLCws="), "Hi There",
               "thcxhlUFcmTii8C2+zeMjvFGvgA=" },
      { Base64.decode("SmVmZQ=="), "what do ya want for nothing?",
               "7/zfauXrL6LSdBbV8YTfnCWafHk=" },
      { Base64.decode("DAwMDAwMDAwMDAwMDAwMDAwMDAw="), "Test With Truncation",
               "TBoDQktV4H/n8nvh1Yu5MkqaWgQ=" },
      {
               Base64
                        .decode("qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqo="),
               "Test Using Larger Than Block-Size Key - Hash Key First",
               "qkrl4VJy0A6VcFY3zoo7Ve1AIRI=" },
      {
               Base64
                        .decode("qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqo="),
               "Test Using Larger Than Block-Size Key and Larger Than One Block-Size Data",
               "6OmdD0UjfXhta7qnllx4CLv/GpE=" } };
   

   @DataProvider(name = "hmacsha1")
   public Object[][] createData1() {
      return base64KeyMessageDigest;
   }

   @Test(dataProvider = "hmacsha1", enabled = false)
   public void testHmacSha1Base64(byte[] key, String message, String base64Digest)
            throws NoSuchProviderException, NoSuchAlgorithmException, InvalidKeyException {
      String b64 = encryptionService.hmacSha1Base64(message, key);
      assertEquals(b64, base64Digest);
   }

   @Test(dataProvider = "hmacsha1", enabled = false)
   void testDigestSerialResponseTime(byte[] key, String message, String base64Digest)
            throws NoSuchProviderException, NoSuchAlgorithmException, InvalidKeyException {
      for (int i = 0; i < 10000; i++)
         testHmacSha1Base64(key, message, base64Digest);
   }

   @Test(dataProvider = "hmacsha1", enabled = false)
   void testDigestParallelResponseTime(final byte[] key, final String message,
            final String base64Digest) throws NoSuchProviderException, NoSuchAlgorithmException,
            InvalidKeyException, InterruptedException, ExecutionException {
      CompletionService<Boolean> completer = new ExecutorCompletionService<Boolean>(exec);
      for (int i = 0; i < 10000; i++)
         completer.submit(new Callable<Boolean>() {
            public Boolean call() throws Exception {
               testHmacSha1Base64(key, message, base64Digest);
               return true;
            }
         });
      for (int i = 0; i < 10000; i++)
         assert completer.take().get();
   }

   @DataProvider(name = "eTag")
   public Object[][] createMD5Data() {
      return base64MD5MessageDigest;
   }

   public final static Object[][] base64MD5MessageDigest = {
            { "apple", "1f3870be274f6c49b3e31a0c6728957f" },
            { "bear", "893b56e3cfe153fb770a120b83bac20c" },
            { "candy", "c48ba993d35c3abe0380f91738fe2a34" },
            { "dogma", "95eb470e4faee302e9cd3063b1923dab" },
            { "emma", "00a809937eddc44521da9521269e75c6" } };


   @Test(dataProvider = "eTag", enabled = false)
   public void testMD5Digest(String message, String base64Digest) throws NoSuchProviderException,
            NoSuchAlgorithmException, InvalidKeyException, UnsupportedEncodingException {
      String b64 = encryptionService.md5Hex(message.getBytes());
      assertEquals(base64Digest, b64);
   }

   byte[] bytes = { 0, 1, 2, 4, 8, 16, 32, 64 };
   String hex = "0001020408102040";

   public void testHexStringEncode() throws UnsupportedEncodingException {
      assertEquals(encryptionService.toHexString(bytes), hex);
   }

   public void testHexStringDecode() throws UnsupportedEncodingException {
      assertEquals(encryptionService.fromHexString(hex), bytes);
   }

   public void testHexStringDecodeOx() throws UnsupportedEncodingException {
      assertEquals(encryptionService.fromHexString("0x" + hex), bytes);
   }
}