1 package org.openecomp.sdc.be.datamodel.utils;
5 import org.openecomp.sdc.be.info.ArtifactTemplateInfo;
6 import org.openecomp.sdc.be.model.ArtifactDefinition;
7 import org.openecomp.sdc.be.model.HeatParameterDefinition;
8 import org.openecomp.sdc.be.model.operations.impl.ArtifactOperation;
9 import org.openecomp.sdc.common.api.ArtifactGroupTypeEnum;
11 import java.util.HashMap;
12 import java.util.LinkedList;
13 import java.util.List;
15 import java.util.ArrayList;
17 import static org.junit.Assert.assertTrue;
19 public class ArtifactUtilsTest {
21 private static final String ARTIFACT_NAME = "myHeatArtifact";
24 @javax.annotation.Resource
25 private ArtifactOperation artifactOperation;
28 private static String RESOURCE_ID = "resourceId";
30 private static String USER_ID = "muUserId";
31 private static String CATEGORY_NAME = "category/mycategory";
34 private ArtifactUtils createTestSubject() {
35 return new ArtifactUtils();
39 public void testFindMasterArtifact() throws Exception {
40 Map<String, ArtifactDefinition> deploymentArtifact = new HashMap<>();
41 List<ArtifactDefinition> artifacts = new LinkedList<>();
42 List<String> artifactsList = new LinkedList<>();
43 ArtifactDefinition result;
46 result = ArtifactUtils.findMasterArtifact(deploymentArtifact, artifacts, artifactsList);
50 public void testFindMasterArtifactWithArtifactDef() throws Exception {
51 ArtifactDefinition artifactWithHeat = createResourceWithHeat();
52 Map<String, ArtifactDefinition> deploymentArtifact = new HashMap<>();
53 deploymentArtifact.put("artifactId",artifactWithHeat);
54 List<ArtifactDefinition> artifacts = new LinkedList<>();
55 artifacts.add(artifactWithHeat);
56 List<String> artifactsList = new LinkedList<>();
57 artifactsList.add("artifactId");
58 ArtifactDefinition result;
61 result = ArtifactUtils.findMasterArtifact(deploymentArtifact, artifacts, artifactsList);
65 public void testBuildJsonForUpdateArtifact() throws Exception {
66 String artifactId = "";
67 String artifactName = "";
68 String artifactType = "";
69 ArtifactGroupTypeEnum artifactGroupType = ArtifactGroupTypeEnum.DEPLOYMENT;
71 String displayName = "";
72 String description = "";
73 byte[] artifactContentent = new byte[] { ' ' };
74 List<ArtifactTemplateInfo> updatedRequiredArtifacts = null;
75 boolean isFromCsar = false;
76 Map<String, Object> result;
80 result = ArtifactUtils.buildJsonForUpdateArtifact(artifactId, artifactName, artifactType, artifactGroupType,
81 label, displayName, description, artifactContentent, updatedRequiredArtifacts, isFromCsar);
85 public void testBuildJsonForArtifact() throws Exception {
86 ArtifactTemplateInfo artifactTemplateInfo = new ArtifactTemplateInfo();
87 artifactTemplateInfo.setFileName("mock.mock.heat");
88 byte[] artifactContentent = new byte[] { ' ' };
89 int atrifactLabelCounter = 0;
90 Map<String, Object> result;
93 result = ArtifactUtils.buildJsonForArtifact(artifactTemplateInfo, artifactContentent, atrifactLabelCounter,false);
97 public void testFindArtifactInList() throws Exception {
98 List<ArtifactDefinition> createdArtifacts = new LinkedList<>();
99 String artifactId = "mock";
100 ArtifactDefinition result;
103 result = ArtifactUtils.findArtifactInList(createdArtifacts, artifactId);
107 public void testFindArtifactInListwithArtifactList() throws Exception {
108 ArtifactDefinition artifactWithHeat = createResourceWithHeat();
109 List<ArtifactDefinition> createdArtifacts = new LinkedList<>();
110 createdArtifacts.add(artifactWithHeat);
111 String artifactId = "artifactId";
112 ArtifactDefinition result;
115 result = ArtifactUtils.findArtifactInList(createdArtifacts, artifactId);
118 public ArtifactDefinition createResourceWithHeat() {
119 ArtifactDefinition artifactDefinition = createArtifactDefinition(USER_ID, RESOURCE_ID, ARTIFACT_NAME);
120 artifactDefinition.setArtifactType("HEAT");
121 artifactDefinition.setArtifactGroupType(ArtifactGroupTypeEnum.DEPLOYMENT);
123 List<HeatParameterDefinition> heatParams = new ArrayList<>();
124 HeatParameterDefinition heatParam = new HeatParameterDefinition();
125 heatParam.setCurrentValue("11");
126 heatParam.setDefaultValue("22");
127 heatParam.setDescription("desc");
128 heatParam.setName("myParam");
129 heatParam.setType("number");
130 heatParams.add(heatParam);
131 artifactDefinition.setListHeatParameters(heatParams);
132 return artifactDefinition;
137 private ArtifactDefinition createArtifactDefinition(String userId, String serviceId, String artifactName) {
138 ArtifactDefinition artifactInfo = new ArtifactDefinition();
140 artifactInfo.setArtifactName(artifactName + ".sh");
141 artifactInfo.setArtifactType("SHELL");
142 artifactInfo.setDescription("hdkfhskdfgh");
143 artifactInfo.setArtifactChecksum("UEsDBAoAAAAIAAeLb0bDQz");
145 artifactInfo.setUserIdCreator(userId);
146 String fullName = "Jim H";
147 artifactInfo.setUpdaterFullName(fullName);
148 long time = System.currentTimeMillis();
149 artifactInfo.setCreatorFullName(fullName);
150 artifactInfo.setCreationDate(time);
151 artifactInfo.setLastUpdateDate(time);
152 artifactInfo.setUserIdLastUpdater(userId);
153 artifactInfo.setArtifactLabel(artifactName);
154 artifactInfo.setUniqueId("artifactId");