O1 PM Bulk - update Traffic Model with all UE

Issue-ID: INT-1945

Signed-off-by: Stanislav Marszalek <s.marszalek2@partner.samsung.com>
Change-Id: I02d3684c98d563d7f386de2fdf032e930ac82b3f

1 files changed, 2 insertions, 4 deletions

diff --git a/src/main/java/org/onap/a1pesimulator/util/RanVesUtils.java b/src/main/java/org/onap/a1pesimulator/util/RanVesUtils.java
index a0f4991..5ecbd6d 100644
--- a/src/main/java/org/onap/a1pesimulator/util/RanVesUtils.java
+++ b/src/main/java/org/onap/a1pesimulator/util/RanVesUtils.java
@@ -54,14 +54,12 @@ public class RanVesUtils {
         commonEventHeader.setStartEpochMicrosec(getStartEpochMicroseconds());
     }
 
-    public static AdditionalMeasurement buildTrafficModelMeasurement(AdditionalMeasurement identityMeasurement,
-            RanUeHolder ranUeHolder, String valuePattern) {
-        String cellId = identityMeasurement.getHashMap().get(Constants.MEASUREMENT_FIELD_VALUE);
+    public static AdditionalMeasurement buildTrafficModelMeasurement(RanUeHolder ranUeHolder, String valuePattern) {
         AdditionalMeasurement trafficModel = new AdditionalMeasurement();
         Map<String, String> hashMap = new HashMap<>();
         trafficModel.setName(UE_PARAM_TRAFFIC_MODEL);
         trafficModel.setHashMap(hashMap);
-        Collection<UserEquipment> cellUes = ranUeHolder.getUserEquipmentsConnectedToCell(cellId);
+        Collection<UserEquipment> cellUes = ranUeHolder.getUserEquipments();
         cellUes.stream().map(UserEquipment::getId).forEach(ueId -> hashMap.put(ueId, valuePattern));
 
         return trafficModel;