Removing AJSC and moving to SpringBoot

[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 d5d77ab..fe72090 100644 (file)
--- a/src/test/java/org/onap/aai/sa/rest/StubEsController.java
+++ b/src/test/java/org/onap/aai/sa/rest/StubEsController.java
@@ -82,6 +82,7 @@ public class StubEsController implements DocumentStoreInterface {
     return opResult;
   }
 
+
   @Override
   public OperationResult deleteIndex(String indexName) throws DocumentStoreOperationException {
 
@@ -153,9 +154,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;
@@ -168,8 +171,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;
@@ -206,6 +213,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 {