catalog-model/src/test/java/org/openecomp/sdc/be/model/jsontitan/operations/ArtifactsOperationsTest.java

   1 package org.openecomp.sdc.be.model.jsontitan.operations;
   2
   3 import fj.data.Either;
   4 import org.junit.Test;
   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;
  12
  13 import java.util.Collections;
  14 import java.util.HashMap;
  15 import java.util.Map;
  16
  17 import static org.junit.Assert.*;
  18 import static org.mockito.Mockito.*;
  19
  20 public class ArtifactsOperationsTest {
  21
  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);
  25
  26     @Test
  27     public void getInstanceArtifacts_collectAllInstanceArtifacts() throws Exception {
  28         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
  29
  30         Map<String, ToscaDataDefinition> instanceDeploymentArtifacts = new HashMap<>();
  31         instanceDeploymentArtifacts.put(INSTANCE_ID, getArtifactsByInstance("name2", "name3"));
  32         instanceDeploymentArtifacts.put("instanceId2", getArtifactsByInstance("name4"));
  33
  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);
  37
  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
  44     }
  45
  46     @Test
  47     public void getInstanceArtifacts_noArtifactsForInstance() throws Exception {
  48         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
  49
  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");
  53
  54         assertTrue(allInstArtifacts.isLeft());
  55         assertTrue(allInstArtifacts.left().value().isEmpty());
  56     }
  57
  58     @Test
  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());
  64     }
  65
  66     @Test
  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());
  72     }
  73
  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());
  79         }
  80         artifactsByInstance.setMapToscaDataDefinition(artifactsByName);
  81         return artifactsByInstance;
  82     }
  83 }