Adding UI extensibility

[aai/sparky-be.git] / src / test / java / org / onap / aai / sparky / synchronizer / entity / AggregationSuggestionEntityTest.java

diff --git a/src/test/java/org/onap/aai/sparky/synchronizer/entity/AggregationSuggestionEntityTest.java b/src/test/java/org/onap/aai/sparky/synchronizer/entity/AggregationSuggestionEntityTest.java
index 5aab75d..09a5299 100644 (file)
--- a/src/test/java/org/onap/aai/sparky/synchronizer/entity/AggregationSuggestionEntityTest.java
+++ b/src/test/java/org/onap/aai/sparky/synchronizer/entity/AggregationSuggestionEntityTest.java
@@ -35,6 +35,7 @@ import java.util.List;
 import java.util.Scanner;
 
 import org.junit.Test;
+import org.onap.aai.sparky.sync.entity.AggregationSuggestionEntity;
 
 public class AggregationSuggestionEntityTest {
   public String getResourceFileContents(String filePath) {
@@ -60,10 +61,12 @@ public class AggregationSuggestionEntityTest {
   public void testGetIndexDocumentJson() {
     AggregationSuggestionEntity aggregationSuggestionEntity = new AggregationSuggestionEntity();
 
+    List<String> filterIds = new ArrayList<>(Arrays.asList("1", "2", "7", "8"));
+    aggregationSuggestionEntity.setFilterIds(filterIds);
+
     String expectedFilterListPayload = getResourceFileContents(
-        "sync/entity/AggregationSuggestionEntity_getIndexDocumentJson_expected.json");
+        "filters/AggregationSuggestionEntity_getIndexDocumentJson_expected.json");
 
-    assertTrue(aggregationSuggestionEntity.getIndexDocumentJson()
-        .contains(expectedFilterListPayload.trim()));
+    assertTrue(aggregationSuggestionEntity.getAsJson().contains(expectedFilterListPayload.trim()));
   }
 }