CertificateStoreLoaderTest.java

/*
 *   Copyright (c) 2012 Mike Heath.  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.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */
package cloudeventbus.pki;

import cloudeventbus.Subject;
import org.testng.annotations.Test;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;

/**
 * @author Mike Heath <elcapo@gmail.com>
 */
public class CertificateStoreLoaderTest {

	@Test
	public void storeAndLoad() throws Exception {
		final KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
		keyPairGenerator.initialize(2048);
		KeyPair keyPair = keyPairGenerator.generateKeyPair();

		final Certificate certificate1 = new Certificate(Certificate.Type.AUTHORITY, 1l, 2l, System.currentTimeMillis(), keyPair.getPublic(), Subject.list("foo.*"), Subject.list("bar.*"), null, null);
		final Certificate certificate2 = new Certificate(Certificate.Type.SERVER, 3l, 4l, System.currentTimeMillis(), keyPair.getPublic(), Subject.list("foo2.*"), Subject.list("bar2.*"), "Certificate 2", null);

		final List<Certificate> certificates = new ArrayList<>();
		certificates.add(certificate1);
		certificates.add(certificate2);

		final ByteArrayOutputStream out = new ByteArrayOutputStream();
		CertificateStoreLoader.store(out, certificates);

		final ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
		final List<Certificate> copiedCertificates = new ArrayList<>();
		CertificateStoreLoader.load(in, copiedCertificates);

		assertEquals(2, copiedCertificates.size());
		final Iterator<Certificate> certificateIterator = copiedCertificates.iterator();
		assertEquals(certificate1, certificateIterator.next());
		assertEquals(certificate2, certificateIterator.next());
		assertFalse(certificateIterator.hasNext());
	}
}