2 * Copyright © 2016-2018 European Support Limited
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
19 import fj.data.Either;
20 import org.junit.Assert;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.mockito.InjectMocks;
24 import org.mockito.Mock;
25 import org.mockito.MockitoAnnotations;
26 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
27 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphDao;
28 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
29 import org.openecomp.sdc.be.datatypes.elements.MapPropertiesDataDefinition;
30 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
31 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
32 import org.openecomp.sdc.be.model.jsonjanusgraph.utils.CapabilityTestUtils;
33 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
35 import java.util.HashMap;
38 import static org.mockito.ArgumentMatchers.any;
39 import static org.mockito.ArgumentMatchers.anyMap;
40 import static org.mockito.ArgumentMatchers.anyString;
41 import static org.mockito.Mockito.when;
43 public class CapabilitiesOperationTest {
46 CapabilitiesOperation operation = new CapabilitiesOperation();
48 private JanusGraphDao mockJanusGraphDao;
50 private TopologyTemplateOperation topologyTemplateOperation;
54 MockitoAnnotations.initMocks(this);
56 when(mockJanusGraphDao.commit()).thenReturn(JanusGraphOperationStatus.OK);
57 when(mockJanusGraphDao.getVertexById(anyString(), any())).thenReturn(Either.left(new GraphVertex()));
59 when(topologyTemplateOperation.updateFullToscaData(any(), any(), any(), anyMap())).thenReturn(StorageOperationStatus.OK);
60 TopologyTemplate topologyTemplate = new TopologyTemplate();
62 Map<String, MapPropertiesDataDefinition> capPropsForTopologyTemplate = CapabilityTestUtils
63 .createCapPropsForTopologyTemplate(topologyTemplate);
64 topologyTemplate.setCapabilitiesProperties(capPropsForTopologyTemplate);
66 when(topologyTemplateOperation.getToscaElement(anyString(), any())).thenReturn(Either.left(topologyTemplate));
70 public void testCreateOrUpdateCapabilitiesProperties() {
72 Map<String, PropertyDataDefinition> mapToscaDataDefinition = new HashMap<>();
73 PropertyDataDefinition propertyDataDefinition = new PropertyDataDefinition();
74 propertyDataDefinition.setUniqueId("ComponentInput1_uniqueId");
75 propertyDataDefinition.setName("propName");
76 mapToscaDataDefinition.put(propertyDataDefinition.getUniqueId(), propertyDataDefinition);
77 MapPropertiesDataDefinition mapPropertiesDataDefinition = new MapPropertiesDataDefinition(mapToscaDataDefinition);
79 Map<String, MapPropertiesDataDefinition> propertiesMap = new HashMap<>();
80 propertiesMap.put(propertyDataDefinition.getUniqueId(), mapPropertiesDataDefinition);
82 StorageOperationStatus operationStatus = operation.createOrUpdateCapabilityProperties("componentId",
85 Assert.assertEquals(StorageOperationStatus.OK, operationStatus);