1 package org.openecomp.sdc.be.model.jsontitan.operations;
5 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
6 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
7 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
8 import org.openecomp.sdc.be.datatypes.elements.MapArtifactDataDefinition;
9 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
10 import org.openecomp.sdc.be.model.ArtifactDefinition;
11 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
13 import java.util.Collections;
14 import java.util.HashMap;
17 import static org.junit.Assert.*;
18 import static org.mockito.Mockito.*;
20 public class ArtifactsOperationsTest {
22 private static final String SERVICE_ID = "serviceId";
23 private static final String INSTANCE_ID = "instanceId";
24 private ArtifactsOperations testInstance = mock(ArtifactsOperations.class, CALLS_REAL_METHODS);
27 public void getInstanceArtifacts_collectAllInstanceArtifacts() throws Exception {
28 Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
30 Map<String, ToscaDataDefinition> instanceDeploymentArtifacts = new HashMap<>();
31 instanceDeploymentArtifacts.put(INSTANCE_ID, getArtifactsByInstance("name2", "name3"));
32 instanceDeploymentArtifacts.put("instanceId2", getArtifactsByInstance("name4"));
34 doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
35 doReturn(Either.left(instanceDeploymentArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
36 Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
38 assertTrue(allInstArtifacts.isLeft());
39 assertEquals(allInstArtifacts.left().value().size(), 3);
40 assertTrue(allInstArtifacts.left().value().containsKey("name1"));
41 assertTrue(allInstArtifacts.left().value().containsKey("name2"));
42 assertTrue(allInstArtifacts.left().value().containsKey("name3"));
43 assertFalse(allInstArtifacts.left().value().containsKey("name4"));//this key is of different instance
47 public void getInstanceArtifacts_noArtifactsForInstance() throws Exception {
48 Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
50 doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
51 doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
52 Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, "someOtherInstance");
54 assertTrue(allInstArtifacts.isLeft());
55 assertTrue(allInstArtifacts.left().value().isEmpty());
59 public void getInstanceArtifacts_errorGettingInstanceArtifacts() throws Exception {
60 doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
61 Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
62 verify(testInstance, times(0)).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
63 assertTrue(allInstArtifacts.isRight());
67 public void getAllInstanceArtifacts_errorGettingDeploymentArtifacts() throws Exception {
68 doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
69 doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
70 Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
71 assertTrue(allInstArtifacts.isRight());
74 private ToscaDataDefinition getArtifactsByInstance(String ... artifactsNames) {
75 MapArtifactDataDefinition artifactsByInstance = new MapArtifactDataDefinition();
76 Map<String, ArtifactDataDefinition> artifactsByName = new HashMap<>();
77 for (String artifactName : artifactsNames) {
78 artifactsByName.put(artifactName, new ArtifactDataDefinition());
80 artifactsByInstance.setMapToscaDataDefinition(artifactsByName);
81 return artifactsByInstance;