edu.wisc.doit.tcrypt.KeyReadingAndWritingTest.java Source code

Introduction

Here is the source code for edu.wisc.doit.tcrypt.KeyReadingAndWritingTest.java

Source

/**
 * Copyright 2012, Board of Regents of the University of
 * Wisconsin System. See the NOTICE file distributed with
 * this work for additional information regarding copyright
 * ownership. Board of Regents of the University of Wisconsin
 * System licenses this file to you 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.
 */
/**
 * @author Brad Leege <leege@doit.wisc.edu>
 * Created on 2/14/13 at 4:29 PM
 */

package edu.wisc.doit.tcrypt;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.io.File;
import java.io.FileReader;
import java.security.KeyPair;
import java.security.PublicKey;

import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;
import org.bouncycastle.openssl.PEMParser;
import org.bouncycastle.openssl.jcajce.JcaPEMKeyConverter;
import org.joda.time.DateTime;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;

import edu.wisc.doit.tcrypt.dao.IKeysKeeper;
import edu.wisc.doit.tcrypt.dao.impl.KeysKeeper;
import edu.wisc.doit.tcrypt.vo.ServiceKey;

public class KeyReadingAndWritingTest {
    @Rule
    public TemporaryFolder testFolder = new TemporaryFolder();

    private IKeysKeeper keysKeeper;

    @Before
    public void setup() throws Exception {
        final BouncyCastleKeyPairGenerator bouncyCastleKeyPairGenerator = new BouncyCastleKeyPairGenerator();
        this.keysKeeper = new KeysKeeper(testFolder.getRoot().getCanonicalPath(), bouncyCastleKeyPairGenerator);
    }

    @Test
    public void testCreateWriteAndReadBackKey() throws Exception {
        // Create ServiceKey
        final KeyPair kp = this.keysKeeper.createServiceKey("example.com", 2048, "username");
        assertNotNull(kp);

        // Step 3: Read ServiceKey from filesystem
        ServiceKey foundKey = keysKeeper.getServiceKey("example.com");
        assertNotNull(foundKey);

        // Compare original ServiceKey content with new ServiceKey read from filesystem
        assertEquals("example.com", foundKey.getServiceName());
        assertEquals("username", foundKey.getCreatedByNetId());
        assertEquals(2048, foundKey.getKeyLength());
        //Verify created in same minute
        assertEquals(DateTime.now().minuteOfHour().roundFloorCopy(),
                foundKey.getDayCreated().minuteOfHour().roundFloorCopy());
        assertNotNull(foundKey.getFileEncrypter());
        assertNotNull(foundKey.getTokenEncrypter());

        final File keyFile = foundKey.getKeyFile();
        assertNotNull(keyFile);

        @SuppressWarnings("resource")
        PEMParser pemParser = new PEMParser(new FileReader(keyFile));
        Object object = pemParser.readObject();
        JcaPEMKeyConverter converter = new JcaPEMKeyConverter().setProvider("BC");
        final PublicKey actualPublicKey = converter.getPublicKey((SubjectPublicKeyInfo) object);

        assertArrayEquals(kp.getPublic().getEncoded(), actualPublicKey.getEncoded());
    }
}