- Assert.assertEquals("Policy type 1", result.get().getResult().getName());
- Assert.assertEquals(String.valueOf(HttpStatus.OK.value()), result.get().getResult().getCode());
- }
-
- @Test
- public void testGetPolicyTypes() throws InterruptedException, ExecutionException {
- GetPolicyTypesInputBuilder inputBuilder = new GetPolicyTypesInputBuilder();
- inputBuilder.setNearRtRicId(nearRtRicId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri = nearRicUrlProvider.getPolicyTypes(inputBuilder.build().getNearRtRicId());
- List<Integer> policyTypesInteger = new ArrayList<>();
- policyTypesInteger.add(20001);
- List<Long> policyTypesLong = new ArrayList<>();
- policyTypesLong.add(20001L);
- ResponseEntity<Object> getPolicyTypesResponse = new ResponseEntity<>(policyTypesInteger, HttpStatus.OK);
- when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(getPolicyTypesResponse);
- ListenableFuture<RpcResult<GetPolicyTypesOutput>> result =
- nonrtRicApiProvider.getPolicyTypes(inputBuilder.build());
- Assert.assertEquals(policyTypesLong, result.get().getResult().getPolicyTypeIdList());
- Assert.assertEquals(String.valueOf(HttpStatus.OK.value()), result.get().getResult().getCode());
- }
-
- @Test
- public void testGetPolicyInstance() throws InterruptedException, ExecutionException {
- GetPolicyInstanceInputBuilder inputBuilder = new GetPolicyInstanceInputBuilder();
- inputBuilder.setNearRtRicId(nearRtRicId);
- inputBuilder.setPolicyTypeId(policyTypeId);
- inputBuilder.setPolicyInstanceId(policyTypeInstanceId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri = nearRicUrlProvider.getPolicyInstanceId(inputBuilder.build().getNearRtRicId(),
- String.valueOf(inputBuilder.build().getPolicyTypeId()), inputBuilder.getPolicyInstanceId());
- String policyInstance =
- "{\"scope\":{\"ue_id\":\"2\"},\"statement\":{\"priority_level\":\"1\"},\"policy_id\":\"pi12\"}";
- ResponseEntity<Object> getPolicyInstanceResponse = new ResponseEntity<>(policyInstance, HttpStatus.OK);
- when(restAdapter.get(eq(uri), eq(String.class))).thenReturn(getPolicyInstanceResponse);
- ListenableFuture<RpcResult<GetPolicyInstanceOutput>> result =
- nonrtRicApiProvider.getPolicyInstance(inputBuilder.build());
- Assert.assertEquals(policyInstance, result.get().getResult().getPolicyInstance());
- Assert.assertEquals(String.valueOf(HttpStatus.OK.value()), result.get().getResult().getCode());
- }
-
- @Test
- public void testGetPolicyInstances() throws InterruptedException, ExecutionException {
- GetPolicyInstancesInputBuilder inputBuilder = new GetPolicyInstancesInputBuilder();
- inputBuilder.setNearRtRicId(nearRtRicId);
- inputBuilder.setPolicyTypeId(policyTypeId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri = nearRicUrlProvider.getPolicyInstances(inputBuilder.build().getNearRtRicId(),
- String.valueOf(inputBuilder.build().getPolicyTypeId()));
- List<String> policyInstances = new ArrayList<>();
- policyInstances.add("3d2157af-6a8f-4a7c-810f-38c2f824bf12");
- ResponseEntity<Object> getPolicyInstancesResponse = new ResponseEntity<>(policyInstances, HttpStatus.OK);
- when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(getPolicyInstancesResponse);
- ListenableFuture<RpcResult<GetPolicyInstancesOutput>> result =
- nonrtRicApiProvider.getPolicyInstances(inputBuilder.build());
- Assert.assertEquals(policyInstances, result.get().getResult().getPolicyInstanceIdList());
- Assert.assertEquals(String.valueOf(HttpStatus.OK.value()), result.get().getResult().getCode());
- }
-
- @Test
- public void testGetStatus() throws InterruptedException, ExecutionException {
- GetStatusInputBuilder inputBuilder = new GetStatusInputBuilder();
- inputBuilder.setNearRtRicId(nearRtRicId);
- inputBuilder.setPolicyTypeId(policyTypeId);
- inputBuilder.setPolicyInstanceId(policyTypeInstanceId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri = nearRicUrlProvider.getPolicyInstanceIdStatus(inputBuilder.build().getNearRtRicId(),
- String.valueOf(inputBuilder.build().getPolicyTypeId()), inputBuilder.getPolicyInstanceId());
- String policyInstanceStatus = "[{\"handler_id\":\"NearRTRIC-1\",\"status\":\"enforced\"}]";
- ResponseEntity<Object> getStatusResponse =
- new ResponseEntity<>(new Gson().fromJson(policyInstanceStatus, List.class), HttpStatus.OK);
- when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(getStatusResponse);
- ListenableFuture<RpcResult<GetStatusOutput>> result =
- nonrtRicApiProvider.getStatus(inputBuilder.build());
- Assert.assertEquals("enforced", result.get().getResult().getStatus());
- Assert.assertEquals(String.valueOf(HttpStatus.OK.value()), result.get().getResult().getCode());