gerrit.onap Code Review - ccsdk/oran.git/blob

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * Copyright (C) 2020 Nordix Foundation. All rights reserved.
   4  * ======================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  * ========================LICENSE_END===================================
  17  */
  18
  19 package org.onap.ccsdk.oran.a1policymanagementservice.controllers.v2;
  20
  21 import static org.assertj.core.api.Assertions.assertThat;
  22 import static org.awaitility.Awaitility.await;
  23 import static org.hamcrest.CoreMatchers.equalTo;
  24 import static org.junit.jupiter.api.Assertions.assertTrue;
  25
  26 import java.io.File;
  27 import java.lang.reflect.Field;
  28 import java.time.Duration;
  29
  30 import org.apache.commons.io.FileUtils;
  31 import org.junit.jupiter.api.BeforeAll;
  32 import org.junit.jupiter.api.Test;
  33 import org.junit.jupiter.api.extension.ExtendWith;
  34 import org.junit.jupiter.api.io.TempDir;
  35 import org.onap.ccsdk.oran.a1policymanagementservice.clients.AsyncRestClient;
  36 import org.onap.ccsdk.oran.a1policymanagementservice.clients.AsyncRestClientFactory;
  37 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.ApplicationConfig;
  38 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.ImmutableWebClientConfig;
  39 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.WebClientConfig;
  40 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Rics;
  41 import org.onap.ccsdk.oran.a1policymanagementservice.tasks.RefreshConfigTask;
  42 import org.springframework.beans.factory.annotation.Autowired;
  43 import org.springframework.boot.test.context.SpringBootTest;
  44 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
  45 import org.springframework.boot.test.context.TestConfiguration;
  46 import org.springframework.boot.web.server.LocalServerPort;
  47 import org.springframework.context.ApplicationContext;
  48 import org.springframework.context.annotation.Bean;
  49 import org.springframework.http.HttpStatus;
  50 import org.springframework.http.MediaType;
  51 import org.springframework.test.context.TestPropertySource;
  52 import org.springframework.test.context.junit.jupiter.SpringExtension;
  53 import org.springframework.web.reactive.function.client.WebClientResponseException;
  54
  55 import reactor.core.publisher.Mono;
  56 import reactor.test.StepVerifier;
  57
  58 @ExtendWith(SpringExtension.class)
  59 @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
  60 @TestPropertySource(properties = { //
  61         "server.ssl.key-store=./config/keystore.jks", //
  62         "app.webclient.trust-store=./config/truststore.jks", //
  63         "app.vardata-directory=./target"})
  64 class ConfigurationControllerTest {
  65     @Autowired
  66     ApplicationContext context;
  67
  68     @Autowired
  69     ApplicationConfig applicationConfig;
  70
  71     @Autowired
  72     private Rics rics;
  73
  74     @TempDir
  75     public static File temporaryFolder;
  76     private static File configFile;
  77
  78     @BeforeAll
  79     private static void setup() throws Exception {
  80         Field f1 = RefreshConfigTask.class.getDeclaredField("configRefreshInterval");
  81         f1.setAccessible(true);
  82         f1.set(null, Duration.ofSeconds(1));
  83     }
  84
  85     public static class MockApplicationConfig extends ApplicationConfig {
  86         @Override
  87         public String getLocalConfigurationFilePath() {
  88             configFile = new File(temporaryFolder, "config.json");
  89             return configFile.getAbsolutePath();
  90         }
  91     }
  92
  93     /**
  94      * Overrides the BeanFactory.
  95      */
  96     @TestConfiguration
  97     static class TestBeanFactory {
  98         @Bean
  99         public ApplicationConfig getApplicationConfig() {
 100             return new MockApplicationConfig();
 101         }
 102     }
 103
 104     @LocalServerPort
 105     private int port;
 106
 107     @Test
 108     void putValidConfigurationWithNewRic_shouldUpdateRepository() throws Exception {
 109         String url = "a1-policy/v2/configuration";
 110
 111         String resp = restClient().put(url, configAsString()).block();
 112
 113         assertThat(resp).isEmpty();
 114         await().until(rics::size, equalTo(2));
 115
 116         // GET config
 117         resp = restClient().get(url).block();
 118         assertThat(resp).contains("config");
 119     }
 120
 121     @Test
 122     void getNoFileExists() {
 123         String url = "a1-policy/v2/configuration";
 124         testErrorCode(restClient().get(url), HttpStatus.NOT_FOUND, "File does not exist");
 125     }
 126
 127     private String configAsString() throws Exception {
 128         File configFile = new File(getClass().getClassLoader()
 129                 .getResource("test_application_configuration_with_dmaap_config.json").getFile());
 130         return FileUtils.readFileToString(configFile, "UTF-8");
 131     }
 132
 133     @Test
 134     void putInvalidConfiguration_shouldReturnError400() throws Exception {
 135         String url = "a1-policy/v2/configuration";
 136
 137         // Valid JSON but invalid configuration.
 138         testErrorCode(restClient().put(url, "{\"error\":\"error\"}"), HttpStatus.BAD_REQUEST,
 139                 "Missing root configuration");
 140     }
 141
 142     private void testErrorCode(Mono<?> request, HttpStatus expStatus, String responseContains) {
 143         StepVerifier.create(request) //
 144                 .expectSubscription() //
 145                 .expectErrorMatches(t -> checkWebClientError(t, expStatus, responseContains)) //
 146                 .verify();
 147     }
 148
 149     private boolean checkWebClientError(Throwable throwable, HttpStatus expStatus, String responseContains) {
 150         assertTrue(throwable instanceof WebClientResponseException);
 151         WebClientResponseException responseException = (WebClientResponseException) throwable;
 152         assertThat(responseException.getStatusCode()).isEqualTo(expStatus);
 153         assertThat(responseException.getResponseBodyAsString()).contains(responseContains);
 154         assertThat(responseException.getHeaders().getContentType()).isEqualTo(MediaType.APPLICATION_PROBLEM_JSON);
 155         return true;
 156     }
 157
 158     private AsyncRestClient restClient() {
 159         WebClientConfig config = this.applicationConfig.getWebClientConfig();
 160         config = ImmutableWebClientConfig.builder() //
 161                 .keyStoreType(config.keyStoreType()) //
 162                 .keyStorePassword(config.keyStorePassword()) //
 163                 .keyStore(config.keyStore()) //
 164                 .keyPassword(config.keyPassword()) //
 165                 .isTrustStoreUsed(true) //
 166                 .trustStore(config.trustStore()) //
 167                 .trustStorePassword(config.trustStorePassword()) //
 168                 .httpProxyConfig(config.httpProxyConfig()) //
 169                 .build();
 170
 171         AsyncRestClientFactory f = new AsyncRestClientFactory(config);
 172         return f.createRestClient("https://localhost:" + port);
 173
 174     }
 175 }