common/src/test/java/org/onap/so/client/adapter/rest/AdapterRestClientTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2019 Nokia.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.so.client.adapter.rest;
  22
  23 import static org.assertj.core.api.Assertions.assertThat;
  24 import static org.assertj.core.api.Assertions.entry;
  25 import static org.mockito.Mockito.mock;
  26 import static org.mockito.Mockito.when;
  27
  28 import java.net.URI;
  29 import java.net.URISyntaxException;
  30 import java.security.GeneralSecurityException;
  31 import java.util.HashMap;
  32 import java.util.Map;
  33 import org.apache.commons.codec.binary.Base64;
  34 import org.junit.Before;
  35 import org.junit.Test;
  36 import org.onap.so.client.policy.JettisonStyleMapperProvider;
  37 import org.onap.so.utils.CryptoUtils;
  38 import org.onap.so.utils.TargetEntity;
  39
  40 public class AdapterRestClientTest {
  41
  42     private static final String CRYPTO_KEY = "546573746F736973546573746F736973";
  43     private static final String INVALID_CRYPTO_KEY = "1234";
  44
  45     private Map<String, String> headerMap;
  46     private AdapterRestProperties adapterRestPropertiesMock;
  47
  48     @Before
  49     public void setup() {
  50         headerMap = new HashMap<>();
  51         adapterRestPropertiesMock = mock(AdapterRestProperties.class);
  52     }
  53
  54     @Test
  55     public void initializeHeaderMap_success() throws URISyntaxException, GeneralSecurityException {
  56         // given
  57         String encyptedMessage = CryptoUtils.encrypt("testAdapter", CRYPTO_KEY);
  58         when(adapterRestPropertiesMock.getAuth()).thenReturn(encyptedMessage);
  59         when(adapterRestPropertiesMock.getKey()).thenReturn(CRYPTO_KEY);
  60         AdapterRestClient testedObject = new AdapterRestClient(adapterRestPropertiesMock, new URI(""));
  61         // when
  62         testedObject.initializeHeaderMap(headerMap);
  63         // then
  64         assertThat(headerMap).containsOnly(entry("Authorization", getExpectedEncodedString(encyptedMessage)));
  65     }
  66
  67     @Test
  68     public void initializeHeaderMap_putNullToMapWhenAuthIsNull() throws URISyntaxException {
  69         // given
  70         AdapterRestClient testedObject = new AdapterRestClient(adapterRestPropertiesMock, new URI(""));
  71         // when
  72         testedObject.initializeHeaderMap(headerMap);
  73         // then
  74         assertThat(headerMap).containsOnly(entry("Authorization", null));
  75     }
  76
  77     @Test
  78     public void initializeHeaderMap_putNullToMapWhenExOccurs() throws URISyntaxException, GeneralSecurityException {
  79         // given
  80         String encyptedMessage = CryptoUtils.encrypt("testAdapter", CRYPTO_KEY);
  81         when(adapterRestPropertiesMock.getAuth()).thenReturn(encyptedMessage);
  82         when(adapterRestPropertiesMock.getKey()).thenReturn(INVALID_CRYPTO_KEY);
  83         AdapterRestClient testedObject = new AdapterRestClient(adapterRestPropertiesMock, new URI(""),
  84                 "accept", "contentType");
  85         // when
  86         testedObject.initializeHeaderMap(headerMap);
  87         // then
  88         assertThat(headerMap).containsOnly(entry("Authorization", null));
  89     }
  90
  91     @Test
  92     public void getTargetEntity_success() throws URISyntaxException {
  93         AdapterRestClient testedObject = new AdapterRestClient(adapterRestPropertiesMock, new URI(""));
  94         assertThat(testedObject.getTargetEntity()).isEqualTo(TargetEntity.OPENSTACK_ADAPTER);
  95     }
  96
  97     @Test
  98     public void getCommonObjectMapperProvider_success() throws URISyntaxException {
  99         AdapterRestClient testedObject = new AdapterRestClient(adapterRestPropertiesMock, new URI(""));
 100         assertThat(testedObject.getCommonObjectMapperProvider()).isInstanceOf(JettisonStyleMapperProvider.class);
 101     }
 102
 103     private String getExpectedEncodedString(String encryptedMessage) throws GeneralSecurityException {
 104         String auth = CryptoUtils.decrypt(encryptedMessage, CRYPTO_KEY);
 105         byte[] encoded = Base64.encodeBase64(auth.getBytes());
 106         String encodedString = new String(encoded);
 107         return "Basic " + encodedString;
 108     }
 109 }