+ @Test
+ public void testAssociateOrAddCalcCapReqToComponent() {
+ StorageOperationStatus result;
+ GraphVertex graphVertex = new GraphVertex();
+ Map<String, MapListRequirementDataDefinition> calcRequirements = new HashMap<>();
+ Map<String, MapListCapabilityDataDefinition> calcCapabilty = new HashMap<>();
+ Map<String, MapCapabilityProperty> calCapabilitiesProps = new HashMap<>();
+ addPolicyToToscaElementWithStatus(TitanOperationStatus.OK);
+ result = topologyTemplateOperation.associateOrAddCalcCapReqToComponent(graphVertex, calcRequirements, calcCapabilty, calCapabilitiesProps);
+ assertEquals(StorageOperationStatus.OK, result);
+ }
+
+ @Test
+ public void testUpdateDistributionStatus() {
+ Either<GraphVertex, StorageOperationStatus> result;
+ String uniqueId = "uniqueId";
+ User user = new User();
+ String userId = "userId";
+ user.setUserId(userId);
+ Iterator<Edge> edgeIterator = new Iterator<Edge>() {
+ @Override
+ public boolean hasNext() {
+ return false;
+ }
+
+ @Override
+ public Edge next() {
+ return null;
+ }
+ };
+ GraphVertex graphVertex = Mockito.mock(GraphVertex.class);
+ TitanVertex titanVertex = Mockito.mock(TitanVertex.class);
+ when(graphVertex.getVertex()).thenReturn(titanVertex);
+ when(titanVertex.edges(Direction.IN, EdgeLabelEnum.LAST_DISTRIBUTION_STATE_MODIFIER.name())).thenReturn(edgeIterator);
+ when(titanDao.getVertexByPropertyAndLabel(GraphPropertyEnum.USERID, userId, VertexTypeEnum.USER, JsonParseFlagEnum.NoParse)).thenReturn(Either.left(graphVertex));
+ when(titanDao.getVertexById(uniqueId, JsonParseFlagEnum.ParseMetadata)).thenReturn(Either.left(graphVertex));
+ when(titanDao.createEdge(graphVertex, graphVertex, EdgeLabelEnum.LAST_DISTRIBUTION_STATE_MODIFIER, null)).thenReturn(TitanOperationStatus.OK);
+ when(titanDao.updateVertex(graphVertex)).thenReturn(Either.left(graphVertex));
+ result = topologyTemplateOperation.updateDistributionStatus(uniqueId, user, DistributionStatusEnum.DISTRIBUTED);
+ assertThat(result.isLeft());
+ }
+