Add A1 controller client in policy-agent
[nonrtric.git] / sdnc-a1-controller / northbound / nonrt-ric-api / provider / src / test / java / org / onap / sdnc / northbound / NonrtRicApiProviderTest.java
index e67304c..2646aff 100644 (file)
 
 package org.onap.sdnc.northbound;
 
-import static org.mockito.Matchers.anyObject;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.when;
-import com.google.common.base.Optional;
 import com.google.common.util.concurrent.ListenableFuture;
 import java.util.ArrayList;
 import java.util.List;
@@ -36,29 +34,25 @@ import org.mockito.Mock;
 import org.mockito.internal.util.reflection.Whitebox;
 import org.mockito.runners.MockitoJUnitRunner;
 import org.onap.sdnc.northbound.provider.NonrtRicApiProvider;
-import org.onap.sdnc.northbound.restadpter.NearRicUrlProvider;
-import org.onap.sdnc.northbound.restadpter.RestAdapter;
+import org.onap.sdnc.northbound.restadapter.NearRicUrlProvider;
+import org.onap.sdnc.northbound.restadapter.RestAdapter;
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
 import org.opendaylight.controller.md.sal.binding.test.AbstractConcurrentDataBrokerTest;
 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.CreatePolicyTypeInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.CreatePolicyTypeOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetHealthCheckInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetHealthCheckOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyInstanceInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyInstanceOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyInstancesInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyInstancesOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyTypeInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyTypeOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyTypesInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetPolicyTypesOutput;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetStatusInputBuilder;
