Improve SONAR coverage in aai/data-router

Change-Id: I568a2eda367e29dbf013881bdf03aeef3312e319
Issue-ID: AAI-214
Signed-off-by: Bansal, Nitin (nb121v) <nitin.bansal@amdocs.com>

3 files changed, 137 insertions, 0 deletions

diff --git a/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyStubbed.java b/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyStubbed.java
new file mode 100644
index 0000000..cdc0fad
--- /dev/null
+++ b/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyStubbed.java
@@ -0,0 +1,34 @@
+package org.openecomp.datarouter.policy;
+
+import java.io.FileNotFoundException;
+
+import org.openecomp.datarouter.entity.DocumentStoreDataEntity;
+
+public class EntityEventPolicyStubbed extends EntityEventPolicy {
+	
+
+	public EntityEventPolicyStubbed(EntityEventPolicyConfig config) throws FileNotFoundException {
+		super(config);
+		
+	}
+
+	protected void handleSearchServiceOperation(DocumentStoreDataEntity eventEntity, String action, String index) {
+		//Stub out the actual call to Search Data service and instead store/update documents in memory
+		try {
+			switch (action.toLowerCase()) {
+			case "create":
+				InMemorySearchDatastore.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if variable == c1
+				break;
+			case "update":
+				InMemorySearchDatastore.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if variable == c1
+				break;
+			case "delete":
+				InMemorySearchDatastore.remove(eventEntity.getId()); // they are executed if variable == c1
+				break;
+			default:
+				break;
+			}
+		} catch (Exception ex) {
+		}
+	}
+}
diff --git a/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyTest.java b/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyTest.java
new file mode 100644
index 0000000..0d97e1d
--- /dev/null
+++ b/src/test/java/org/openecomp/datarouter/policy/EntityEventPolicyTest.java
@@ -0,0 +1,79 @@
+package org.openecomp.datarouter.policy;
+
+import static org.junit.Assert.*;
+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.openecomp.datarouter.util.NodeUtils;
+import org.openecomp.datarouter.util.SearchServiceAgent;
+import org.powermock.api.mockito.PowerMockito;
+
+
+
+public class EntityEventPolicyTest {
+	EntityEventPolicy policy;
+	String eventJson;
+	
+	@SuppressWarnings("unchecked")
+    @Before
+    public void init() throws Exception {
+		EntityEventPolicyConfig config = PowerMockito.mock(EntityEventPolicyConfig.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 EntityEventPolicyStubbed(config);
+		FileInputStream event = new FileInputStream( new File("src/test/resources/aai_event.json"));
+		eventJson = IOUtils.toString(event, "UTF-8");
+
+	}
+
+	@Test
+	 public void testProcess() throws Exception {
+		policy.process(getExchangeEvent("event1","create"));
+		policy.process(getExchangeEvent("event2","create"));
+		
+		assertNotNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("event1")));
+		assertNotNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("event2")));
+		
+		policy.process(getExchangeEvent("event1","update"));
+		policy.process(getExchangeEvent("event2","update"));
+		assertNotNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("event1")));
+		assertNotNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("event2")));
+		
+		policy.process(getExchangeEvent("event2","delete"));
+		assertNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("event2")));
+	}
+	
+	private Exchange getExchangeEvent(String link,String action){
+		Object obj = eventJson.replace("$LINK",link ).replace("$ACTION",action) ;
+		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;
+		
+	}
+
+	
+
+}
diff --git a/src/test/java/org/openecomp/datarouter/policy/InMemorySearchDatastore.java b/src/test/java/org/openecomp/datarouter/policy/InMemorySearchDatastore.java
new file mode 100644
index 0000000..4b25f6e
--- /dev/null
+++ b/src/test/java/org/openecomp/datarouter/policy/InMemorySearchDatastore.java
@@ -0,0 +1,24 @@
+package org.openecomp.datarouter.policy;
+
+import java.util.concurrent.ConcurrentHashMap;
+
+public final class InMemorySearchDatastore {
+
+	private final static ConcurrentHashMap<String, String> documents = new ConcurrentHashMap<String, String>();
+
+	public static ConcurrentHashMap<String, String> getAll() {
+		return documents;
+	}
+
+	public static void put(String key, String value) {
+		documents.put(key, value);
+	}
+
+	public static String get(String key) {
+		return documents.get(key);
+	}
+
+	public static void remove(String key) {
+		documents.remove(key);
+	}
+}