catalog-model/src/test/java/org/openecomp/sdc/be/model/jsonjanusgraph/operations/GroupsOperationMockTest.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2023 Nordix Foundation. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
  22
  23 import static org.junit.jupiter.api.Assertions.assertEquals;
  24 import static org.junit.jupiter.api.Assertions.assertTrue;
  25 import static org.mockito.Mockito.CALLS_REAL_METHODS;
  26 import static org.mockito.Mockito.doReturn;
  27 import static org.mockito.Mockito.mock;
  28
  29 import com.google.gson.Gson;
  30 import fj.data.Either;
  31 import java.util.ArrayList;
  32 import java.util.HashMap;
  33 import java.util.List;
  34 import java.util.Map;
  35 import java.util.Set;
  36 import org.janusgraph.core.JanusGraphVertex;
  37 import org.junit.jupiter.api.BeforeEach;
  38 import org.junit.jupiter.api.Test;
  39 import org.junit.jupiter.api.extension.ExtendWith;
  40 import org.mockito.Mock;
  41 import org.mockito.junit.jupiter.MockitoExtension;
  42 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphDao;
  43 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  44 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  45 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  46 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  47 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  48 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathDataDefinition;
  49 import org.openecomp.sdc.be.datatypes.elements.GroupDataDefinition;
  50 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
  51 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  52 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  53 import org.openecomp.sdc.be.model.GroupDefinition;
  54 import org.openecomp.sdc.be.model.LifecycleStateEnum;
  55 import org.openecomp.sdc.be.model.Service;
  56 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
  57 import org.openecomp.sdc.be.model.jsonjanusgraph.utils.ModelConverter;
  58 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  59
  60 @ExtendWith(MockitoExtension.class)
  61 public class GroupsOperationMockTest {
  62
  63     @Mock
  64     private JanusGraphDao janusGraphDao;
  65
  66     @Mock
  67     private JanusGraphVertex vertex;
  68
  69     protected static final String CONTAINER_ID = "containerId";
  70     protected static final String CONTAINER_NAME = "containerName";
  71
  72     private GroupsOperation test = mock(GroupsOperation.class, CALLS_REAL_METHODS);
  73
  74     @BeforeEach
  75     public void setUp() throws Exception {
  76         test.setJanusGraphDao(janusGraphDao);
  77     }
  78
  79     @Test
  80     public void deleteCalculatedCapabilitiesWithProperties() {
  81         GroupDefinition g1 = createGroupDefinition("g1");
  82         GroupDefinition g2 = createGroupDefinition("g2");
  83         List<GroupDefinition> listGroup = new ArrayList<>();
  84         listGroup.add(g1);
  85         listGroup.add(g2);
  86         GraphVertex createdGraphVertex = createBasicContainerGraphVertex();
  87         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getChildVertex(createdGraphVertex, EdgeLabelEnum.CALCULATED_CAPABILITIES,
  88             JsonParseFlagEnum.ParseJson);
  89         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse);
  90         doReturn(Either.left(createdGraphVertex)).when(test).updateOrCopyOnUpdate(createdGraphVertex, createdGraphVertex,
  91             EdgeLabelEnum.CALCULATED_CAPABILITIES);
  92         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getChildVertex(createdGraphVertex, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES,
  93             JsonParseFlagEnum.ParseJson);
  94         doReturn(Either.left(createdGraphVertex)).when(test).updateOrCopyOnUpdate(createdGraphVertex, createdGraphVertex,
  95             EdgeLabelEnum.CALCULATED_CAP_PROPERTIES);
  96         final StorageOperationStatus ret = test.deleteCalculatedCapabilitiesWithProperties(CONTAINER_ID, listGroup);
  97         assertEquals(StorageOperationStatus.OK, ret);
  98     }
  99
 100     @Test
 101     public void deleteCalculatedCapabilitiesWithProperties_NotFound() {
 102         GroupDefinition g1 = createGroupDefinition("g1");
 103         GroupDefinition g2 = createGroupDefinition("g2");
 104         List<GroupDefinition> listGroup = new ArrayList<>();
 105         listGroup.add(g1);
 106         listGroup.add(g2);
 107         GraphVertex createdGraphVertex = createBasicContainerGraphVertex();
 108         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse);
 109         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getChildVertex(createdGraphVertex, EdgeLabelEnum.CALCULATED_CAPABILITIES,
 110             JsonParseFlagEnum.ParseJson);
 111         doReturn(Either.left(createdGraphVertex)).when(test).updateOrCopyOnUpdate(createdGraphVertex, createdGraphVertex,
 112             EdgeLabelEnum.CALCULATED_CAPABILITIES);
 113         doReturn(Either.left(createdGraphVertex)).when(janusGraphDao).getChildVertex(createdGraphVertex, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES,
 114             JsonParseFlagEnum.ParseJson);
 115         doReturn(Either.right(JanusGraphOperationStatus.NOT_FOUND)).when(test).updateOrCopyOnUpdate(createdGraphVertex, createdGraphVertex,
 116             EdgeLabelEnum.CALCULATED_CAP_PROPERTIES);
 117         final StorageOperationStatus ret = test.deleteCalculatedCapabilitiesWithProperties(CONTAINER_ID, listGroup);
 118         assertEquals(StorageOperationStatus.NOT_FOUND, ret);
 119     }
 120
 121     private GroupDefinition createGroupDefinition(String id) {
 122         GroupDefinition groupDefinition = new GroupDefinition();
 123         groupDefinition.setUniqueId(id);
 124         groupDefinition.setInvariantName("name" + id);
 125         return groupDefinition;
 126     }
 127
 128     private GraphVertex createBasicContainerGraphVertex() {
 129         GraphVertex resource = new GraphVertex(VertexTypeEnum.TOPOLOGY_TEMPLATE);
 130         resource.addMetadataProperty(GraphPropertyEnum.UNIQUE_ID, CONTAINER_ID);
 131         resource.addMetadataProperty(GraphPropertyEnum.NAME, CONTAINER_NAME);
 132         resource.setJsonMetadataField(JsonPresentationFields.NAME, CONTAINER_NAME);
 133         resource.setJsonMetadataField(JsonPresentationFields.COMPONENT_TYPE, ComponentTypeEnum.RESOURCE.name());
 134         Map<String, GroupDataDefinition> groups = new HashMap<>();
 135         GroupDefinition g1 = createGroupDefinition("g1");
 136         GroupDefinition g2 = createGroupDefinition("g2");
 137         groups.put(g1.getInvariantName(), new GroupDataDefinition(g1));
 138         groups.put(g2.getInvariantName(), new GroupDataDefinition(g2));
 139         resource.setJson(groups);
 140         return resource;
 141     }
 142 }