-import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev191002.GetStatusOutput;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyIdentitiesInputBuilder;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyIdentitiesOutput;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyTypeIdentitiesInputBuilder;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyTypeIdentitiesOutput;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyTypeInputBuilder;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.GetPolicyTypeOutput;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.PutPolicyInputBuilder;
+import org.opendaylight.yang.gen.v1.org.onap.sdnc.northbound.a1.adapter.rev200122.PutPolicyOutput;
 import org.opendaylight.yangtools.yang.common.RpcResult;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
 
 /**
  * This class Tests all the methods in NonrtRicApiProvider
@@ -80,9 +74,9 @@ public class NonrtRicApiProviderTest extends AbstractConcurrentDataBrokerTest {
   @Mock
   private RestAdapter restAdapter;
   private NearRicUrlProvider nearRicUrlProvider;
-  private static String nearRtRicId = "NearRtRic1";
-  private static Long policyTypeId = 11L;
-  private static String policyTypeInstanceId = "12";
+  private static String nearRtRicUrl = "http://ric1:8085";
+  private static String policyTypeId = "STD_QoSNudging_0.1.0";
+  private static String policyId = "3d2157af-6a8f-4a7c-810f-38c2f824bf12";
 
 
   @Before
@@ -94,112 +88,65 @@ public class NonrtRicApiProviderTest extends AbstractConcurrentDataBrokerTest {
   }
 
   @Test
-  public void testCreatePolicyType() throws InterruptedException, ExecutionException {
-    CreatePolicyTypeInputBuilder inputBuilder = new CreatePolicyTypeInputBuilder();
-    inputBuilder.setNearRtRicId(nearRtRicId);
-    inputBuilder.setPolicyTypeId(policyTypeId);
+  public void testGetPolicyTypeIdentities() throws InterruptedException, ExecutionException {
+    GetPolicyTypeIdentitiesInputBuilder inputBuilder = new GetPolicyTypeIdentitiesInputBuilder();
+    inputBuilder.setNearRtRicUrl(nearRtRicUrl);
     Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
-    String uri = nearRicUrlProvider.getPolicyTypeId(inputBuilder.build().getNearRtRicId(),
-                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());
+    String uri = nearRicUrlProvider.getPolicyTypeIdentitiesUrl(inputBuilder.build().getNearRtRicUrl());
+    List<String> policyTypeIdentities = new ArrayList<>();
+    policyTypeIdentities.add(policyTypeId);
+    ResponseEntity<Object> getPolicyTypeIdentitiesResponse = new ResponseEntity<>(policyTypeIdentities, HttpStatus.OK);
+    when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(getPolicyTypeIdentitiesResponse);
+    ListenableFuture<RpcResult<GetPolicyTypeIdentitiesOutput>> result =
+        nonrtRicApiProvider.getPolicyTypeIdentities(inputBuilder.build());
+    Assert.assertEquals(policyTypeIdentities, result.get().getResult().getPolicyTypeIdList());
   }
 
   @Test
-  public void testGetPolicyType() throws InterruptedException, ExecutionException {
-    GetPolicyTypeInputBuilder inputBuilder = new GetPolicyTypeInputBuilder();
-    inputBuilder.setNearRtRicId(nearRtRicId);
-    inputBuilder.setPolicyTypeId(policyTypeId);
+  public void testGetPolicyIdentities() throws InterruptedException, ExecutionException {
+    GetPolicyIdentitiesInputBuilder inputBuilder = new GetPolicyIdentitiesInputBuilder();
+    inputBuilder.setNearRtRicUrl(nearRtRicUrl);
     Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
-    String uri = nearRicUrlProvider.getPolicyTypeId(inputBuilder.build().getNearRtRicId(),
-            String.valueOf(inputBuilder.build().getPolicyTypeId()));
-    String policyType =
-            "{\"name\":\"Policy type 1\",\"description\":\"PT 1\",\"policy_type_id\":1,\"create_schema\":{}}";
-    when(restAdapter.get(eq(uri), anyObject())).thenReturn(Optional.of(policyType));
-    ListenableFuture<RpcResult<GetPolicyTypeOutput>> result =
-        nonrtRicApiProvider.getPolicyType(inputBuilder.build());
-    Assert.assertEquals("Policy type 1", result.get().getResult().getName());
+    String uri = nearRicUrlProvider.getPolicyIdentitiesUrl(inputBuilder.build().getNearRtRicUrl());
+    List<String> policyIdentities = new ArrayList<>();
+    policyIdentities.add(policyId);
+    ResponseEntity<Object> getPolicyIdentitiesResponse = new ResponseEntity<>(policyIdentities, HttpStatus.OK);
+    when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(getPolicyIdentitiesResponse);
+    ListenableFuture<RpcResult<GetPolicyIdentitiesOutput>> result =
+        nonrtRicApiProvider.getPolicyIdentities(inputBuilder.build());
+    Assert.assertEquals(policyIdentities, result.get().getResult().getPolicyIdList());
   }
 
   @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);
-    when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(Optional.of(policyTypesInteger));
-    ListenableFuture<RpcResult<GetPolicyTypesOutput>> result =
-        nonrtRicApiProvider.getPolicyTypes(inputBuilder.build());
-    Assert.assertEquals(policyTypesLong, result.get().getResult().getPolicyTypeIdList());
-  }
-
-  @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\"}";
-    when(restAdapter.get(eq(uri), eq(String.class)))
-        .thenReturn(Optional.of(policyInstance));
-    ListenableFuture<RpcResult<GetPolicyInstanceOutput>> result =
-        nonrtRicApiProvider.getPolicyInstance(inputBuilder.build());
-    Assert.assertEquals(policyInstance, result.get().getResult().getPolicyInstance());
-  }
-
-  @Test
-  public void testGetPolicyInstances() throws InterruptedException, ExecutionException {
-    GetPolicyInstancesInputBuilder inputBuilder = new GetPolicyInstancesInputBuilder();
-    inputBuilder.setNearRtRicId(nearRtRicId);
+  public void testGetPolicyType() throws InterruptedException, ExecutionException {
+    GetPolicyTypeInputBuilder inputBuilder = new GetPolicyTypeInputBuilder();
+    inputBuilder.setNearRtRicUrl(nearRtRicUrl);
     inputBuilder.setPolicyTypeId(policyTypeId);
     Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
-    String uri = nearRicUrlProvider.getPolicyInstances(inputBuilder.build().getNearRtRicId(),
+    String uri = nearRicUrlProvider.getPolicyTypeUrl(inputBuilder.build().getNearRtRicUrl(),
             String.valueOf(inputBuilder.build().getPolicyTypeId()));
-    List<String> policyInstances = new ArrayList<>();
-    policyInstances.add("3d2157af-6a8f-4a7c-810f-38c2f824bf12");
-    when(restAdapter.get(eq(uri), eq(List.class))).thenReturn(Optional.of(policyInstances));
-    ListenableFuture<RpcResult<GetPolicyInstancesOutput>> result =
-        nonrtRicApiProvider.getPolicyInstances(inputBuilder.build());
-    Assert.assertEquals(policyInstances, result.get().getResult().getPolicyInstanceIdList());
-  }
-
-  @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 = "{\"status\":\"enforced\"}";
-    when(restAdapter.get(eq(uri), eq(String.class))).thenReturn(Optional.of(policyInstanceStatus));
-    ListenableFuture<RpcResult<GetStatusOutput>> result =
-        nonrtRicApiProvider.getStatus(inputBuilder.build());
-    Assert.assertEquals("enforced", result.get().getResult().getStatus());
+    String testPolicyType = "{}";
+    ResponseEntity<Object> getPolicyTypeResponse = new ResponseEntity<>(testPolicyType, HttpStatus.OK);
+    when(restAdapter.get(eq(uri), eq(String.class))).thenReturn(getPolicyTypeResponse);
+    ListenableFuture<RpcResult<GetPolicyTypeOutput>> result =
+        nonrtRicApiProvider.getPolicyType(inputBuilder.build());
+    Assert.assertEquals(testPolicyType, result.get().getResult().getPolicyType());
   }
 
   @Test
-  public void testHealthCheck() throws InterruptedException, ExecutionException {
-    GetHealthCheckInputBuilder inputBuilder = new GetHealthCheckInputBuilder();
-    inputBuilder.setNearRtRicId(nearRtRicId);
+  public void testPutPolicy() throws InterruptedException, ExecutionException {
+    PutPolicyInputBuilder inputBuilder = new PutPolicyInputBuilder();
+    String testPolicy = "{}";
+    inputBuilder.setNearRtRicUrl(nearRtRicUrl);
+    inputBuilder.setPolicyId(policyId);
+    inputBuilder.setPolicy(testPolicy);
     Whitebox.setInternalState(nonrtRicApiProvider, "restAdapter", restAdapter);
-    String uri = nearRicUrlProvider.getHealthCheck(inputBuilder.build().getNearRtRicId());
-    String healthCheckStatus = "";
-    when(restAdapter.get(eq(uri), eq(String.class))).thenReturn(null);
-    ListenableFuture<RpcResult<GetHealthCheckOutput>> result =
-        nonrtRicApiProvider.getHealthCheck(inputBuilder.build());
-    Assert.assertEquals(true, result.get().getResult().isHealthStatus());
+    String uri = nearRicUrlProvider.getPolicyUrl(inputBuilder.build().getNearRtRicUrl(),
+            inputBuilder.getPolicyId());
+    ResponseEntity<String> putPolicyResponse = new ResponseEntity<>(testPolicy, HttpStatus.CREATED);
+    when(restAdapter.put(eq(uri), eq(testPolicy), eq(String.class))).thenReturn(putPolicyResponse);
+    ListenableFuture<RpcResult<PutPolicyOutput>> result =
+        nonrtRicApiProvider.putPolicy(inputBuilder.build());
+    Assert.assertEquals(testPolicy, result.get().getResult().getReturnedPolicy());
   }
 }