2 * ========================LICENSE_START=================================
4 * ======================================================================
5 * Copyright (C) 2020 Nordix Foundation. All rights reserved.
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
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.onap.ccsdk.oran.a1policymanagementservice.clients;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.junit.jupiter.api.Assertions.assertEquals;
25 import static org.mockito.ArgumentMatchers.anyString;
26 import static org.mockito.ArgumentMatchers.eq;
27 import static org.mockito.Mockito.mock;
28 import static org.mockito.Mockito.verify;
29 import static org.mockito.Mockito.when;
31 import java.util.Arrays;
32 import java.util.List;
34 import org.json.JSONException;
35 import org.junit.jupiter.api.BeforeEach;
36 import org.junit.jupiter.api.Test;
37 import org.junit.jupiter.api.extension.ExtendWith;
38 import org.mockito.ArgumentCaptor;
39 import org.mockito.junit.jupiter.MockitoExtension;
40 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.RicConfig;
42 import reactor.core.publisher.Flux;
43 import reactor.core.publisher.Mono;
44 import reactor.test.StepVerifier;
46 @ExtendWith(MockitoExtension.class)
47 class StdA1ClientV2Test {
49 private static final String RIC_URL = "https://ric.com";
51 private static final String RIC_BASE_URL = RIC_URL + "/A1-P/v2";
53 private static final String POLICYTYPES_IDENTITIES_URL = RIC_BASE_URL + "/policytypes";
54 private static final String POLICIES = "/policies";
55 private static final String POLICYTYPES_URL = RIC_BASE_URL + "/policytypes/";
56 private static final String POLICY_TYPE_1_ID = "type1";
57 private static final String POLICY_TYPE_2_ID = "type2";
58 private static final String POLICY_TYPE_SCHEMA_VALID = "{\"type\":\"type1\"}";
59 private static final String POLICY_TYPE_SCHEMA_INVALID = "\"type\":\"type1\"}";
60 private static final String POLICY_1_ID = "policy1";
61 private static final String POLICY_2_ID = "policy2";
62 private static final String POLICY_JSON_VALID = "{\"policyId\":\"policy1\"}";
64 StdA1ClientVersion2 clientUnderTest;
66 AsyncRestClient asyncRestClientMock;
70 RicConfig ricConfig = RicConfig.builder() //
74 asyncRestClientMock = mock(AsyncRestClient.class);
75 clientUnderTest = new StdA1ClientVersion2(ricConfig, asyncRestClientMock);
79 void testGetPolicyTypeIdentities() {
80 List<String> policyTypeIds = Arrays.asList(POLICY_TYPE_1_ID, POLICY_TYPE_2_ID);
81 Mono<String> policyTypeIdsResp = Mono.just(policyTypeIds.toString());
82 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeIdsResp);
84 Mono<List<String>> returnedMono = clientUnderTest.getPolicyTypeIdentities();
85 verify(asyncRestClientMock).get(POLICYTYPES_IDENTITIES_URL);
86 StepVerifier.create(returnedMono).expectNext(policyTypeIds).expectComplete().verify();
90 void testGetPolicyIdentities() {
91 Mono<String> policyTypeIdsResp = Mono.just(Arrays.asList(POLICY_TYPE_1_ID, POLICY_TYPE_2_ID).toString());
92 Mono<String> policyIdsType1Resp = Mono.just(Arrays.asList(POLICY_1_ID).toString());
93 Mono<String> policyIdsType2Resp = Mono.just(Arrays.asList(POLICY_2_ID).toString());
94 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeIdsResp).thenReturn(policyIdsType1Resp)
95 .thenReturn(policyIdsType2Resp);
97 List<String> returned = clientUnderTest.getPolicyIdentities().block();
99 assertEquals(2, returned.size(), "");
100 verify(asyncRestClientMock).get(POLICYTYPES_IDENTITIES_URL);
101 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_1_ID + POLICIES);
102 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_2_ID + POLICIES);
106 void testGetValidPolicyType() {
107 String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_VALID + "}";
108 Mono<String> policyTypeResp = Mono.just(policyType);
110 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeResp);
112 String response = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID).block();
113 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_1_ID);
114 assertThat(response).contains("title");
118 void testGetInValidPolicyTypeJson() {
119 String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_INVALID + "}";
120 Mono<String> policyTypeResp = Mono.just(policyType);
122 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeResp);
124 Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
125 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_1_ID);
126 StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof JSONException).verify();
130 void testGetPolicyTypeWithoutCreateSchema() {
131 Mono<String> policyTypeResp = Mono.just(POLICY_TYPE_SCHEMA_VALID);
133 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeResp);
135 Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
136 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_1_ID);
137 StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof Exception).verify();
141 void testPutPolicy() {
142 when(asyncRestClientMock.put(anyString(), anyString())).thenReturn(Mono.empty());
145 .putPolicy(A1ClientHelper.createPolicy(RIC_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID))
147 ArgumentCaptor<String> urlCaptor = ArgumentCaptor.forClass(String.class);
148 verify(asyncRestClientMock).put(urlCaptor.capture(), eq(POLICY_JSON_VALID));
149 String actualUrl = urlCaptor.getValue();
150 String expUrl = POLICYTYPES_URL + POLICY_TYPE_1_ID + POLICIES + "/" + POLICY_1_ID;
151 assertThat(actualUrl).contains(expUrl);
155 void testDeletePolicy() {
156 when(asyncRestClientMock.delete(anyString())).thenReturn(Mono.empty());
158 Mono<String> returnedMono = clientUnderTest
159 .deletePolicy(A1ClientHelper.createPolicy(RIC_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
160 verify(asyncRestClientMock).delete(POLICYTYPES_URL + POLICY_TYPE_1_ID + POLICIES + "/" + POLICY_1_ID);
161 StepVerifier.create(returnedMono).expectComplete().verify();
165 void testDeleteAllPolicies() {
166 Mono<String> policyTypeIdsResp = Mono.just(Arrays.asList(POLICY_TYPE_1_ID, POLICY_TYPE_2_ID).toString());
167 Mono<String> policyIdsType1Resp = Mono.just(Arrays.asList(POLICY_1_ID).toString());
168 Mono<String> policyIdsType2Resp = Mono.just(Arrays.asList(POLICY_2_ID).toString());
169 when(asyncRestClientMock.get(anyString())).thenReturn(policyTypeIdsResp).thenReturn(policyIdsType1Resp)
170 .thenReturn(policyIdsType2Resp);
171 when(asyncRestClientMock.delete(anyString())).thenReturn(Mono.empty());
173 Flux<String> returnedFlux = clientUnderTest.deleteAllPolicies();
174 StepVerifier.create(returnedFlux).expectComplete().verify();
175 verify(asyncRestClientMock).get(POLICYTYPES_IDENTITIES_URL);
176 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_1_ID + POLICIES);
177 verify(asyncRestClientMock).delete(POLICYTYPES_URL + POLICY_TYPE_1_ID + POLICIES + "/" + POLICY_1_ID);
178 verify(asyncRestClientMock).get(POLICYTYPES_URL + POLICY_TYPE_2_ID + POLICIES);
179 verify(asyncRestClientMock).delete(POLICYTYPES_URL + POLICY_TYPE_2_ID + POLICIES + "/" + POLICY_2_ID);