1 files changed, 16 insertions, 5 deletions
diff --git a/main/src/main/java/org/onap/policy/pap/main/comm/PdpMessageGenerator.java b/main/src/main/java/org/onap/policy/pap/main/comm/PdpMessageGenerator.java
index fb47e929..99408eb2 100644
--- a/main/src/main/java/org/onap/policy/pap/main/comm/PdpMessageGenerator.java
+++ b/main/src/main/java/org/onap/policy/pap/main/comm/PdpMessageGenerator.java
@@ -72,7 +72,6 @@ public class PdpMessageGenerator {
      */
     private final Long heartBeatMs;
 
-
     /**
      * Constructs the object.
      *
@@ -94,21 +93,33 @@ public class PdpMessageGenerator {
     }
 
     protected PdpUpdate createPdpUpdateMessage(final String pdpGroupName, final PdpSubGroup subGroup,
-        final String pdpInstanceId, final PolicyModelsProvider databaseProvider)
-        throws PfModelException {
+                    final String pdpInstanceId, final PolicyModelsProvider databaseProvider,
+                    final List<ToscaPolicy> policies, final List<ToscaPolicy> policiesToBeDeployed,
+                    final List<ToscaConceptIdentifier> policiesToBeUndeployed) throws PfModelException {
 
         final PdpUpdate update = new PdpUpdate();
+
         update.setName(pdpInstanceId);
         update.setPdpGroup(pdpGroupName);
         update.setPdpSubgroup(subGroup.getPdpType());
-        update.setPolicies(getToscaPolicies(subGroup, databaseProvider));
+        update.setPolicies(policies);
+        update.setPoliciesToBeDeployed(policiesToBeDeployed);
+        update.setPoliciesToBeUndeployed(policiesToBeUndeployed);
         update.setPdpHeartbeatIntervalMs(heartBeatMs);
 
         LOGGER.debug("Created PdpUpdate message - {}", update);
         return update;
     }
 
-    private List<ToscaPolicy> getToscaPolicies(final PdpSubGroup subGroup, final PolicyModelsProvider databaseProvider)
+    /**
+     * Method to return a list of policies.
+     *
+     * @param subGroup PdpSubGroup to retrieve policies from
+     * @param databaseProvider PolicyModelsProvider used to retrieve list of policies
+     * @returns a list of ToscaPolicy
+     **/
+    public List<ToscaPolicy> getToscaPolicies(final PdpSubGroup subGroup,
+            final PolicyModelsProvider databaseProvider)
                     throws PfModelException {
 
         final List<ToscaPolicy> policies = new LinkedList<>();