Organise imports to ONAP Java standards

[aai/search-data-service.git] / src / test / java / org / onap / aai / sa / rest / StubEsController.java

diff --git a/src/test/java/org/onap/aai/sa/rest/StubEsController.java b/src/test/java/org/onap/aai/sa/rest/StubEsController.java
index 326d03a..0472283 100644 (file)
--- a/src/test/java/org/onap/aai/sa/rest/StubEsController.java
+++ b/src/test/java/org/onap/aai/sa/rest/StubEsController.java
@@ -20,19 +20,20 @@
  */
 package org.onap.aai.sa.rest;
 
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
 import org.json.simple.JSONObject;
-import org.onap.aai.sa.rest.AnalysisConfiguration;
-import org.onap.aai.sa.rest.BulkRequest;
 import org.onap.aai.sa.searchdbabstraction.elasticsearch.dao.DocumentStoreDataEntity;
 import org.onap.aai.sa.searchdbabstraction.elasticsearch.dao.DocumentStoreInterface;
 import org.onap.aai.sa.searchdbabstraction.elasticsearch.exception.DocumentStoreOperationException;
-import org.onap.aai.sa.searchdbabstraction.entity.*;
-import org.onap.aai.sa.searchdbabstraction.util.DocumentSchemaUtil;
 import org.onap.aai.sa.searchdbabstraction.entity.Document;
-import org.onap.aai.sa.rest.DocumentSchema;
-
-import java.util.HashMap;
-import java.util.Map;
+import org.onap.aai.sa.searchdbabstraction.entity.DocumentOperationResult;
+import org.onap.aai.sa.searchdbabstraction.entity.OperationResult;
+import org.onap.aai.sa.searchdbabstraction.entity.SearchHit;
+import org.onap.aai.sa.searchdbabstraction.entity.SearchHits;
+import org.onap.aai.sa.searchdbabstraction.entity.SearchOperationResult;
+import org.onap.aai.sa.searchdbabstraction.util.DocumentSchemaUtil;
 
 /**
  * This class implements a stubbed version of the document store DAO so
@@ -67,12 +68,25 @@ public class StubEsController implements DocumentStoreInterface {
     OperationResult opResult = new OperationResult();
     opResult.setResultCode(200);
 
-    opResult.setResult(index + "@" + analysisConfig.getEsIndexSettings() + "@"
-        + DocumentSchemaUtil.generateDocumentMappings(documentSchema));
+    try {
+               opResult.setResult(index + "@" + analysisConfig.getEsIndexSettings() + "@"
+                   + DocumentSchemaUtil.generateDocumentMappings(documentSchema));
+       } catch (IOException e) {
+               e.printStackTrace();
+       }
 
     return opResult;
   }
 
+  @Override
+  public OperationResult createDynamicIndex(String index, String dynamicSchema) {
+    OperationResult opResult = new OperationResult();
+    opResult.setResultCode(200);
+    // Directly return the json as this flow should not edit the json in any way
+    opResult.setResult(dynamicSchema);
+    return opResult;
+  }
+
 
   @Override
   public OperationResult deleteIndex(String indexName) throws DocumentStoreOperationException {
@@ -145,9 +159,11 @@ public class StubEsController implements DocumentStoreInterface {
       if (opResult.getDocument() != null) {
         opResult.getDocument().setEtag(null);
         opResult.getDocument().setUrl(null);
+        opResult.setResultVersion("1");
       }
       opResult.setResultCode(200);
       opResult.setResult(indexName + "@" + document.getId());
+      opResult.setResultVersion("1");
     }
 
     return opResult;
@@ -160,8 +176,12 @@ public class StubEsController implements DocumentStoreInterface {
 
     if (indexName.equals(DOES_NOT_EXIST_INDEX)) {
       opResult.setResultCode(404);
+      // Adding it to make the tests pass.
+      opResult.setResultVersion ("1");
     } else {
       opResult.setResultCode(200);
+      // Adding it to make the tests pass.
+      opResult.setResultVersion ("1");
     }
 
     return opResult;
@@ -198,6 +218,21 @@ public class StubEsController implements DocumentStoreInterface {
     return opResult;
   }
 
+  @Override
+  public SearchOperationResult suggestionQueryWithPayload(String indexName, String query)
+          throws DocumentStoreOperationException {
+    SearchOperationResult opResult = new SearchOperationResult();
+
+    if (indexName.equals(DOES_NOT_EXIST_INDEX)) {
+      opResult.setResultCode(404);
+    } else {
+      opResult.setResultCode(200);
+      opResult.setResult(indexName + "@" + query);
+    }
+
+    return opResult;
+  }
+
   @Override
   public OperationResult performBulkOperations(BulkRequest[] requests) throws DocumentStoreOperationException {