policy-agent/src/test/java/org/oransc/policyagent/clients/SdncOscA1ClientTest.java

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2020 Nordix Foundation
   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
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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===================================
  19  */
  20
  21 package org.oransc.policyagent.clients;
  22
  23 import static org.mockito.ArgumentMatchers.anyString;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.verify;
  26 import static org.mockito.Mockito.when;
  27
  28 import java.util.Arrays;
  29 import java.util.List;
  30
  31 import org.json.JSONException;
  32 import org.junit.jupiter.api.BeforeEach;
  33 import org.junit.jupiter.api.Test;
  34 import org.junit.jupiter.api.extension.ExtendWith;
  35 import org.mockito.junit.jupiter.MockitoExtension;
  36 import org.mockito.stubbing.OngoingStubbing;
  37
  38 import reactor.core.publisher.Flux;
  39 import reactor.core.publisher.Mono;
  40 import reactor.test.StepVerifier;
  41
  42 @ExtendWith(MockitoExtension.class)
  43 public class SdncOscA1ClientTest {
  44     private static final String CONTROLLER_USERNAME = "username";
  45     private static final String CONTROLLER_PASSWORD = "password";
  46     private static final String RIC_1_URL = "RicUrl";
  47     private static final String POLICYTYPES_IDENTITIES_URL = "/A1-ADAPTER-API:getPolicyTypeIdentities";
  48     private static final String POLICIES_IDENTITIES_URL = "/A1-ADAPTER-API:getPolicyIdentities";
  49     private static final String POLICYTYPES_URL = "/A1-ADAPTER-API:getPolicyType";
  50     private static final String PUT_POLICY_URL = "/A1-ADAPTER-API:putPolicy";
  51     private static final String DELETE_POLICY_URL = "/A1-ADAPTER-API:deletePolicy";
  52
  53     private static final String POLICY_TYPE_1_ID = "type1";
  54     private static final String POLICY_TYPE_2_ID = "type2";
  55     private static final String POLICY_TYPE_SCHEMA_VALID = "{\"type\":\"type1\"}";
  56     private static final String POLICY_TYPE_SCHEMA_INVALID = "\"type\":\"type1\"}";
  57     private static final String POLICY_1_ID = "policy1";
  58     private static final String POLICY_2_ID = "policy2";
  59     private static final String POLICY_JSON_VALID = "{\"scope\":{\"ueId\":\"ue1\"}}";
  60     private static final String POLICY_JSON_INVALID = "\"scope\":{\"ueId\":\"ue1\"}}";
  61
  62     SdncOscA1Client clientUnderTest;
  63
  64     AsyncRestClient asyncRestClientMock;
  65
  66     @BeforeEach
  67     public void init() {
  68         asyncRestClientMock = mock(AsyncRestClient.class);
  69         clientUnderTest = new SdncOscA1Client(A1ClientHelper.createRic(RIC_1_URL).getConfig(), CONTROLLER_USERNAME,
  70             CONTROLLER_PASSWORD, asyncRestClientMock);
  71     }
  72
  73     @Test
  74     public void testGetPolicyTypeIdentities() {
  75         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
  76             .nearRtRicUrl(RIC_1_URL) //
  77             .build();
  78         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
  79
  80         List<String> policyTypeIds = Arrays.asList(POLICY_TYPE_1_ID, POLICY_TYPE_2_ID);
  81         Mono<String> policyTypeIdsResp =
  82             A1ClientHelper.createOutputJsonResponse("policy-type-id-list", policyTypeIds.toString());
  83         whenAsyncPostThenReturn(policyTypeIdsResp);
  84
  85         Mono<List<String>> returnedMono = clientUnderTest.getPolicyTypeIdentities();
  86         verify(asyncRestClientMock).postWithAuthHeader(POLICYTYPES_IDENTITIES_URL, inputJsonString, CONTROLLER_USERNAME,
  87             CONTROLLER_PASSWORD);
  88         StepVerifier.create(returnedMono).expectNext(policyTypeIds).expectComplete().verify();
  89     }
  90
  91     @Test
  92     public void testGetPolicyIdentities() {
  93         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
  94             .nearRtRicUrl(RIC_1_URL) //
  95             .build();
  96         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
  97
  98         List<String> policyIds = Arrays.asList(POLICY_1_ID, POLICY_2_ID);
  99         Mono<String> policyIdsResp = A1ClientHelper.createOutputJsonResponse("policy-id-list", policyIds.toString());
 100         whenAsyncPostThenReturn(policyIdsResp);
 101
 102         Mono<List<String>> returnedMono = clientUnderTest.getPolicyIdentities();
 103         verify(asyncRestClientMock).postWithAuthHeader(POLICIES_IDENTITIES_URL, inputJsonString, CONTROLLER_USERNAME,
 104             CONTROLLER_PASSWORD);
 105         StepVerifier.create(returnedMono).expectNext(policyIds).expectComplete().verify();
 106     }
 107
 108     @Test
 109     public void testGetValidPolicyType() {
 110         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 111             .nearRtRicUrl(RIC_1_URL) //
 112             .policyTypeId(POLICY_TYPE_1_ID) //
 113             .build();
 114         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 115
 116         String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_VALID + ", \"statusSchema\": {} }";
 117         Mono<String> policyTypeResp = A1ClientHelper.createOutputJsonResponse("policy-type", policyType);
 118         whenAsyncPostThenReturn(policyTypeResp);
 119
 120         Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
 121         verify(asyncRestClientMock).postWithAuthHeader(POLICYTYPES_URL, inputJsonString, CONTROLLER_USERNAME,
 122             CONTROLLER_PASSWORD);
 123         StepVerifier.create(returnedMono).expectNext(POLICY_TYPE_SCHEMA_VALID).expectComplete().verify();
 124     }
 125
 126     @Test
 127     public void testGetInvalidPolicyType() {
 128         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 129             .nearRtRicUrl(RIC_1_URL) //
 130             .policyTypeId(POLICY_TYPE_1_ID) //
 131             .build();
 132         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 133
 134         String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_INVALID + ", \"statusSchema\": {} }";
 135         Mono<String> policyTypeResp = A1ClientHelper.createOutputJsonResponse("policy-type", policyType);
 136         whenAsyncPostThenReturn(policyTypeResp);
 137
 138         Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
 139         verify(asyncRestClientMock).postWithAuthHeader(POLICYTYPES_URL, inputJsonString, CONTROLLER_USERNAME,
 140             CONTROLLER_PASSWORD);
 141         StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof JSONException).verify();
 142     }
 143
 144     @Test
 145     public void testPutPolicyValidResponse() {
 146         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 147             .nearRtRicUrl(RIC_1_URL) //
 148             .policyTypeId(POLICY_TYPE_1_ID) //
 149             .policyId(POLICY_1_ID) //
 150             .policy(POLICY_JSON_VALID) //
 151             .build();
 152         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 153
 154         Mono<String> policyResp = A1ClientHelper.createOutputJsonResponse("returned-policy", POLICY_JSON_VALID);
 155         whenAsyncPostThenReturn(policyResp);
 156
 157         Mono<String> returnedMono = clientUnderTest
 158             .putPolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 159         verify(asyncRestClientMock).postWithAuthHeader(PUT_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 160             CONTROLLER_PASSWORD);
 161         StepVerifier.create(returnedMono).expectNext(POLICY_JSON_VALID).expectComplete().verify();
 162     }
 163
 164     @Test
 165     public void testPutPolicyInvalidResponse() {
 166         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 167             .nearRtRicUrl(RIC_1_URL) //
 168             .policyTypeId(POLICY_TYPE_1_ID) //
 169             .policyId(POLICY_1_ID) //
 170             .policy(POLICY_JSON_VALID) //
 171             .build();
 172         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 173
 174         Mono<String> policyResp = A1ClientHelper.createOutputJsonResponse("returned-policy", POLICY_JSON_INVALID);
 175         whenAsyncPostThenReturn(policyResp);
 176
 177         Mono<String> returnedMono = clientUnderTest
 178             .putPolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 179         verify(asyncRestClientMock).postWithAuthHeader(PUT_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 180             CONTROLLER_PASSWORD);
 181         StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof JSONException).verify();
 182     }
 183
 184     @Test
 185     public void testDeletePolicy() {
 186         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 187             .nearRtRicUrl(RIC_1_URL) //
 188             .policyId(POLICY_1_ID) //
 189             .build();
 190         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 191
 192         whenAsyncPostThenReturn(Mono.empty());
 193
 194         Mono<String> returnedMono = clientUnderTest
 195             .deletePolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 196         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 197             CONTROLLER_PASSWORD);
 198         StepVerifier.create(returnedMono).expectComplete().verify();
 199     }
 200
 201     @Test
 202     public void testDeleteAllPolicies() {
 203         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 204             .nearRtRicUrl(RIC_1_URL) //
 205             .build();
 206         String inputJsonStringGetIds = A1ClientHelper.createInputJsonString(inputParams);
 207         inputParams = ImmutableSdncOscAdapterInput.builder() //
 208             .nearRtRicUrl(RIC_1_URL) //
 209             .policyId(POLICY_1_ID) //
 210             .build();
 211         String inputJsonStringDeletePolicy1 = A1ClientHelper.createInputJsonString(inputParams);
 212         inputParams = ImmutableSdncOscAdapterInput.builder() //
 213             .nearRtRicUrl(RIC_1_URL) //
 214             .policyId(POLICY_2_ID) //
 215             .build();
 216         String inputJsonStringDeletePolicy2 = A1ClientHelper.createInputJsonString(inputParams);
 217
 218         List<String> policyIds = Arrays.asList(POLICY_1_ID, POLICY_2_ID);
 219         Mono<String> policyIdsResp = A1ClientHelper.createOutputJsonResponse("policy-id-list", policyIds.toString());
 220         whenAsyncPostThenReturn(policyIdsResp).thenReturn(Mono.empty());
 221
 222         Flux<String> returnedFlux = clientUnderTest.deleteAllPolicies();
 223         StepVerifier.create(returnedFlux).expectComplete().verify();
 224         verify(asyncRestClientMock).postWithAuthHeader(POLICIES_IDENTITIES_URL, inputJsonStringGetIds,
 225             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 226         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonStringDeletePolicy1,
 227             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 228         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonStringDeletePolicy2,
 229             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 230     }
 231
 232     private OngoingStubbing<Mono<String>> whenAsyncPostThenReturn(Mono<String> response) {
 233         return when(asyncRestClientMock.postWithAuthHeader(anyString(), anyString(), anyString(), anyString()))
 234             .thenReturn(response);
 235     }
 236 }