JWKAPITestCase.java example

Explorer
picketlink-master
/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2012, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.picketlink.test.json.api;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.picketlink.test.json.api.JWEAPITestCase.assertJwEquals;

import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.interfaces.RSAPublicKey;

import org.junit.Before;
import org.junit.Test;
import org.picketlink.json.jose.JWK;
import org.picketlink.json.jose.JWKBuilder;
import org.picketlink.json.jose.JWKSet;

/**
 * The Class JWKAPITestCase.
 *
 * @author Giriraj Sharma
 */
public class JWKAPITestCase {

    private KeyPair keyPair1;
    
    private KeyPair keyPair2;
    
    private KeyPair keyPair3;
    
    private KeyPair keyPair4;

    /**
     * On before.
     *
     * @throws Exception the exception
     */
    @Before
    public void onBefore() throws Exception {
        this.keyPair1 = KeyPairGenerator.getInstance("RSA").generateKeyPair();
        this.keyPair2 = KeyPairGenerator.getInstance("RSA").generateKeyPair();
        this.keyPair3 = KeyPairGenerator.getInstance("RSA").generateKeyPair();
        this.keyPair4 = KeyPairGenerator.getInstance("RSA").generateKeyPair();
    }
    
    /**
     * Creates the JWK.
     *
     * @param publicKey the public key
     * @param keyId the key id
     * @return the JWK
     */
    public JWK createJWK(RSAPublicKey publicKey, String keyId) {
        return new JWKBuilder()
            .modulus(publicKey.getModulus())
            .publicExponent(publicKey.getPublicExponent())
            .keyIdentifier(keyId)
            .keyType("RSA")
            .keyAlgorithm(publicKey.getAlgorithm())
            .keyUse("enc")
            .build();
    }

    /**
     * Test RSA_JWK.
     */
    @Test
    public void testRSAJWK() {
        RSAPublicKey publicKey = (RSAPublicKey) this.keyPair1.getPublic();

        JWK jwk = createJWK(publicKey, "1");

        String jsonString = jwk.toString();

        assertEquals("{\"n\":\"" + jwk.getModulus() + "\",\"e\":\"AQAB\",\"kid\":\"1\",\"kty\":\"RSA\",\"alg\":\"RSA\",\"use\":\"enc\"}", jsonString);
        
        String jsonEncoded = jwk.encode();
        
        JWK parsedJwk = new JWKBuilder().build(jsonEncoded);

        assertNotNull(parsedJwk);
        assertEquals(this.keyPair1.getPublic(), parsedJwk.toRSAPublicKey());
        
        assertEquals(parsedJwk.toRSAPublicKey().getModulus(), ((RSAPublicKey) this.keyPair1.getPublic()).getModulus());
        assertEquals(parsedJwk.toRSAPublicKey().getPublicExponent(), ((RSAPublicKey) this.keyPair1.getPublic()).getPublicExponent());
        assertEquals(parsedJwk.getKeyIdentifier(), jwk.getKeyIdentifier());
        assertEquals(parsedJwk.getKeyType(), jwk.getKeyType());
        assertEquals(parsedJwk.getKeyAlgorithm(), jwk.getKeyAlgorithm());
        assertEquals(parsedJwk.getKeyUse(), jwk.getKeyUse());
    }

    /**
     * Test JWK SET.
     */
    @Test
    public void testJWKSet() {
        JWK jwkKeyPair1 = createJWK((RSAPublicKey) this.keyPair1.getPublic(), "1");
        JWK jwkKeyPair2 = createJWK((RSAPublicKey) this.keyPair2.getPublic(), "2");
        JWK jwkKeyPair3 = createJWK((RSAPublicKey) this.keyPair3.getPublic(), "3");
        JWK jwkKeyPair4 = createJWK((RSAPublicKey) this.keyPair4.getPublic(), "4");

        JWKSet jwkSet = new JWKSet(jwkKeyPair1, jwkKeyPair2, jwkKeyPair3, jwkKeyPair4);
        String jsonKeySet = jwkSet.toString();

        assertJwEquals("{\"keys\":[{\"n\":\"" + jwkKeyPair3.getModulus() + "\",\"e\":\"AQAB\",\"kid\":\"3\",\"kty\":\"RSA\",\"alg\":\"RSA\",\"use\":\"enc\"},{\"n\":\"" + jwkKeyPair2.getModulus() + "\",\"e\":\"AQAB\",\"kid\":\"2\",\"kty\":\"RSA\",\"alg\":\"RSA\",\"use\":\"enc\"},{\"n\":\"" + jwkKeyPair1.getModulus() + "\",\"e\":\"AQAB\",\"kid\":\"1\",\"kty\":\"RSA\",\"alg\":\"RSA\",\"use\":\"enc\"},{\"n\":\"" + jwkKeyPair4.getModulus() + "\",\"e\":\"AQAB\",\"kid\":\"4\",\"kty\":\"RSA\",\"alg\":\"RSA\",\"use\":\"enc\"}]}", jsonKeySet);

        JWKSet parsedKeySet = new JWKSet(jsonKeySet);

        assertNotNull(parsedKeySet.get("1"));
        assertNotNull(parsedKeySet.get("2"));
        assertNotNull(parsedKeySet.get("3"));
        assertNotNull(parsedKeySet.get("4"));
        
        assertEquals(this.keyPair1.getPublic(), parsedKeySet.get("1").toRSAPublicKey());
        assertEquals(this.keyPair2.getPublic(), parsedKeySet.get("2").toRSAPublicKey());
        assertEquals(this.keyPair3.getPublic(), parsedKeySet.get("3").toRSAPublicKey());
        assertEquals(this.keyPair4.getPublic(), parsedKeySet.get("4").toRSAPublicKey());
        
        assertEquals(parsedKeySet.get("2").toRSAPublicKey().getModulus(), ((RSAPublicKey) this.keyPair2.getPublic()).getModulus());
        assertEquals(parsedKeySet.get("2").toRSAPublicKey().getPublicExponent(), ((RSAPublicKey) this.keyPair2.getPublic()).getPublicExponent());
        assertEquals(parsedKeySet.get("2").getKeyIdentifier(), jwkKeyPair2.getKeyIdentifier());
        assertEquals(parsedKeySet.get("2").getKeyType(), jwkKeyPair2.getKeyType());
        assertEquals(parsedKeySet.get("2").getKeyAlgorithm(), jwkKeyPair2.getKeyAlgorithm());
        assertEquals(parsedKeySet.get("2").getKeyUse(), jwkKeyPair2.getKeyUse());
    }
}