- nonrtRicApiProvider = new NonrtRicApiProvider(dataBroker, mockNotificationPublishService,
- mockRpcProviderRegistry);
- }
-
- @Test
- public void testCreatePolicyType() throws InterruptedException, ExecutionException {
- CreatePolicyTypeInputBuilder inputBuilder = new CreatePolicyTypeInputBuilder();
- inputBuilder.setPolicyTypeId(policyTypeId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri =
- nearRicUrlProvider.getPolicyTypeId(String.valueOf(inputBuilder.build().getPolicyTypeId()));
- Optional<Object> createPolicyTyperesponse = null;
- when(restAdapter.put(eq(uri), anyObject())).thenReturn(createPolicyTyperesponse);
- ListenableFuture<RpcResult<CreatePolicyTypeOutput>> result =
- nonrtRicApiProvider.createPolicyType(inputBuilder.build());
- Assert.assertEquals("Success", result.get().getResult().getStatus());
- }
-
- @Test
- public void testGetPolicyType() throws InterruptedException, ExecutionException {
- GetPolicyTypeInputBuilder inputBuilder = new GetPolicyTypeInputBuilder();
- inputBuilder.setPolicyTypeId(policyTypeId);
- Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
- String uri =
- nearRicUrlProvider.getPolicyTypeId(String.valueOf(inputBuilder.build().getPolicyTypeId()));
- PolicyTypeSchema policyTypeSchema = new PolicyTypeSchema();
- policyTypeSchema.setName("AdmissionControlPolicy");
- policyTypeSchema.setCreateSchema("{}");
- when(restAdapter.get(eq(uri), anyObject())).thenReturn(Optional.of(policyTypeSchema));
- ListenableFuture<RpcResult<GetPolicyTypeOutput>> result =
- nonrtRicApiProvider.getPolicyType(inputBuilder.build());
- Assert.assertEquals(policyTypeSchema.getName(), result.get().getResult().getName());