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 POLICY_TYPES_IDENTITIES_URL = "/A1-ADAPTER-API:getPolicyTypeIdentities";
  48     private static final String POLICY_IDENTITIES_URL = "/A1-ADAPTER-API:getPolicyIdentities";
  49     private static final String POLICY_TYPES_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
  87         verify(asyncRestClientMock).postWithAuthHeader(POLICY_TYPES_IDENTITIES_URL, inputJsonString,
  88             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
  89         StepVerifier.create(returnedMono).expectNext(policyTypeIds).expectComplete().verify();
  90     }
  91
  92     @Test
  93     public void testGetPolicyIdentities() {
  94         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
  95             .nearRtRicUrl(RIC_1_URL) //
  96             .build();
  97         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
  98
  99         List<String> policyIds = Arrays.asList(POLICY_1_ID, POLICY_2_ID);
 100         Mono<String> policyIdsResp = A1ClientHelper.createOutputJsonResponse("policy-id-list", policyIds.toString());
 101         whenAsyncPostThenReturn(policyIdsResp);
 102
 103         Mono<List<String>> returnedMono = clientUnderTest.getPolicyIdentities();
 104
 105         verify(asyncRestClientMock).postWithAuthHeader(POLICY_IDENTITIES_URL, inputJsonString, CONTROLLER_USERNAME,
 106             CONTROLLER_PASSWORD);
 107         StepVerifier.create(returnedMono).expectNext(policyIds).expectComplete().verify();
 108     }
 109
 110     @Test
 111     public void testGetValidPolicyType() {
 112         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 113             .nearRtRicUrl(RIC_1_URL) //
 114             .policyTypeId(POLICY_TYPE_1_ID) //
 115             .build();
 116         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 117
 118         String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_VALID + ", \"statusSchema\": {} }";
 119         Mono<String> policyTypeResp = A1ClientHelper.createOutputJsonResponse("policy-type", policyType);
 120         whenAsyncPostThenReturn(policyTypeResp);
 121
 122         Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
 123
 124         verify(asyncRestClientMock).postWithAuthHeader(POLICY_TYPES_URL, inputJsonString, CONTROLLER_USERNAME,
 125             CONTROLLER_PASSWORD);
 126         StepVerifier.create(returnedMono).expectNext(POLICY_TYPE_SCHEMA_VALID).expectComplete().verify();
 127     }
 128
 129     @Test
 130     public void testGetInvalidPolicyType() {
 131         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 132             .nearRtRicUrl(RIC_1_URL) //
 133             .policyTypeId(POLICY_TYPE_1_ID) //
 134             .build();
 135         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 136
 137         String policyType = "{\"policySchema\": " + POLICY_TYPE_SCHEMA_INVALID + ", \"statusSchema\": {} }";
 138         Mono<String> policyTypeResp = A1ClientHelper.createOutputJsonResponse("policy-type", policyType);
 139         whenAsyncPostThenReturn(policyTypeResp);
 140
 141         Mono<String> returnedMono = clientUnderTest.getPolicyTypeSchema(POLICY_TYPE_1_ID);
 142
 143         verify(asyncRestClientMock).postWithAuthHeader(POLICY_TYPES_URL, inputJsonString, CONTROLLER_USERNAME,
 144             CONTROLLER_PASSWORD);
 145         StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof JSONException).verify();
 146     }
 147
 148     @Test
 149     public void testPutPolicyValidResponse() {
 150         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 151             .nearRtRicUrl(RIC_1_URL) //
 152             .policyTypeId(POLICY_TYPE_1_ID) //
 153             .policyId(POLICY_1_ID) //
 154             .policy(POLICY_JSON_VALID) //
 155             .build();
 156         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 157
 158         Mono<String> policyResp = A1ClientHelper.createOutputJsonResponse("returned-policy", POLICY_JSON_VALID);
 159         whenAsyncPostThenReturn(policyResp);
 160
 161         Mono<String> returnedMono = clientUnderTest
 162             .putPolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 163
 164         verify(asyncRestClientMock).postWithAuthHeader(PUT_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 165             CONTROLLER_PASSWORD);
 166         StepVerifier.create(returnedMono).expectNext(POLICY_JSON_VALID).expectComplete().verify();
 167     }
 168
 169     @Test
 170     public void testPutPolicyInvalidResponse() {
 171         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 172             .nearRtRicUrl(RIC_1_URL) //
 173             .policyTypeId(POLICY_TYPE_1_ID) //
 174             .policyId(POLICY_1_ID) //
 175             .policy(POLICY_JSON_VALID) //
 176             .build();
 177         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 178
 179         Mono<String> policyResp = A1ClientHelper.createOutputJsonResponse("returned-policy", POLICY_JSON_INVALID);
 180         whenAsyncPostThenReturn(policyResp);
 181
 182         Mono<String> returnedMono = clientUnderTest
 183             .putPolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 184
 185         verify(asyncRestClientMock).postWithAuthHeader(PUT_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 186             CONTROLLER_PASSWORD);
 187         StepVerifier.create(returnedMono).expectErrorMatches(throwable -> throwable instanceof JSONException).verify();
 188     }
 189
 190     @Test
 191     public void testDeletePolicy() {
 192         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 193             .nearRtRicUrl(RIC_1_URL) //
 194             .policyId(POLICY_1_ID) //
 195             .build();
 196         String inputJsonString = A1ClientHelper.createInputJsonString(inputParams);
 197
 198         whenAsyncPostThenReturn(Mono.empty());
 199
 200         Mono<String> returnedMono = clientUnderTest
 201             .deletePolicy(A1ClientHelper.createPolicy(RIC_1_URL, POLICY_1_ID, POLICY_JSON_VALID, POLICY_TYPE_1_ID));
 202
 203         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonString, CONTROLLER_USERNAME,
 204             CONTROLLER_PASSWORD);
 205         StepVerifier.create(returnedMono).expectComplete().verify();
 206     }
 207
 208     @Test
 209     public void testDeleteAllPolicies() {
 210         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
 211             .nearRtRicUrl(RIC_1_URL) //
 212             .build();
 213         String inputJsonStringGetIds = A1ClientHelper.createInputJsonString(inputParams);
 214         inputParams = ImmutableSdncOscAdapterInput.builder() //
 215             .nearRtRicUrl(RIC_1_URL) //
 216             .policyId(POLICY_1_ID) //
 217             .build();
 218         String inputJsonStringDeletePolicy1 = A1ClientHelper.createInputJsonString(inputParams);
 219         inputParams = ImmutableSdncOscAdapterInput.builder() //
 220             .nearRtRicUrl(RIC_1_URL) //
 221             .policyId(POLICY_2_ID) //
 222             .build();
 223         String inputJsonStringDeletePolicy2 = A1ClientHelper.createInputJsonString(inputParams);
 224
 225         List<String> policyIds = Arrays.asList(POLICY_1_ID, POLICY_2_ID);
 226         Mono<String> policyIdsResp = A1ClientHelper.createOutputJsonResponse("policy-id-list", policyIds.toString());
 227         whenAsyncPostThenReturn(policyIdsResp).thenReturn(Mono.empty());
 228
 229         Flux<String> returnedFlux = clientUnderTest.deleteAllPolicies();
 230
 231         StepVerifier.create(returnedFlux).expectComplete().verify();
 232         verify(asyncRestClientMock).postWithAuthHeader(POLICY_IDENTITIES_URL, inputJsonStringGetIds,
 233             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 234         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonStringDeletePolicy1,
 235             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 236         verify(asyncRestClientMock).postWithAuthHeader(DELETE_POLICY_URL, inputJsonStringDeletePolicy2,
 237             CONTROLLER_USERNAME, CONTROLLER_PASSWORD);
 238     }
 239
 240     private OngoingStubbing<Mono<String>> whenAsyncPostThenReturn(Mono<String> response) {
 241         return when(asyncRestClientMock.postWithAuthHeader(anyString(), anyString(), anyString(), anyString()))
 242             .thenReturn(response);
 243     }
 244 }