1 files changed, 16 insertions, 0 deletions
diff --git a/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetVfModulesByVnf.java b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetVfModulesByVnf.java
index 798a02642..7a35abc12 100644
--- a/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetVfModulesByVnf.java
+++ b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetVfModulesByVnf.java
@@ -5,9 +5,16 @@ import org.springframework.http.HttpMethod;
 
 public class PresetAAIGetVfModulesByVnf extends BaseAAIPreset {
     private final String vnfInstanceId;
+    private final Placement vfModule1Placement;
 
     public PresetAAIGetVfModulesByVnf(String vnfInstanceId) {
         this.vnfInstanceId = vnfInstanceId;
+        vfModule1Placement = null;
+    }
+
+    public PresetAAIGetVfModulesByVnf(String vnfInstanceId, Placement vfModule2Placement) {
+        this.vnfInstanceId = vnfInstanceId;
+        this.vfModule1Placement = vfModule2Placement;
     }
 
     @Override
@@ -20,6 +27,14 @@ public class PresetAAIGetVfModulesByVnf extends BaseAAIPreset {
         return getRootPath() + "/network/generic-vnfs/generic-vnf/" + this.vnfInstanceId + "/vf-modules";
     }
 
+    private String placementRelationship(Placement placement) {
+        if (placement != null) {
+            return "," + Placement.Util.placementRelationship("vserver", placement);
+        } else {
+            return "";
+        }
+    }
+
     @Override
     public Object getResponseBody() {
         return "" +
@@ -72,6 +87,7 @@ public class PresetAAIGetVfModulesByVnf extends BaseAAIPreset {
                 "              }" +
                 "            ]" +
                 "          }" +
+            placementRelationship(vfModule1Placement) +
                 "        ]" +
                 "      }" +
                 "    }" +