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.assertEquals;
  18 import static org.junit.Assert.assertFalse;
  19 import static org.junit.Assert.assertTrue;
  20 import static org.mockito.Mockito.*;
  21
  22 public class ArtifactsOperationsTest {
  23
  24     private static final String SERVICE_ID = "serviceId";
  25     private static final String INSTANCE_ID = "instanceId";
  26     private ArtifactsOperations testInstance = mock(ArtifactsOperations.class, CALLS_REAL_METHODS);
  27
  28     @Test
  29     public void getInstanceArtifacts_collectAllInstanceArtifacts() throws Exception {
  30         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
  31
  32         Map<String, ToscaDataDefinition> instanceDeploymentArtifacts = new HashMap<>();
  33         instanceDeploymentArtifacts.put(INSTANCE_ID, getArtifactsByInstance("name2", "name3"));
  34         instanceDeploymentArtifacts.put("instanceId2", getArtifactsByInstance("name4"));
  35
  36         doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
  37         doReturn(Either.left(instanceDeploymentArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
  38         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
  39
  40         assertTrue(allInstArtifacts.isLeft());
  41         assertEquals(allInstArtifacts.left().value().size(), 3);
  42         assertTrue(allInstArtifacts.left().value().containsKey("name1"));
  43         assertTrue(allInstArtifacts.left().value().containsKey("name2"));
  44         assertTrue(allInstArtifacts.left().value().containsKey("name3"));
  45         assertFalse(allInstArtifacts.left().value().containsKey("name4"));//this key is of different instance
  46     }
  47
  48     @Test
  49     public void getInstanceArtifacts_noArtifactsForInstance() throws Exception {
  50         Map<String, ToscaDataDefinition> instanceArtifacts = Collections.singletonMap(INSTANCE_ID, getArtifactsByInstance("name1"));
  51
  52         doReturn(Either.left(instanceArtifacts)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
  53         doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
  54         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, "someOtherInstance");
  55
  56         assertTrue(allInstArtifacts.isLeft());
  57         assertTrue(allInstArtifacts.left().value().isEmpty());
  58     }
  59
  60     @Test
  61     public void getInstanceArtifacts_errorGettingInstanceArtifacts() throws Exception {
  62         doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
  63         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
  64         verify(testInstance, times(0)).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
  65         assertTrue(allInstArtifacts.isRight());
  66     }
  67
  68     @Test
  69     public void getAllInstanceArtifacts_errorGettingDeploymentArtifacts() throws Exception {
  70         doReturn(Either.left(new HashMap<>())).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INSTANCE_ARTIFACTS);
  71         doReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)).when(testInstance).getDataFromGraph(SERVICE_ID, EdgeLabelEnum.INST_DEPLOYMENT_ARTIFACTS);
  72         Either<Map<String, ArtifactDefinition>, StorageOperationStatus> allInstArtifacts = testInstance.getAllInstanceArtifacts(SERVICE_ID, INSTANCE_ID);
  73         assertTrue(allInstArtifacts.isRight());
  74     }
  75
  76     private ToscaDataDefinition getArtifactsByInstance(String ... artifactsNames) {
  77         MapArtifactDataDefinition artifactsByInstance = new MapArtifactDataDefinition();
  78         Map<String, ArtifactDataDefinition> artifactsByName = new HashMap<>();
  79         for (String artifactName : artifactsNames) {
  80             artifactsByName.put(artifactName, new ArtifactDataDefinition());
  81         }
  82         artifactsByInstance.setMapToscaDataDefinition(artifactsByName);
  83         return artifactsByInstance;
  84     }
  85 }