X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fdatarouter%2Fpolicy%2FSpikeAggregateGenericVnfProcessorTest.java;h=03970bd64565e731af8682bbf7f87c494408affa;hb=72da66902a3efaff3068ed2b3f4f7eb3bb5fe43f;hp=d1c01af44f18295f980e8b260a87bc6c48802a9f;hpb=7753d7e90f5d91b143043a642e90dd30806da2ba;p=aai%2Fdata-router.git

diff --git a/src/test/java/org/onap/aai/datarouter/policy/SpikeAggregateGenericVnfProcessorTest.java b/src/test/java/org/onap/aai/datarouter/policy/SpikeAggregateGenericVnfProcessorTest.java
index d1c01af..03970bd 100644
--- a/src/test/java/org/onap/aai/datarouter/policy/SpikeAggregateGenericVnfProcessorTest.java
+++ b/src/test/java/org/onap/aai/datarouter/policy/SpikeAggregateGenericVnfProcessorTest.java
@@ -34,68 +34,80 @@ import org.apache.commons.io.IOUtils;
 import org.junit.Before;
 import org.junit.Test;
 import org.onap.aai.datarouter.util.NodeUtils;
-import org.onap.aai.datarouter.util.SearchServiceAgent;
 import org.powermock.api.mockito.PowerMockito;
 
-
-
 public class SpikeAggregateGenericVnfProcessorTest {
-  SpikeAggregateGenericVnfProcessor policy;
-  String eventJson;
-
+  private SpikeAggregateGenericVnfProcessor policy;
+  private InMemorySearchDatastore searchDb;
 
-  @SuppressWarnings("unchecked")
   @Before
   public void init() throws Exception {
     SpikeEventPolicyConfig config = PowerMockito.mock(SpikeEventPolicyConfig.class);
     PowerMockito.when(config.getSearchKeystorePwd()).thenReturn("password");
     PowerMockito.when(config.getSourceDomain()).thenReturn("JUNIT");
 
+    searchDb = new InMemorySearchDatastore();
+    policy = new SpikeAggregateGenericVnfProcessorStubbed(config).withSearchDb(searchDb);
+  }
 
-    SearchServiceAgent searchServiceAgent = PowerMockito.mock(SearchServiceAgent.class);
-    PowerMockito.whenNew(SearchServiceAgent.class).withAnyArguments()
-        .thenReturn(searchServiceAgent);
+  @Test
+  public void testProcess_success() throws Exception {
 
+    String genericVnfEventJsonTemplate = IOUtils.toString(
+        new FileInputStream(new File("src/test/resources/generic-vnf-spike-event.json")), "UTF-8");
 
-    policy = new SpikeAggregateGenericVnfProcessorStubbed(config);
-    FileInputStream event = new FileInputStream(new File("src/test/resources/spike_event.json"));
-    eventJson = IOUtils.toString(event, "UTF-8");  
+    policy.process(
+        getExchangeEvent(genericVnfEventJsonTemplate, "update-notification", "CREATE", "gvnf123"));
 
+    assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("generic-vnf/gvnf123")));
+
+    policy.process(
+        getExchangeEvent(genericVnfEventJsonTemplate, "update-notification", "DELETE", "gvnf123"));
+
+    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("generic-vnf/gvnf123")));
+
+    
   }
+  /*
+   * Failure test cases - no searchable attribute for type
+   */
 
   @Test
-  public void testProcess_success() throws Exception {
-    policy.process(getExchangeEvent("12345", "create", "generic-vnf"));
-    policy.process(getExchangeEvent("23456", "create", "generic-vnf"));
-    
-    assertNotNull(
-        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/12345")));
-    assertNotNull(
-        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/23456")));
-
-   
-    policy.process(getExchangeEvent("23456", "delete", "generic-vnf"));
-    assertNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("23456")));
+  public void testProcess_failure_unknownOxmEntityType() throws Exception {
+
+    String pserverEventJsonTemplate = IOUtils.toString(
+        new FileInputStream(new File("src/test/resources/optical-router-spike-event.json")),
+        "UTF-8");
+
+    policy.process(
+        getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "optronic123"));
+
+    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("optical-router/optronic123")));
   }
+
   @Test
-  public void testProcess_fail() throws Exception {
-    policy.process(getExchangeEvent("666666", "create", "NotValid"));
-    assertNull(
-        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("NotValid/666666")));
-    
-    policy.process(getExchangeEvent("", "create", "generic-vnf"));
-    assertNull(
-        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/")));
+  public void testProcess_failure_missingMandatoryFieldsFromBodyObject() throws Exception {
 
+    String pserverEventJsonTemplate = IOUtils.toString(
+        new FileInputStream(
+            new File("src/test/resources/pserver-missing-mandtory-field-spike-event.json")),
+        "UTF-8");
+
+    policy.process(
+        getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "pserver123"));
+
+    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("pserver/pserver123")));
   }
 
-  
-  private Exchange getExchangeEvent(String key, String action, String type) {
-    Object obj = eventJson.replace("$KEY", key).replace("$ACTION", action).replace("$TYPE", type);
+  private Exchange getExchangeEvent(String payloadTemplate, String eventType, String operationType,
+      String entityKey) {
+    Object obj = payloadTemplate.replace("$EVENT_TYPE", eventType)
+        .replace("$OPERATION_TYPE", operationType).replace("$ENTITY_KEY", entityKey);
+
     Exchange exchange = PowerMockito.mock(Exchange.class);
     Message inMessage = PowerMockito.mock(Message.class);
     Message outMessage = PowerMockito.mock(Message.class);
-    PowerMockito.when(exchange.getIn()).thenReturn(inMessage); 
+    PowerMockito.when(exchange.getIn()).thenReturn(inMessage);
     PowerMockito.when(inMessage.getBody()).thenReturn(obj);
 
     PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
@@ -106,6 +118,4 @@ public class SpikeAggregateGenericVnfProcessorTest {
 
   }
 
-
-
 }