a8fb60999c07eb795b4f05a7ba568c23d24a16f1
[sdc.git] /
1 package org.openecomp.sdc.be.model.jsontitan.operations;
2
3 import static org.junit.Assert.assertEquals;
4 import static org.junit.Assert.assertFalse;
5 import static org.junit.Assert.assertTrue;
6 import static org.mockito.Mockito.CALLS_REAL_METHODS;
7 import static org.mockito.Mockito.doReturn;
8 import static org.mockito.Mockito.mock;
9 import static org.mockito.Mockito.times;
10 import static org.mockito.Mockito.verify;
11
12 import java.util.Collections;
13 import java.util.HashMap;
14 import java.util.Map;
15
16 import org.junit.Test;
17 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
18 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
19 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
20 import org.openecomp.sdc.be.datatypes.elements.MapArtifactDataDefinition;
21 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
22 import org.openecomp.sdc.be.model.ArtifactDefinition;
23 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
24
25 import fj.data.Either;
26
27 public class ArtifactsOperationsTest {
28
29     private static final String SERVICE_ID = "serviceId";
30     private static final String INSTANCE_ID = "instanceId";
31     private ArtifactsOperations testInstance = mock(ArtifactsOperations.class, CALLS_REAL_METHODS);
32
33     @Test
34     public void getInstanceArtifacts_collectAllInstanceArtifacts() throws Exception {
35         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
36
37         Map<String, ToscaDataDefinition> instanceDeploymentArtifacts = new HashMap<>();
38         instanceDeploymentArtifacts.put(INSTANCE_ID, getArtifactsByInstance("name2", "name3"));
39         instanceDeploymentArtifacts.put("instanceId2", getArtifactsByInstance("name4"));
40
41         doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
42         doReturn(Either.left(instanceDeploymentArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
43         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
44
45         assertTrue(allInstArtifacts.isLeft());
46         assertEquals(allInstArtifacts.left().value().size(), 3);
47         assertTrue(allInstArtifacts.left().value().containsKey("name1"));
48         assertTrue(allInstArtifacts.left().value().containsKey("name2"));
49         assertTrue(allInstArtifacts.left().value().containsKey("name3"));
50         assertFalse(allInstArtifacts.left().value().containsKey("name4"));//this key is of different instance
51     }
52
53     @Test
54     public void getInstanceArtifacts_noArtifactsForInstance() throws Exception {
55         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
56
57         doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
58         doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
59         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, "someOtherInstance");
60
61         assertTrue(allInstArtifacts.isLeft());
62         assertTrue(allInstArtifacts.left().value().isEmpty());
63     }
64
65     @Test
66     public void getInstanceArtifacts_errorGettingInstanceArtifacts() throws Exception {
67         doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
68         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
69         verify(testInstance, times(0)).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
70         assertTrue(allInstArtifacts.isRight());
71     }
72
73     @Test
74     public void getAllInstanceArtifacts_errorGettingDeploymentArtifacts() throws Exception {
75         doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
76         doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
77         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
78         assertTrue(allInstArtifacts.isRight());
79     }
80
81     private ToscaDataDefinition getArtifactsByInstance(String ... artifactsNames) {
82         MapArtifactDataDefinition artifactsByInstance = new MapArtifactDataDefinition();
83         Map<String, ArtifactDataDefinition> artifactsByName = new HashMap<>();
84         for (String artifactName : artifactsNames) {
85             artifactsByName.put(artifactName, new ArtifactDataDefinition());
86         }
87         artifactsByInstance.setMapToscaDataDefinition(artifactsByName);
88         return artifactsByInstance;
89     }
90 }