1 files changed, 117 insertions, 0 deletions
diff --git a/src/test/java/org/onap/aai/datarouter/policy/SpikeEntitySearchProcessorTest.java b/src/test/java/org/onap/aai/datarouter/policy/SpikeEntitySearchProcessorTest.java
new file mode 100644
index 0000000..08518b1
--- /dev/null
+++ b/src/test/java/org/onap/aai/datarouter/policy/SpikeEntitySearchProcessorTest.java
@@ -0,0 +1,117 @@
+package org.onap.aai.datarouter.policy;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.mockito.Matchers.anyObject;
+import static org.mockito.Matchers.anyString;
+
+import java.io.File;
+import java.io.FileInputStream;
+
+import org.apache.camel.Exchange;
+import org.apache.camel.Message;
+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 SpikeEntitySearchProcessorTest {
+  SpikeEntitySearchProcessor policy;
+  String eventJson;
+
+  @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");
+
+
+    SearchServiceAgent searchServiceAgent = PowerMockito.mock(SearchServiceAgent.class);
+    PowerMockito.whenNew(SearchServiceAgent.class).withAnyArguments()
+        .thenReturn(searchServiceAgent);
+
+
+    policy = new SpikeEntitySearchProcessorStubbed(config);
+    FileInputStream event = new FileInputStream(new File("src/test/resources/spike_event.json"));
+    eventJson = IOUtils.toString(event, "UTF-8");  
+
+  }
+
+  @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("generic-vnf/23456")));
+    
+    policy.process(getExchangeEvent("333333", "", "generic-vnf"));
+    assertNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/333333")));
+  }
+  @Test
+  public void testProcess_fail() throws Exception {
+    policy.process(getExchangeEvent("xxxxx", "create", "NotValid"));
+    assertNull(
+        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("NotValid/xxxxx")));
+    
+    policy.process(getExchangeEvent("", "create", "generic-vnf"));
+    assertNull(
+        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/")));
+   
+    policy.process(getExchangeEvent("yyyy", "create", ""));
+    assertNull(
+        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("/yyyy")));
+    policy.process(getExchangeEvent("", "create", ""));
+    assertNull(
+        InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("/")));
+  }
+  @Test
+  public void testProcess_null() throws Exception {
+    policy.process(getExchangeEvent());
+  }
+  
+  private Exchange getExchangeEvent(String key, String action, String type) {
+    Object obj = eventJson.replace("$KEY", key).replace("$ACTION", action).replace("$TYPE", type);
+    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(inMessage.getBody()).thenReturn(obj);
+
+    PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
+    PowerMockito.doNothing().when(outMessage).setBody(anyObject());
+    PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());
+
+    return exchange;
+
+  }
+  
+  private Exchange getExchangeEvent() {
+    Object obj = "";
+    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(inMessage.getBody()).thenReturn(obj);
+
+    PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
+    PowerMockito.doNothing().when(outMessage).setBody(anyObject());
+    PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());
+    
+    return exchange;
+
+  }
+
+
+}