catalog-be/src/test/java/org/openecomp/sdc/be/components/lifecycle/LifecycleTestBase.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. 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.components.lifecycle;
  22
  23 import fj.data.Either;
  24 import org.junit.BeforeClass;
  25 import org.mockito.InjectMocks;
  26 import org.mockito.Mockito;
  27 import org.mockito.invocation.InvocationOnMock;
  28 import org.mockito.stubbing.Answer;
  29 import org.openecomp.sdc.be.auditing.impl.AuditingManager;
  30 import org.openecomp.sdc.be.components.impl.ArtifactsBusinessLogic;
  31 import org.openecomp.sdc.be.components.impl.ResponseFormatManager;
  32 import org.openecomp.sdc.be.config.ConfigurationManager;
  33 import org.openecomp.sdc.be.dao.api.ActionStatus;
  34 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
  35 import org.openecomp.sdc.be.datatypes.components.ResourceMetadataDataDefinition;
  36 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
  37 import org.openecomp.sdc.be.datatypes.enums.ResourceTypeEnum;
  38 import org.openecomp.sdc.be.impl.ComponentsUtils;
  39 import org.openecomp.sdc.be.impl.WebAppContextWrapper;
  40 import org.openecomp.sdc.be.model.*;
  41 import org.openecomp.sdc.be.model.jsontitan.datamodel.ToscaElement;
  42 import org.openecomp.sdc.be.model.jsontitan.datamodel.ToscaElementTypeEnum;
  43 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaElementLifecycleOperation;
  44 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
  45 import org.openecomp.sdc.be.model.jsontitan.utils.ModelConverter;
  46 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  47 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
  48 import org.openecomp.sdc.be.user.Role;
  49 import org.openecomp.sdc.be.user.UserBusinessLogic;
  50 import org.openecomp.sdc.common.api.ConfigurationSource;
  51 import org.openecomp.sdc.common.api.Constants;
  52 import org.openecomp.sdc.common.impl.ExternalConfiguration;
  53 import org.openecomp.sdc.common.impl.FSConfigurationSource;
  54 import org.openecomp.sdc.exception.ResponseFormat;
  55 import org.slf4j.Logger;
  56 import org.slf4j.LoggerFactory;
  57 import org.springframework.web.context.WebApplicationContext;
  58
  59 import javax.servlet.ServletContext;
  60 import java.util.ArrayList;
  61 import java.util.List;
  62
  63 import static org.mockito.Mockito.when;
  64
  65 public class LifecycleTestBase {
  66     private static final Logger log = LoggerFactory.getLogger(LifecycleTestBase.class);
  67     @InjectMocks
  68     protected final ServletContext servletContext = Mockito.mock(ServletContext.class);
  69     protected UserBusinessLogic mockUserAdmin = Mockito.mock(UserBusinessLogic.class);
  70     protected WebAppContextWrapper webAppContextWrapper = Mockito.mock(WebAppContextWrapper.class);
  71     protected WebApplicationContext webAppContext = Mockito.mock(WebApplicationContext.class);
  72     protected ToscaElementLifecycleOperation toscaElementLifecycleOperation = Mockito.mock(ToscaElementLifecycleOperation.class);
  73     protected ArtifactsBusinessLogic artifactsManager = Mockito.mock(ArtifactsBusinessLogic.class);;
  74     protected User user = null;
  75     protected Resource resourceResponse;
  76     protected Service serviceResponse;
  77     protected static ConfigurationManager configurationManager = null;
  78     protected ResponseFormatManager responseManager = null;
  79     protected TitanDao titanDao = Mockito.mock(TitanDao.class);
  80     protected ToscaOperationFacade toscaOperationFacade = Mockito.mock(ToscaOperationFacade.class);
  81     protected static ComponentsUtils componentsUtils;
  82
  83     @BeforeClass
  84     public static void setupClass() {
  85         ExternalConfiguration.setAppName("catalog-be");
  86
  87         // init Configuration
  88         String appConfigDir = "src/test/resources/config/catalog-be";
  89         ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), appConfigDir);
  90         configurationManager = new ConfigurationManager(configurationSource);
  91
  92         componentsUtils = new ComponentsUtils(Mockito.mock(AuditingManager.class));
  93     }
  94
  95     public void setup() {
  96
  97         // User data and management
  98         user = new User();
  99         user.setUserId("jh003");
 100         user.setFirstName("Jimmi");
 101         user.setLastName("Hendrix");
 102         user.setRole(Role.ADMIN.name());
 103
 104         Either<User, ActionStatus> eitherGetUser = Either.left(user);
 105         when(mockUserAdmin.getUser("jh003", false)).thenReturn(eitherGetUser);
 106
 107         // Servlet Context attributes
 108         when(servletContext.getAttribute(Constants.CONFIGURATION_MANAGER_ATTR)).thenReturn(configurationManager);
 109         when(servletContext.getAttribute(Constants.WEB_APPLICATION_CONTEXT_WRAPPER_ATTR)).thenReturn(webAppContextWrapper);
 110         when(webAppContextWrapper.getWebAppContext(servletContext)).thenReturn(webAppContext);
 111         when(webAppContext.getBean(ToscaElementLifecycleOperation.class)).thenReturn(toscaElementLifecycleOperation);
 112         when(webAppContext.getBean(ArtifactsBusinessLogic.class)).thenReturn(artifactsManager);
 113
 114         // createResource
 115         resourceResponse = createResourceObject();
 116         Either<ToscaElement, StorageOperationStatus> eitherComponent = Either.left(ModelConverter.convertToToscaElement(resourceResponse));
 117         when(toscaElementLifecycleOperation.checkoutToscaElement(Mockito.any(String.class), Mockito.any(String.class), Mockito.any(String.class)))
 118                 .thenAnswer(createAnswer(eitherComponent));
 119
 120         when(toscaElementLifecycleOperation.checkinToscaELement(Mockito.any(LifecycleStateEnum.class), Mockito.any(String.class), Mockito.any(String.class), Mockito.any(String.class)))
 121                 .thenAnswer(createAnswer(eitherComponent));
 122
 123         when(toscaElementLifecycleOperation.requestCertificationToscaElement(Mockito.any(String.class), Mockito.any(String.class), Mockito.any(String.class)))
 124                 .thenAnswer(createAnswer(eitherComponent));
 125
 126         Either<User, StorageOperationStatus> getOwnerResult = Either.left(user);
 127         when(toscaElementLifecycleOperation.getToscaElementOwner(Mockito.anyString())).thenReturn(getOwnerResult);
 128
 129         Either<Component, StorageOperationStatus> eitherlatestDerived = Either.right(StorageOperationStatus.OK);
 130         when(toscaOperationFacade.shouldUpgradeToLatestDerived(Mockito.any(Resource.class))).thenReturn(eitherlatestDerived);
 131
 132         responseManager = ResponseFormatManager.getInstance();
 133
 134     }
 135
 136     public static <T> Answer<T> createAnswer(final T value) {
 137         return new Answer<T>() {
 138             @Override
 139             public T answer(InvocationOnMock invocation) throws Throwable {
 140                 return value;
 141             }
 142
 143         };
 144     }
 145
 146     protected Resource createResourceObject() {
 147         return createResourceObject(ComponentTypeEnum.RESOURCE);
 148     }
 149
 150     protected Resource createResourceObject(ComponentTypeEnum componentType) {
 151         Resource resource = new Resource();
 152         resource.setUniqueId("uid");
 153         resource.setComponentType(componentType);
 154         resource.setName("MyResourceName");
 155         resource.setUniqueId("uid");
 156         resource.addCategory("VoIP", "INfra");
 157         resource.setDescription("My short description");
 158         List<String> tgs = new ArrayList<>();
 159         tgs.add("test");
 160         resource.setTags(tgs);
 161         List<String> template = new ArrayList<>();
 162         template.add("Root");
 163         resource.setDerivedFrom(template);
 164         resource.setVendorName("Motorola");
 165         resource.setVendorRelease("1.0.0");
 166         resource.setContactId("yavivi");
 167         resource.setIcon("MyIcon.jpg");
 168         resource.setToscaType(ToscaElementTypeEnum.TOPOLOGY_TEMPLATE.getValue());
 169
 170         return resource;
 171     }
 172
 173     protected Resource createResourceVFCMTObject() {
 174         ResourceMetadataDataDefinition rMetadataDataDefinition = new ResourceMetadataDataDefinition();
 175         rMetadataDataDefinition.setResourceType(ResourceTypeEnum.VFCMT);
 176         ComponentMetadataDefinition cMetadataDataDefinition = new ComponentMetadataDefinition(rMetadataDataDefinition) ;
 177
 178         Resource resource = new Resource(cMetadataDataDefinition);
 179         resource.setUniqueId("rid");
 180         resource.setName("MyResourceVFCMTName");
 181         resource.addCategory("VoIP", "INfra");
 182         resource.setDescription("My short description");
 183         List<String> tgs = new ArrayList<>();
 184         tgs.add("test1");
 185         resource.setTags(tgs);
 186         List<String> template = new ArrayList<>();
 187         template.add("Root");
 188         resource.setDerivedFrom(template);
 189         resource.setVendorName("Motorola");
 190         resource.setVendorRelease("1.0.0");
 191         resource.setContactId("yavivi");
 192         resource.setIcon("MyIcon.jpg");
 193         resource.setToscaType(ToscaElementTypeEnum.NODE_TYPE.getValue());
 194
 195         return resource;
 196     }
 197
 198     protected Service createServiceObject() {
 199         Service service = new Service();
 200         service.setName("MyServiceName");
 201         service.setUniqueId("sid");
 202         service.addCategory("VoIP", null);
 203         service.setDescription("My short description");
 204         List<String> tgs = new ArrayList<>();
 205         tgs.add("test");
 206         service.setTags(tgs);
 207         List<String> template = new ArrayList<>();
 208         template.add("Root");
 209         service.setContactId("aa0001");
 210         service.setIcon("MyIcon.jpg");
 211
 212         return service;
 213     }
 214
 215     protected void assertResponse(Either<? extends Component, ResponseFormat> createResponse, ActionStatus expectedStatus, String... variables) {
 216         ResponseFormat expectedResponse = responseManager.getResponseFormat(expectedStatus, variables);
 217         ResponseFormat actualResponse = createResponse.right().value();
 218     }
 219
 220     protected void assertServiceResponse(Either<Service, ResponseFormat> createResponse, ActionStatus expectedStatus, String... variables) {
 221         ResponseFormat expectedResponse = responseManager.getResponseFormat(expectedStatus, variables);
 222         ResponseFormat actualResponse = createResponse.right().value();
 223     }
 224
 225     protected static ArtifactDefinition getArtifactPlaceHolder(String resourceId, String logicalName) {
 226         ArtifactDefinition artifact = new ArtifactDefinition();
 227
 228         artifact.setUniqueId(UniqueIdBuilder.buildPropertyUniqueId(resourceId, logicalName.toLowerCase()));
 229         artifact.setArtifactLabel(logicalName.toLowerCase());
 230
 231         return artifact;
 232     }
 233 }