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.io.TempDir;
  34 import org.onap.ccsdk.oran.a1policymanagementservice.clients.AsyncRestClient;
  35 import org.onap.ccsdk.oran.a1policymanagementservice.clients.AsyncRestClientFactory;
  36 import org.onap.ccsdk.oran.a1policymanagementservice.clients.SecurityContext;
  37 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.ApplicationConfig;
  38 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.WebClientConfig;
  39 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Rics;
  40 import org.onap.ccsdk.oran.a1policymanagementservice.tasks.RefreshConfigTask;
  41 import org.springframework.beans.factory.annotation.Autowired;
  42 import org.springframework.boot.test.context.SpringBootTest;
  43 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
  44 import org.springframework.boot.test.context.TestConfiguration;
  45 import org.springframework.boot.web.embedded.tomcat.TomcatServletWebServerFactory;
  46 import org.springframework.boot.web.server.LocalServerPort;
  47 import org.springframework.boot.web.servlet.server.ServletWebServerFactory;
  48 import org.springframework.context.ApplicationContext;
  49 import org.springframework.context.annotation.Bean;
  50 import org.springframework.http.HttpStatus;
  51 import org.springframework.http.MediaType;
  52 import org.springframework.test.context.TestPropertySource;
  53 import org.springframework.web.reactive.function.client.WebClientResponseException;
  54
  55 import reactor.core.publisher.Mono;
  56 import reactor.test.StepVerifier;
  57
  58 @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
  59 @TestPropertySource(properties = { //
  60         "server.ssl.key-store=./config/keystore.jks", //
  61         "app.webclient.trust-store=./config/truststore.jks", //
  62         "app.vardata-directory=./target", //
  63         "app.config-file-schema-path=/application_configuration_schema.json" //
  64 })
  65 class ConfigurationControllerTest {
  66     @Autowired
  67     ApplicationContext context;
  68
  69     @Autowired
  70     ApplicationConfig applicationConfig;
  71
  72     @Autowired
  73     private Rics rics;
  74
  75     @TempDir
  76     public static File temporaryFolder;
  77     private static File configFile;
  78
  79     @BeforeAll
  80     private static void setup() throws Exception {
  81         Field f1 = RefreshConfigTask.class.getDeclaredField("configRefreshInterval");
  82         f1.setAccessible(true);
  83         f1.set(null, Duration.ofSeconds(1));
  84     }
  85
  86     public static class MockApplicationConfig extends ApplicationConfig {
  87         @Override
  88         public String getLocalConfigurationFilePath() {
  89             configFile = new File(temporaryFolder, "config.json");
  90             return configFile.getAbsolutePath();
  91         }
  92     }
  93
  94     /**
  95      * Overrides the BeanFactory.
  96      */
  97     @TestConfiguration
  98     static class TestBeanFactory {
  99         @Bean
 100         public ApplicationConfig getApplicationConfig() {
 101             return new MockApplicationConfig();
 102         }
 103
 104         @Bean
 105         public ServletWebServerFactory servletContainer() {
 106             return new TomcatServletWebServerFactory();
 107         }
 108     }
 109
 110     @LocalServerPort
 111     private int port;
 112
 113     @Test
 114     void putValidConfigurationWithNewRic_shouldUpdateRepository() throws Exception {
 115         String url = "a1-policy/v2/configuration";
 116
 117         String resp = restClient().put(url, configAsString()).block();
 118
 119         assertThat(resp).isEmpty();
 120         await().until(rics::size, equalTo(2));
 121
 122         // GET config
 123         resp = restClient().get(url).block();
 124         assertThat(resp).contains("config");
 125     }
 126
 127     @Test
 128     void getNoFileExists() {
 129         String url = "a1-policy/v2/configuration";
 130         testErrorCode(restClient().get(url), HttpStatus.NOT_FOUND, "File does not exist");
 131     }
 132
 133     private String configAsString() throws Exception {
 134         File configFile = new File(getClass().getClassLoader()
 135                 .getResource("test_application_configuration_with_dmaap_config.json").getFile());
 136         return FileUtils.readFileToString(configFile, "UTF-8");
 137     }
 138
 139     @Test
 140     void putInvalidConfiguration_shouldReturnError400() throws Exception {
 141         String url = "a1-policy/v2/configuration";
 142
 143         // Valid JSON but invalid configuration.
 144         testErrorCode(restClient().put(url, "{\"error\":\"error\"}"), HttpStatus.BAD_REQUEST, "");
 145     }
 146
 147     private void testErrorCode(Mono<?> request, HttpStatus expStatus, String responseContains) {
 148         StepVerifier.create(request) //
 149                 .expectSubscription() //
 150                 .expectErrorMatches(t -> checkWebClientError(t, expStatus, responseContains)) //
 151                 .verify();
 152     }
 153
 154     private boolean checkWebClientError(Throwable throwable, HttpStatus expStatus, String responseContains) {
 155         assertTrue(throwable instanceof WebClientResponseException);
 156         WebClientResponseException responseException = (WebClientResponseException) throwable;
 157         assertThat(responseException.getStatusCode()).isEqualTo(expStatus);
 158         assertThat(responseException.getResponseBodyAsString()).contains(responseContains);
 159         assertThat(responseException.getHeaders().getContentType()).isEqualTo(MediaType.APPLICATION_PROBLEM_JSON);
 160         return true;
 161     }
 162
 163     private AsyncRestClient restClient() {
 164         WebClientConfig config = this.applicationConfig.getWebClientConfig();
 165         config = WebClientConfig.builder() //
 166                 .keyStoreType(config.getKeyStoreType()) //
 167                 .keyStorePassword(config.getKeyStorePassword()) //
 168                 .keyStore(config.getKeyStore()) //
 169                 .keyPassword(config.getKeyPassword()) //
 170                 .isTrustStoreUsed(false) //
 171                 .trustStore(config.getTrustStore()) //
 172                 .trustStorePassword(config.getTrustStorePassword()) //
 173                 .httpProxyConfig(config.getHttpProxyConfig()) //
 174                 .build();
 175
 176         AsyncRestClientFactory f = new AsyncRestClientFactory(config, new SecurityContext(""));
 177         return f.createRestClientNoHttpProxy("https://localhost:" + port);
 178
 179     }
 180 }