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