1 files changed, 122 insertions, 0 deletions
diff --git a/sparkybe-onap-service/src/test/java/org/onap/aai/sparky/sync/ElasticSearchIndexCleanerTest.java b/sparkybe-onap-service/src/test/java/org/onap/aai/sparky/sync/ElasticSearchIndexCleanerTest.java
new file mode 100644
index 0000000..30394de
--- /dev/null
+++ b/sparkybe-onap-service/src/test/java/org/onap/aai/sparky/sync/ElasticSearchIndexCleanerTest.java
@@ -0,0 +1,122 @@
+package org.onap.aai.sparky.sync;
+
+import static org.junit.Assert.*;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.onap.aai.restclient.client.OperationResult;
+import org.onap.aai.sparky.dal.ElasticSearchAdapter;
+import org.onap.aai.sparky.sync.config.ElasticSearchEndpointConfig;
+import org.onap.aai.sparky.sync.config.ElasticSearchSchemaConfig;
+import org.onap.aai.sparky.sync.enumeration.OperationState;
+import org.onap.aai.sparky.util.TestResourceLoader;
+
+public class ElasticSearchIndexCleanerTest {
+
+	private ElasticSearchIndexCleaner esIndexCleaner;
+
+	private ElasticSearchAdapter esAdapter;
+	private ElasticSearchEndpointConfig esRestEndpointConfig;
+	private ElasticSearchSchemaConfig esSchemaConfig;
+
+	@Before
+	public void init() throws Exception {
+
+		esAdapter = Mockito.mock( ElasticSearchAdapter.class);
+		esRestEndpointConfig = new ElasticSearchEndpointConfig();
+		esSchemaConfig = new ElasticSearchSchemaConfig();
+		
+		esRestEndpointConfig.setScrollContextBatchRequestSize(5000);
+		esRestEndpointConfig.setEsIpAddress("127.0.0.1");
+		esRestEndpointConfig.setEsServerPort("9200");
+		esRestEndpointConfig.setScrollContextTimeToLiveInMinutes(5);
+		
+	}
+
+	@Test
+	public void validateBasicConstruction() throws Exception {
+		esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
+	}
+	
+	@Test
+	public void validatePreOperationCollection() throws Exception {
+
+		String beforeSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
+				"/sync/ElasticSearch/BeforeSync_ElasticSearch_ScrollApi_Successful.json");
+
+		OperationResult scrollApiOpResult = new OperationResult();
+		scrollApiOpResult.setResultCode(200);
+		scrollApiOpResult.setResult(beforeSyncScrollApiResponse);
+
+		Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
+				.thenReturn(scrollApiOpResult);
+
+		esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
+		OperationState opState = esIndexCleaner.populatePreOperationCollection();
+
+		assertEquals(OperationState.OK, opState);
+
+	}
+	
+	@Test
+	public void validatePostOperationCollection() throws Exception {
+
+		String afterSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
+				"/sync/ElasticSearch/AfterSync_ElasticSearch_ScrollApi_Successful.json");
+
+		OperationResult scrollApiOpResult = new OperationResult();
+		scrollApiOpResult.setResultCode(200);
+		scrollApiOpResult.setResult(afterSyncScrollApiResponse);
+
+		Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
+				.thenReturn(scrollApiOpResult);
+
+		esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
+		OperationState opState = esIndexCleaner.populatePostOperationCollection();
+
+		assertEquals(OperationState.OK, opState);
+
+	}
+	
+
+	@Test
+	public void validatePerformCleanup() throws Exception {
+
+		String beforeSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
+				"/sync/ElasticSearch/BeforeSync_ElasticSearch_ScrollApi_Successful.json");
+
+		OperationResult beforeScrollApiOpResult = new OperationResult();
+		beforeScrollApiOpResult.setResultCode(200);
+		beforeScrollApiOpResult.setResult(beforeSyncScrollApiResponse);
+
+		String afterSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
+				"/sync/ElasticSearch/AfterSync_ElasticSearch_ScrollApi_Successful.json");
+		
+		OperationResult afterScrollApiOpResult = new OperationResult();
+		afterScrollApiOpResult.setResultCode(200);
+		afterScrollApiOpResult.setResult(afterSyncScrollApiResponse);
+
+		Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
+				.thenReturn(beforeScrollApiOpResult,afterScrollApiOpResult);
+		
+		esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
+		
+		OperationState beforeOpState = esIndexCleaner.populatePreOperationCollection();
+		OperationState afterOpState = esIndexCleaner.populatePostOperationCollection();
+
+		assertEquals(OperationState.OK, beforeOpState);
+		assertEquals(OperationState.OK, afterOpState);
+		
+		/*
+		 * Now we can start the test work
+		 */
+		
+		OperationState cleanupState = esIndexCleaner.performCleanup();
+		assertEquals(OperationState.OK, cleanupState);
+		
+
+	}
+
+
+}