X-Git-Url: https://gerrit.onap.org/r/gitweb?p=sdc.git;a=blobdiff_plain;f=common-be%2Fsrc%2Fmain%2Fjava%2Forg%2Fopenecomp%2Fsdc%2Fbe%2Fdatatypes%2Felements%2FOperationDataDefinition.java;h=a4bc7f1e0fdd8ff411e9814305056393bc7a413d;hp=f93e41cf512bb5ff2b87ad1f2af395e7890bf8a7;hb=629837b3f7a282c74604987fd531ff4523f98a0c;hpb=615f2405b8d2dee1d91cf20188a180591ec84ab3

diff --git a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
index f93e41cf51..a4bc7f1e0f 100644
--- a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
+++ b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
@@ -21,7 +21,9 @@
 package org.openecomp.sdc.be.datatypes.elements;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+import java.util.Map;
 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
+import org.openecomp.sdc.be.datatypes.enums.MilestoneTypeEnum;
 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
 
 import java.io.Serializable;
@@ -46,6 +48,7 @@ public class OperationDataDefinition extends ToscaDataDefinition implements Seri
     }
 
     public OperationDataDefinition(OperationDataDefinition p) {
+        setMilestones(p.getMilestones());
         setDescription(p.getDescription());
         setImplementation(p.getImplementation());
         setInputs(p.getInputs());
@@ -59,6 +62,15 @@ public class OperationDataDefinition extends ToscaDataDefinition implements Seri
         setWorkflowVersion(p.getWorkflowVersion());
     }
 
+    public void setMilestones(Map<String, MilestoneDataDefinition> milestones) {
+        setToscaPresentationValue(JsonPresentationFields.OPERATION_MILESTONES, milestones);
+    }
+
+    public Map<String, MilestoneDataDefinition>  getMilestones() {
+        return (Map<String, MilestoneDataDefinition> ) getToscaPresentationValue(
+            JsonPresentationFields.OPERATION_MILESTONES);
+    }
+
     public String getDescription() {
         return (String) getToscaPresentationValue(JsonPresentationFields.DESCRIPTION);
     }