certServiceClient/src/test/java/org/onap/oom/certservice/client/certification/CsrFactoryTest.java

   1 /*============LICENSE_START=======================================================
   2  * oom-certservice-client
   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.certservice.client.certification;
  21
  22
  23 import static org.assertj.core.api.Assertions.assertThat;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.when;
  26
  27 import java.security.KeyPair;
  28 import java.util.List;
  29 import org.bouncycastle.asn1.x509.GeneralName;
  30 import org.junit.jupiter.api.Test;
  31 import org.onap.oom.certservice.client.certification.exception.CsrGenerationException;
  32 import org.onap.oom.certservice.client.certification.exception.KeyPairGenerationException;
  33 import org.onap.oom.certservice.client.configuration.model.CsrConfiguration;
  34 import org.onap.oom.certservice.client.configuration.model.San;
  35
  36 class CsrFactoryTest {
  37
  38     private CsrConfiguration config = mock(CsrConfiguration.class);
  39
  40     @Test
  41     void createEncodedCsr_shouldSucceedWhenAllFieldsAreSetCorrectly()
  42         throws KeyPairGenerationException, CsrGenerationException {
  43
  44         KeyPair keyPair = createKeyPair();
  45
  46         mockRequiredConfigFields();
  47         mockOptionalConfigFields();
  48
  49         assertThat(new CsrFactory(config).createCsrInPem(keyPair)).isNotEmpty();
  50     }
  51
  52     @Test
  53     void createEncodedCsr_shouldSucceedWhenRequiredFieldsAreSetCorrectly()
  54         throws KeyPairGenerationException, CsrGenerationException {
  55
  56         KeyPair keyPair = createKeyPair();
  57
  58         mockRequiredConfigFields();
  59         mockOptionalConfigFieldsEmpty();
  60
  61         assertThat(new CsrFactory(config).createCsrInPem(keyPair)).isNotEmpty();
  62     }
  63
  64     private KeyPair createKeyPair() {
  65         return new KeyPairFactory(EncryptionAlgorithmConstants.RSA_ENCRYPTION_ALGORITHM,
  66             EncryptionAlgorithmConstants.KEY_SIZE).create();
  67     }
  68
  69     private void mockRequiredConfigFields() {
  70         when(config.getCommonName()).thenReturn("onap.org");
  71         when(config.getOrganization()).thenReturn("Linux-Foundation");
  72         when(config.getCountry()).thenReturn("US");
  73         when(config.getState()).thenReturn("California");
  74     }
  75
  76     private void mockOptionalConfigFields() {
  77         San san1 = new San("onapexample.com", GeneralName.dNSName);
  78         San san2 = new San("onapexample.com.pl", GeneralName.dNSName);
  79
  80         when(config.getLocation()).thenReturn("San-Francisco");
  81         when(config.getSans()).thenReturn(List.of(san1, san2));
  82         when(config.getOrganizationUnit()).thenReturn("ONAP");
  83     }
  84
  85     private void mockOptionalConfigFieldsEmpty() {
  86         when(config.getLocation()).thenReturn(null);
  87         when(config.getSans()).thenReturn(null);
  88         when(config.getOrganizationUnit()).thenReturn(null);
  89     }
  90 }
  91