trustStoreMerger/src/test/java/org/onap/oom/truststoremerger/certification/file/model/PemTruststoreTest.java

   1 /*============LICENSE_START=======================================================
   2  * oom-truststore-merger
   3  * ================================================================================
   4  * Copyright (C) 2020 Nokia. All rights reserved.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  * ============LICENSE_END=========================================================
  18  */
  19
  20 package org.onap.oom.truststoremerger.certification.file.model;
  21
  22 import static org.assertj.core.api.Assertions.assertThat;
  23 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.when;
  26 import static org.onap.oom.truststoremerger.api.CertificateConstants.X_509_CERTIFICATE;
  27
  28 import java.io.IOException;
  29 import java.security.cert.Certificate;
  30 import java.security.cert.CertificateEncodingException;
  31 import java.util.ArrayList;
  32 import java.util.List;
  33 import org.junit.jupiter.api.AfterAll;
  34 import org.junit.jupiter.api.Test;
  35 import org.onap.oom.truststoremerger.api.ExitableException;
  36 import org.onap.oom.truststoremerger.certification.file.provider.entry.CertificateWithAlias;
  37 import org.onap.oom.truststoremerger.certification.file.provider.entry.CertificateWithAliasFactory;
  38 import org.onap.oom.truststoremerger.certification.file.TestCertificateProvider;
  39 import org.onap.oom.truststoremerger.certification.file.exception.MissingTruststoreException;
  40 import org.onap.oom.truststoremerger.certification.file.exception.TruststoreDataOperationException;
  41 import org.onap.oom.truststoremerger.certification.file.exception.WriteTruststoreFileException;
  42
  43 class PemTruststoreTest {
  44
  45     public static final int EXPECTED_ONE = 1;
  46     public static final int EXPECTED_THREE = 3;
  47     public static final int FIRST_ELEMENT = 0;
  48
  49     private final CertificateWithAliasFactory factory = new CertificateWithAliasFactory();
  50
  51     @Test
  52     void pemTruststoreShouldReadCertificatesFromFile() throws ExitableException {
  53
  54         //given
  55         PemTruststore pemTruststore = TestCertificateProvider.getSamplePemTruststoreFile();
  56
  57         //when
  58         List<CertificateWithAlias> certificates = pemTruststore.getCertificates();
  59         Certificate certificate = certificates.get(FIRST_ELEMENT).getCertificate();
  60         //then
  61
  62         assertThat(certificates).hasSize(EXPECTED_ONE);
  63         assertThat(certificate.getType()).isEqualTo(X_509_CERTIFICATE);
  64     }
  65
  66     @Test
  67     void pemTruststoreShouldAddDifferentCertificates() throws IOException, ExitableException {
  68
  69         //given
  70         PemTruststore tmpPemTruststoreFile = TestCertificateProvider.createTmpPemTruststoreFile();
  71         List<CertificateWithAlias> jksTruststoreCertificates = TestCertificateProvider
  72             .getSampleJksTruststoreFileWithUniqueAlias().getCertificates();
  73         List<CertificateWithAlias> p12TruststoreCertificates = TestCertificateProvider.getSampleP12Truststore()
  74             .getCertificates();
  75
  76         //when
  77         tmpPemTruststoreFile.addCertificate(jksTruststoreCertificates);
  78         tmpPemTruststoreFile.addCertificate(p12TruststoreCertificates);
  79         tmpPemTruststoreFile.saveFile();
  80
  81         PemTruststore tmpPemTruststoreSaved = TestCertificateProvider.getTmpPemTruststoreFile();
  82         List<CertificateWithAlias> addedCertificates = tmpPemTruststoreSaved.getCertificates();
  83         Certificate certificate = addedCertificates.get(FIRST_ELEMENT).getCertificate();
  84
  85         //then
  86         assertThat(addedCertificates).hasSize(EXPECTED_THREE);
  87         assertThat(certificate.getType()).isEqualTo(X_509_CERTIFICATE);
  88
  89     }
  90
  91     @Test
  92     void privateKeyIsSkippedWhileReadingCertificates() throws ExitableException {
  93         //given
  94         PemTruststore pemTruststore = TestCertificateProvider.getPemWithPrivateKeyTruststoreFile();
  95
  96         //when
  97         List<CertificateWithAlias> certificate = pemTruststore.getCertificates();
  98         //then
  99
 100         assertThat(certificate).hasSize(EXPECTED_ONE);
 101     }
 102
 103     @Test
 104     void shouldThrowExceptionWhenCannotSaveFile() throws IOException, ExitableException {
 105         //given
 106         PemTruststore tmpPemTruststoreFile = TestCertificateProvider.createTmpPemTruststoreFile();
 107         List<CertificateWithAlias> pemTruststoreCertificates =
 108             TestCertificateProvider.getSamplePemTruststoreFile().getCertificates();
 109         //when
 110         tmpPemTruststoreFile.addCertificate(pemTruststoreCertificates);
 111         tmpPemTruststoreFile.getFile().setWritable(false);
 112         //then
 113         assertThatExceptionOfType(WriteTruststoreFileException.class)
 114             .isThrownBy(tmpPemTruststoreFile::saveFile);
 115
 116     }
 117
 118     @Test
 119     void shouldThrowExceptionWhenFileNotContainsCertificate() throws IOException {
 120         //given
 121         PemTruststore tmpPemTruststoreFile = TestCertificateProvider.createEmptyTmpPemTruststoreFile();
 122         //when//then
 123         assertThatExceptionOfType(MissingTruststoreException.class)
 124             .isThrownBy(tmpPemTruststoreFile::getCertificates);
 125     }
 126
 127     @Test
 128     void shouldThrowExceptionWhenCannotConvertCertificateToPem() throws Exception {
 129         //given
 130         PemTruststore pemTruststore = TestCertificateProvider.createTmpPemTruststoreFile();
 131         Certificate certificate = mock(Certificate.class);
 132
 133         when(certificate.getEncoded()).thenThrow(new CertificateEncodingException());
 134
 135         List<CertificateWithAlias> certificatesWithAliases = new ArrayList<>();
 136         certificatesWithAliases.add(factory.createPemCertificate(certificate));
 137         pemTruststore.addCertificate(certificatesWithAliases);
 138
 139         //when //then
 140         assertThatExceptionOfType(TruststoreDataOperationException.class)
 141             .isThrownBy(pemTruststore::saveFile);
 142     }
 143
 144     @AfterAll
 145     static void removeTemporaryFiles() throws IOException {
 146         TestCertificateProvider.removeTemporaryFiles();
 147     }
 148
 149 }