X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=artifactbroker%2Fmodel%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Fpolicy%2Fdistribution%2Fmodel%2FCloudArtifact.java;h=8a7943a8b8f8525c3d9717773b4dcf66a8c07db7;hb=refs%2Fchanges%2F15%2F84215%2F1;hp=2447c68d8c5b6b5915a209cf713fe81bdf7b2b50;hpb=2cf6469e7af254a65f4fde20b2ef61efad32cd57;p=multicloud%2Fframework.git

diff --git a/artifactbroker/model/src/main/java/org/onap/policy/distribution/model/CloudArtifact.java b/artifactbroker/model/src/main/java/org/onap/policy/distribution/model/CloudArtifact.java
index 2447c68..8a7943a 100644
--- a/artifactbroker/model/src/main/java/org/onap/policy/distribution/model/CloudArtifact.java
+++ b/artifactbroker/model/src/main/java/org/onap/policy/distribution/model/CloudArtifact.java
@@ -20,20 +20,22 @@
 
 package org.onap.policy.distribution.model;
 
-import java.util.List;
-import java.util.Map;
+import java.util.ArrayList;
+import java.util.HashMap;
+
+import org.onap.sdc.api.notification.IArtifactInfo;
 
 /**
  * Represents a CloudArtifact that a {@link Policy} can be decoded from.
  */
 public class CloudArtifact implements PolicyInput {
 
-    List<VfModuelModel> vfModulePayload;
-    Map<String, String> artifactTypeMap;
+    ArrayList<VfModuleModel> vfModulePayload;
+    HashMap<String, IArtifactInfo> artifactMap;
 
-    public CloudArtifact(List<VfModuelModel> vfModulePayload, Map<String, String> artifactTypeMap) {
+    public CloudArtifact(ArrayList<VfModuleModel> vfModulePayload, HashMap<String, IArtifactInfo> artifactMap) {
         this.vfModulePayload = vfModulePayload;
-        this.artifactTypeMap = artifactTypeMap;
+        this.artifactMap = artifactMap;
     }
 
     /**
@@ -41,7 +43,7 @@ public class CloudArtifact implements PolicyInput {
      *
      * @return the path of the TOSCA file
      */
-    List<VfModuelModel> getVfModulePayload() {
+    public ArrayList<VfModuleModel> getVfModulePayload() {
         return vfModulePayload;
     }
 
@@ -50,8 +52,8 @@ public class CloudArtifact implements PolicyInput {
      *
      * @return the path of the TOSCA file
      */
-    Map<String, String> getArtifactTypeMap() {
-        return artifactTypeMap;
+    public HashMap<String, IArtifactInfo> getArtifactTypeMap() {
+        return artifactMap;
     }
 
 }