Upgrade SDC from Titan to Janus Graph
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / components / impl / AdditionalInformationBusinessLogicTest.java
1 /*
2
3  * Copyright (c) 2018 AT&T Intellectual Property.
4
5  *
6
7  * Licensed under the Apache License, Version 2.0 (the "License");
8
9  * you may not use this file except in compliance with the License.
10
11  * You may obtain a copy of the License at
12
13  *
14
15  *     http://www.apache.org/licenses/LICENSE-2.0
16
17  *
18
19  * Unless required by applicable law or agreed to in writing, software
20
21  * distributed under the License is distributed on an "AS IS" BASIS,
22
23  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
24
25  * See the License for the specific language governing permissions and
26
27  * limitations under the License.
28
29  */
30 package org.openecomp.sdc.be.components.impl;
31
32 import fj.data.Either;
33 import org.junit.Assert;
34 import org.junit.Test;
35 import org.junit.runner.RunWith;
36 import org.mockito.ArgumentMatchers;
37 import org.mockito.InjectMocks;
38 import org.mockito.Mock;
39 import org.mockito.junit.MockitoJUnitRunner;
40 import org.openecomp.sdc.be.components.validation.UserValidations;
41 import org.openecomp.sdc.be.config.Configuration;
42 import org.openecomp.sdc.be.config.ConfigurationManager;
43 import org.openecomp.sdc.be.dao.jsongraph.JanusGraphDao;
44 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
45 import org.openecomp.sdc.be.datatypes.elements.AdditionalInfoParameterInfo;
46 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
47 import org.openecomp.sdc.be.impl.ComponentsUtils;
48 import org.openecomp.sdc.be.model.Resource;
49 import org.openecomp.sdc.be.model.AdditionalInformationDefinition;
50 import org.openecomp.sdc.be.model.User;
51 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
52 import org.openecomp.sdc.be.model.operations.api.IAdditionalInformationOperation;
53 import org.openecomp.sdc.be.model.operations.api.IGraphLockOperation;
54 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
55 import org.openecomp.sdc.common.api.ConfigurationSource;
56 import org.openecomp.sdc.common.impl.ExternalConfiguration;
57 import org.openecomp.sdc.common.impl.FSConfigurationSource;
58 import org.openecomp.sdc.exception.ResponseFormat;
59
60 import java.util.ArrayList;
61 import java.util.List;
62
63 import static org.mockito.ArgumentMatchers.anyString;
64 import static org.mockito.ArgumentMatchers.eq;
65 import static org.mockito.Mockito.when;
66
67 @RunWith(MockitoJUnitRunner.class)
68 public class AdditionalInformationBusinessLogicTest {
69         @InjectMocks
70         private AdditionalInformationBusinessLogic additionalInformationBusinessLogic;
71         @Mock
72         private IGraphLockOperation graphLockOperation;
73         @Mock
74         private UserValidations userValidations;
75         @Mock
76         private ToscaOperationFacade toscaOperationFacade;
77         @Mock
78         private ComponentsUtils componentsUtils;
79         @Mock
80         private JanusGraphDao janusGraphDao;
81         @Mock
82         private IAdditionalInformationOperation additionalInformationOperation;
83         @Mock
84         private Resource resource;
85
86         NodeTypeEnum nodeType = NodeTypeEnum.Resource;
87         static ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), "src/test/resources/config/catalog-be");
88
89         @Test
90         public void testCreateAdditionalInformation() {
91                 Either<AdditionalInfoParameterInfo, ResponseFormat> result;
92                 String resourceId = "resourceId";
93                 AdditionalInfoParameterInfo additionalInfoParameterInfo = getAdditionalInfoParamInfo();
94                 AdditionalInformationDefinition additionalInformationDefinition = getAdditionalInfoDef(additionalInfoParameterInfo);
95                 User user = new User();
96                 String userId = "userId";
97                 user.setUserId(userId);
98                 ConfigurationManager configurationManager = new ConfigurationManager(configurationSource);
99                 configurationManager.setConfiguration(new Configuration());
100                 configurationManager.getConfiguration().setAdditionalInformationMaxNumberOfKeys(0);
101                 when(additionalInformationBusinessLogic.validateUserExists(anyString(), anyString(), eq(false))).thenReturn(user);
102                 when(toscaOperationFacade.getToscaElement(ArgumentMatchers.eq("resourceId"), ArgumentMatchers.eq(JsonParseFlagEnum.ParseMetadata))).thenReturn(Either.left(resource));
103                 when(graphLockOperation.lockComponent(resourceId, nodeType)).thenReturn(StorageOperationStatus.OK);
104                 when(additionalInformationOperation.getNumberOfAdditionalInformationParameters(nodeType,resourceId,true)).thenReturn(Either.left(0));
105                 when(additionalInformationOperation.createAdditionalInformationParameter(nodeType, resourceId, additionalInfoParameterInfo.getKey(),additionalInfoParameterInfo.getValue(), true)).thenReturn(Either.left(additionalInformationDefinition));
106
107                 result = additionalInformationBusinessLogic.createAdditionalInformation(nodeType,resourceId,additionalInfoParameterInfo,userId);
108                 Assert.assertTrue(result.isLeft());
109         }
110
111         @Test
112         public void testUpdateAdditionalInformation() {
113                 Either<AdditionalInfoParameterInfo, ResponseFormat> result;
114                 String resourceId = "resourceId";
115                 AdditionalInfoParameterInfo additionalInfoParameterInfo = getAdditionalInfoParamInfo();
116                 additionalInfoParameterInfo.setUniqueId(resourceId);
117                 AdditionalInformationDefinition additionalInformationDefinition = getAdditionalInfoDef(additionalInfoParameterInfo);
118                 User user = new User();
119                 String userId = "userId";
120                 user.setUserId(userId);
121                 when(toscaOperationFacade.getToscaElement(ArgumentMatchers.eq("resourceId"), ArgumentMatchers.eq(JsonParseFlagEnum.ParseMetadata))).thenReturn(Either.left(resource));
122                 when(graphLockOperation.lockComponent(resourceId, nodeType)).thenReturn(StorageOperationStatus.OK);
123                 when(additionalInformationOperation.updateAdditionalInformationParameter(nodeType, resourceId, additionalInfoParameterInfo.getUniqueId(), additionalInfoParameterInfo.getKey(), additionalInfoParameterInfo.getValue(), true)).thenReturn(Either.left(additionalInformationDefinition));
124
125                 result = additionalInformationBusinessLogic.updateAdditionalInformation(nodeType, resourceId, additionalInfoParameterInfo, userId);
126                 Assert.assertTrue(result.isLeft());
127         }
128
129         @Test
130         public void testDeleteAdditionalInformation() {
131                 Either<AdditionalInfoParameterInfo, ResponseFormat> result;
132                 String resourceId = "resourceId";
133                 String userId = "userId";
134                 AdditionalInfoParameterInfo additionalInfoParameterInfo = getAdditionalInfoParamInfo();
135                 AdditionalInformationDefinition additionalInformationDefinition = getAdditionalInfoDef(additionalInfoParameterInfo);
136                 when(toscaOperationFacade.getToscaElement(ArgumentMatchers.eq("resourceId"), ArgumentMatchers.eq(JsonParseFlagEnum.ParseMetadata))).thenReturn(Either.left(resource));
137                 when(graphLockOperation.lockComponent(resourceId, nodeType)).thenReturn(StorageOperationStatus.OK);
138                 when(additionalInformationOperation.getAdditionalInformationParameter(nodeType, resourceId, additionalInfoParameterInfo.getUniqueId(), true)).thenReturn(Either.left(additionalInfoParameterInfo));
139                 when(additionalInformationOperation.deleteAdditionalInformationParameter(nodeType, resourceId, additionalInfoParameterInfo.getUniqueId(), true)).thenReturn(Either.left(additionalInformationDefinition));
140
141                 result = additionalInformationBusinessLogic.deleteAdditionalInformation(nodeType, resourceId, additionalInfoParameterInfo, userId);
142                 Assert.assertTrue(result.isLeft());
143         }
144
145         @Test
146         public void testGetAdditionalInformation() {
147                 Either<AdditionalInfoParameterInfo, ResponseFormat> result;
148                 String resourceId = "resourceId";
149                 String userId = "userId";
150                 AdditionalInfoParameterInfo additionalInfoParameterInfo = getAdditionalInfoParamInfo();
151                 when(additionalInformationOperation.getAdditionalInformationParameter(nodeType, resourceId, additionalInfoParameterInfo.getUniqueId(), true)).thenReturn(Either.left(additionalInfoParameterInfo));
152
153                 result = additionalInformationBusinessLogic.getAdditionalInformation(nodeType, resourceId, additionalInfoParameterInfo, userId);
154                 Assert.assertTrue(result.isLeft());
155         }
156
157         @Test
158         public void testGetAllAdditionalInformation() {
159                 Either<AdditionalInformationDefinition, ResponseFormat> result;
160                 String resourceId = "resourceId";
161                 String userId = "userId";
162                 AdditionalInfoParameterInfo additionalInfoParameterInfo = getAdditionalInfoParamInfo();
163                 AdditionalInformationDefinition additionalInformationDefinition = getAdditionalInfoDef(additionalInfoParameterInfo);
164                 when(additionalInformationOperation.getAllAdditionalInformationParameters(nodeType, resourceId, false)).thenReturn(Either.left(additionalInformationDefinition));
165                 result = additionalInformationBusinessLogic.getAllAdditionalInformation(nodeType, resourceId, userId);
166                 Assert.assertTrue(result.isLeft());
167         }
168
169         private AdditionalInfoParameterInfo getAdditionalInfoParamInfo() {
170                 AdditionalInfoParameterInfo additionalInfoParameterInfo = new AdditionalInfoParameterInfo();
171                 additionalInfoParameterInfo.setKey("key");
172                 additionalInfoParameterInfo.setValue("value");
173                 return additionalInfoParameterInfo;
174         }
175
176         private AdditionalInformationDefinition getAdditionalInfoDef(AdditionalInfoParameterInfo additionalInfoParameterInfo) {
177                 AdditionalInformationDefinition additionalInformationDefinition = new AdditionalInformationDefinition();
178                 List<AdditionalInfoParameterInfo> parameters = new ArrayList<>();
179                 parameters.add(additionalInfoParameterInfo);
180                 additionalInformationDefinition.setParameters(parameters);
181                 return additionalInformationDefinition;
182         }
183 }