2 * ========================LICENSE_START=================================
5 * Copyright (C) 2019 Nordix Foundation
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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===================================
21 package org.oransc.policyagent.configuration;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.junit.Assert.assertTrue;
25 import static org.mockito.ArgumentMatchers.any;
26 import static org.mockito.Mockito.doReturn;
27 import static org.mockito.Mockito.mock;
28 import static org.mockito.Mockito.spy;
29 import static org.mockito.Mockito.times;
30 import static org.mockito.Mockito.verify;
32 import ch.qos.logback.classic.spi.ILoggingEvent;
33 import ch.qos.logback.core.read.ListAppender;
35 import com.google.common.base.Charsets;
36 import com.google.common.io.Resources;
37 import com.google.gson.JsonIOException;
38 import com.google.gson.JsonObject;
39 import com.google.gson.JsonParser;
40 import com.google.gson.JsonSyntaxException;
42 import java.io.ByteArrayInputStream;
43 import java.io.IOException;
44 import java.io.InputStream;
45 import java.io.InputStreamReader;
47 import java.nio.charset.StandardCharsets;
48 import java.util.Arrays;
49 import java.util.Properties;
50 import java.util.Vector;
52 import org.junit.jupiter.api.Assertions;
53 import org.junit.jupiter.api.Test;
54 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClient;
55 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.EnvProperties;
56 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.ImmutableEnvProperties;
57 import org.oransc.policyagent.exceptions.ServiceException;
58 import org.oransc.policyagent.utils.LoggingUtils;
60 import reactor.core.publisher.Flux;
61 import reactor.core.publisher.Mono;
62 import reactor.test.StepVerifier;
64 public class ApplicationConfigTest {
66 private ApplicationConfig appConfigUnderTest;
67 CbsClient cbsClient = mock(CbsClient.class);
69 public static final ImmutableRicConfig CORRECT_RIC_CONIFG = ImmutableRicConfig.builder() //
71 .baseUrl("http://localhost:8080/") //
72 .managedElementIds(new Vector<String>(Arrays.asList("kista_1", "kista_2"))) //
75 private static EnvProperties properties() {
76 return ImmutableEnvProperties.builder() //
77 .consulHost("host") //
79 .cbsName("cbsName") //
80 .appName("appName") //
85 public void whenTheConfigurationFits() throws IOException, ServiceException {
87 appConfigUnderTest = spy(ApplicationConfig.class);
88 appConfigUnderTest.systemEnvironment = new Properties();
90 doReturn(getCorrectJson()).when(appConfigUnderTest).createInputStream(any());
91 appConfigUnderTest.initialize();
94 verify(appConfigUnderTest, times(1)).loadConfigurationFromFile(any());
96 Vector<RicConfig> ricConfigs = appConfigUnderTest.getRicConfigs();
97 RicConfig ricConfig = ricConfigs.firstElement();
98 assertThat(ricConfigs).isNotNull();
99 assertThat(ricConfig).isEqualTo(CORRECT_RIC_CONIFG);
103 public void whenFileIsExistsButJsonIsIncorrect() throws IOException, ServiceException {
105 appConfigUnderTest = spy(ApplicationConfig.class);
106 appConfigUnderTest.systemEnvironment = new Properties();
109 doReturn(getIncorrectJson()).when(appConfigUnderTest).createInputStream(any());
110 appConfigUnderTest.loadConfigurationFromFile(any());
113 verify(appConfigUnderTest, times(1)).loadConfigurationFromFile(any());
114 Assertions.assertNull(appConfigUnderTest.getRicConfigs());
118 public void whenPeriodicConfigRefreshNoEnvironmentVariables() {
120 appConfigUnderTest = spy(ApplicationConfig.class);
121 appConfigUnderTest.systemEnvironment = new Properties();
123 final ListAppender<ILoggingEvent> logAppender = LoggingUtils.getLogListAppender(ApplicationConfig.class);
124 Flux<ApplicationConfig> task = appConfigUnderTest.createRefreshTask();
126 StepVerifier.create(task).expectSubscription().verifyComplete();
128 assertTrue(logAppender.list.toString().contains("$CONSUL_HOST environment has not been defined"));
132 public void whenPeriodicConfigRefreshNoConsul() {
133 appConfigUnderTest = spy(ApplicationConfig.class);
134 appConfigUnderTest.systemEnvironment = new Properties();
136 EnvProperties props = properties();
137 doReturn(Mono.just(props)).when(appConfigUnderTest).getEnvironment(any());
139 doReturn(Mono.just(cbsClient)).when(appConfigUnderTest).createCbsClient(props);
140 Flux<JsonObject> err = Flux.error(new IOException());
141 doReturn(err).when(cbsClient).updates(any(), any(), any());
143 final ListAppender<ILoggingEvent> logAppender = LoggingUtils.getLogListAppender(ApplicationConfig.class);
144 Flux<ApplicationConfig> task = appConfigUnderTest.createRefreshTask();
148 .expectSubscription() //
152 logAppender.list.toString().contains("Could not refresh application configuration java.io.IOException"));
156 public void whenPeriodicConfigRefreshSuccess() throws JsonIOException, JsonSyntaxException, IOException {
157 appConfigUnderTest = spy(ApplicationConfig.class);
158 appConfigUnderTest.systemEnvironment = new Properties();
160 EnvProperties props = properties();
161 doReturn(Mono.just(props)).when(appConfigUnderTest).getEnvironment(any());
162 doReturn(Mono.just(cbsClient)).when(appConfigUnderTest).createCbsClient(props);
164 Flux<JsonObject> json = Flux.just(getJsonRootObject());
165 doReturn(json).when(cbsClient).updates(any(), any(), any());
167 Flux<ApplicationConfig> task = appConfigUnderTest.createRefreshTask();
171 .expectSubscription() //
172 .expectNext(appConfigUnderTest) //
175 Assertions.assertNotNull(appConfigUnderTest.getRicConfigs());
178 private JsonObject getJsonRootObject() throws JsonIOException, JsonSyntaxException, IOException {
179 JsonObject rootObject = (new JsonParser()).parse(new InputStreamReader(getCorrectJson())).getAsJsonObject();
183 private static InputStream getCorrectJson() throws IOException {
184 URL url = ApplicationConfigParser.class.getClassLoader().getResource("test_application_configuration.json");
185 String string = Resources.toString(url, Charsets.UTF_8);
186 return new ByteArrayInputStream((string.getBytes(StandardCharsets.UTF_8)));
189 private static InputStream getIncorrectJson() {
190 String string = "{" + //
191 " \"config\": {" + //
193 return new ByteArrayInputStream((string.getBytes(StandardCharsets.UTF_8)));