KeyTest.java

package com.googlecode.totallylazy.security;

import org.junit.Test;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.not;
import static org.hamcrest.MatcherAssert.assertThat;

public class KeyTest {
    @Test
    public void canGenerateEncodedKey() throws Exception {
        isValidSecretKey(Key.key());
    }

    @Test
    public void canProvideSecretForKey() throws Exception {
        isValidSecretKey(Key.key("lpYsPLjBJnK4Up4mrgomWQ=="));
    }

    @Test
    public void canEncryptAndDecrypt() throws Exception {
        Key key = Key.key();
        String value = "Hi Mum!";
        String encrypted = key.encrypt(value);
        assertThat(encrypted, is(not(value)));
        assertThat(key.decrypt(encrypted), is(value));
    }

    @Test
    public void canEncryptAndDecryptUrlSafely() throws Exception {
        Key key = Key.key();
        String value = "Hi Mum!";
        String encrypted = key.encryptUrlSafe(value);
        assertThat(encrypted, is(not(value)));
        assertThat(key.decryptUrlSafe(encrypted), is(value));
    }

    private void isValidSecretKey(Key secret) throws Exception {
        SecretKeySpec keySpec = new SecretKeySpec(Base64.decode(secret.toString()), Key.ALGORITHM);
        Cipher.getInstance(Key.ALGORITHM).init(Cipher.ENCRYPT_MODE, keySpec);
    }
}