Backend support for operation milestones with activities

[sdc.git] / catalog-model / src / main / java / org / openecomp / sdc / be / model / Operation.java

diff --git a/catalog-model/src/main/java/org/openecomp/sdc/be/model/Operation.java b/catalog-model/src/main/java/org/openecomp/sdc/be/model/Operation.java
index 92c7c59..a25fd01 100644 (file)
--- a/catalog-model/src/main/java/org/openecomp/sdc/be/model/Operation.java
+++ b/catalog-model/src/main/java/org/openecomp/sdc/be/model/Operation.java
@@ -19,11 +19,14 @@
  */
 package org.openecomp.sdc.be.model;
 
+import java.util.Map;
 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
+import org.openecomp.sdc.be.datatypes.elements.MilestoneDataDefinition;
 import org.openecomp.sdc.be.datatypes.elements.OperationDataDefinition;
 import org.openecomp.sdc.be.datatypes.elements.OperationInputDefinition;
 import org.openecomp.sdc.be.datatypes.elements.OperationOutputDefinition;
+import org.openecomp.sdc.be.datatypes.enums.MilestoneTypeEnum;
 
 /**
  * Defines an operation available to manage particular aspects of the Node Type.
@@ -48,11 +51,12 @@ public class Operation extends OperationDataDefinition implements IOperationPara
     }
 
     public Operation(ArtifactDataDefinition implementation, String description, ListDataDefinition<OperationInputDefinition> inputs,
-                     ListDataDefinition<OperationOutputDefinition> outputs) {
+                     ListDataDefinition<OperationOutputDefinition> outputs, Map<String, MilestoneDataDefinition> milestones) {
         super(description);
         setImplementation(implementation);
         setInputs(inputs);
         setOutputs(outputs);
+        setMilestones(milestones);
     }
 
     @Override