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.tasks;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.mockito.ArgumentMatchers.any;
25 import static org.mockito.Mockito.doReturn;
26 import static org.mockito.Mockito.spy;
27 import static org.mockito.Mockito.times;
28 import static org.mockito.Mockito.verify;
30 import ch.qos.logback.classic.spi.ILoggingEvent;
31 import ch.qos.logback.core.read.ListAppender;
33 import com.google.common.base.Charsets;
34 import com.google.common.io.Resources;
35 import com.google.gson.JsonIOException;
36 import com.google.gson.JsonObject;
37 import com.google.gson.JsonParser;
38 import com.google.gson.JsonSyntaxException;
40 import java.io.ByteArrayInputStream;
41 import java.io.IOException;
42 import java.io.InputStream;
43 import java.io.InputStreamReader;
45 import java.nio.charset.StandardCharsets;
46 import java.time.Duration;
47 import java.util.Arrays;
48 import java.util.Properties;
49 import java.util.Vector;
51 import org.junit.jupiter.api.Test;
52 import org.junit.jupiter.api.extension.ExtendWith;
53 import org.mockito.Mock;
54 import org.mockito.Spy;
55 import org.mockito.junit.jupiter.MockitoExtension;
56 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClient;
57 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.EnvProperties;
58 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.ImmutableEnvProperties;
59 import org.oransc.policyagent.clients.A1ClientFactory;
60 import org.oransc.policyagent.configuration.ApplicationConfig;
61 import org.oransc.policyagent.configuration.ApplicationConfigParser;
62 import org.oransc.policyagent.configuration.ImmutableRicConfig;
63 import org.oransc.policyagent.configuration.RicConfig;
64 import org.oransc.policyagent.repository.Policies;
65 import org.oransc.policyagent.repository.PolicyTypes;
66 import org.oransc.policyagent.repository.Rics;
67 import org.oransc.policyagent.repository.Services;
68 import org.oransc.policyagent.utils.LoggingUtils;
70 import reactor.core.publisher.Flux;
71 import reactor.core.publisher.Mono;
72 import reactor.test.StepVerifier;
74 @ExtendWith(MockitoExtension.class)
75 public class RefreshConfigTaskTest {
77 private RefreshConfigTask refreshTaskUnderTest;
80 ApplicationConfig appConfig;
85 private static final String RIC_1_NAME = "ric1";
86 public static final ImmutableRicConfig CORRECT_RIC_CONIFG = ImmutableRicConfig.builder() //
88 .baseUrl("http://localhost:8080/") //
89 .managedElementIds(new Vector<String>(Arrays.asList("kista_1", "kista_2"))) //
92 private static EnvProperties properties() {
93 return ImmutableEnvProperties.builder() //
94 .consulHost("host") //
96 .cbsName("cbsName") //
97 .appName("appName") //
101 private RefreshConfigTask createTestObject(boolean configFileExists) {
102 RefreshConfigTask obj = spy(new RefreshConfigTask(appConfig, new Rics(), new Policies(), new Services(),
103 new PolicyTypes(), new A1ClientFactory(appConfig)));
104 doReturn(configFileExists).when(obj).configFileExists();
109 public void whenTheConfigurationFits_thenConfiguredRicsArePutInRepository() throws Exception {
110 refreshTaskUnderTest = this.createTestObject(true);
111 refreshTaskUnderTest.systemEnvironment = new Properties();
113 doReturn(getCorrectJson()).when(refreshTaskUnderTest).createInputStream(any());
114 doReturn("fileName").when(appConfig).getLocalConfigurationFilePath();
116 StepVerifier.create(refreshTaskUnderTest.createRefreshTask()) //
117 .expectSubscription() //
118 .expectNext(this.appConfig) //
119 .expectNext(this.appConfig) //
124 verify(refreshTaskUnderTest, times(1)).loadConfigurationFromFile();
126 Iterable<RicConfig> ricConfigs = appConfig.getRicConfigs();
127 RicConfig ricConfig = ricConfigs.iterator().next();
128 assertThat(ricConfigs).isNotNull();
129 assertThat(ricConfig).isEqualTo(CORRECT_RIC_CONIFG);
133 public void whenFileExistsButJsonIsIncorrect_thenNoRicsArePutInRepository() throws Exception {
134 refreshTaskUnderTest = this.createTestObject(true);
135 refreshTaskUnderTest.systemEnvironment = new Properties();
138 doReturn(getIncorrectJson()).when(refreshTaskUnderTest).createInputStream(any());
139 doReturn("fileName").when(appConfig).getLocalConfigurationFilePath();
141 StepVerifier.create(refreshTaskUnderTest.createRefreshTask()) //
142 .expectSubscription() //
143 .expectNoEvent(Duration.ofMillis(100)) //
148 verify(refreshTaskUnderTest, times(1)).loadConfigurationFromFile();
149 assertThat(appConfig.getRicConfigs().size()).isEqualTo(0);
153 public void whenPeriodicConfigRefreshNoConsul_thenErrorIsLogged() {
154 refreshTaskUnderTest = this.createTestObject(false);
155 refreshTaskUnderTest.systemEnvironment = new Properties();
157 EnvProperties props = properties();
158 doReturn(Mono.just(props)).when(refreshTaskUnderTest).getEnvironment(any());
160 doReturn(Mono.just(cbsClient)).when(refreshTaskUnderTest).createCbsClient(props);
161 Flux<?> err = Flux.error(new IOException());
162 doReturn(err).when(cbsClient).updates(any(), any(), any());
164 final ListAppender<ILoggingEvent> logAppender = LoggingUtils.getLogListAppender(RefreshConfigTask.class);
165 Flux<?> task = refreshTaskUnderTest.createRefreshTask();
169 .expectSubscription() //
170 .expectNoEvent(Duration.ofMillis(100)) //
175 logAppender.list.toString().contains("Could not refresh application configuration. java.io.IOException"))
180 public void whenPeriodicConfigRefreshSuccess_thenNewConfigIsCreated() throws Exception {
181 refreshTaskUnderTest = this.createTestObject(false);
182 refreshTaskUnderTest.systemEnvironment = new Properties();
184 EnvProperties props = properties();
185 doReturn(Mono.just(props)).when(refreshTaskUnderTest).getEnvironment(any());
186 doReturn(Mono.just(cbsClient)).when(refreshTaskUnderTest).createCbsClient(props);
188 JsonObject configAsJson = getJsonRootObject();
189 String newBaseUrl = "newBaseUrl";
190 modifyTheRicConfiguration(configAsJson, newBaseUrl);
191 Flux<JsonObject> json = Flux.just(configAsJson);
192 doReturn(json).when(cbsClient).updates(any(), any(), any());
194 Flux<ApplicationConfig> task = refreshTaskUnderTest.createRefreshTask();
198 .expectSubscription() //
199 .expectNext(appConfig) //
200 .expectNext(appConfig) //
204 verify(refreshTaskUnderTest, times(2)).runRicSynchronization(any());
205 assertThat(appConfig.getRicConfigs()).isNotNull();
206 assertThat(appConfig.getRic(RIC_1_NAME).baseUrl()).isEqualTo(newBaseUrl);
209 private void modifyTheRicConfiguration(JsonObject configAsJson, String newBaseUrl) {
210 ((JsonObject) configAsJson.getAsJsonObject("config").getAsJsonArray("ric").get(0)).addProperty("baseUrl",
214 private JsonObject getJsonRootObject() throws JsonIOException, JsonSyntaxException, IOException {
215 JsonObject rootObject = JsonParser.parseReader(new InputStreamReader(getCorrectJson())).getAsJsonObject();
219 private static InputStream getCorrectJson() throws IOException {
220 URL url = ApplicationConfigParser.class.getClassLoader().getResource("test_application_configuration.json");
221 String string = Resources.toString(url, Charsets.UTF_8);
222 return new ByteArrayInputStream((string.getBytes(StandardCharsets.UTF_8)));
225 private static InputStream getIncorrectJson() {
226 String string = "{" + //
227 " \"config\": {" + //
229 return new ByteArrayInputStream((string.getBytes(StandardCharsets.UTF_8)));