catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/ComponentLockerTest.java

   1 /*-\r
   2  * ============LICENSE_START=======================================================\r
   3  * SDC\r
   4  * ================================================================================\r
   5  * Copyright (C) 2020 Fujitsu Ltd. All rights reserved.\r
   6  * ================================================================================\r
   7  * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * you may not use this file except in compliance with the License.\r
   9  * You may obtain a copy of the License at\r
  10  *\r
  11  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  *\r
  13  * Unless required by applicable law or agreed to in writing, software\r
  14  * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * See the License for the specific language governing permissions and\r
  17  * limitations under the License.\r
  18  * ============LICENSE_END=========================================================\r
  19  */\r
  20 package org.openecomp.sdc.be.components.impl;\r
  21 \r
  22 import org.junit.Before;\r
  23 import org.junit.Test;\r
  24 import org.mockito.Mock;\r
  25 import org.mockito.InjectMocks;\r
  26 import org.mockito.MockitoAnnotations;\r
  27 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;\r
  28 import org.openecomp.sdc.be.model.Component;\r
  29 import org.openecomp.sdc.be.model.operations.StorageException;\r
  30 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;\r
  31 import org.openecomp.sdc.be.model.operations.impl.GraphLockOperation;\r
  32 import org.openecomp.sdc.be.model.Service;\r
  33 import static org.mockito.Mockito.when;\r
  34 import static org.mockito.Mockito.verify;\r
  35 import static org.mockito.Mockito.times;\r
  36 \r
  37 public class ComponentLockerTest {\r
  38 \r
  39     private static final String USER_ID = "userId";\r
  40 \r
  41     @Mock\r
  42     private GraphLockOperation graphLockOperation;\r
  43 \r
  44     @InjectMocks\r
  45     private ComponentLocker testInstance;\r
  46 \r
  47     @Before\r
  48     public void init() {\r
  49         MockitoAnnotations.openMocks(this);\r
  50     }\r
  51 \r
  52     @Test\r
  53     public void lock_Component() {\r
  54         Component component= new Service();\r
  55         component.setComponentType(ComponentTypeEnum.SERVICE);\r
  56         component.setUniqueId(USER_ID);\r
  57         when(graphLockOperation.lockComponent(USER_ID, component.getComponentType().getNodeType())).thenReturn(StorageOperationStatus.OK);\r
  58         testInstance.lock(component);\r
  59         verify(graphLockOperation, times(1)).lockComponent(USER_ID, component.getComponentType().getNodeType());\r
  60     }\r
  61 \r
  62     @Test(expected = StorageException.class)\r
  63     public void lock_Component_Exception() {\r
  64         Component component= new Service();\r
  65         component.setComponentType(ComponentTypeEnum.SERVICE);\r
  66         component.setUniqueId(USER_ID);\r
  67         when(graphLockOperation.lockComponent(USER_ID, component.getComponentType().getNodeType())).thenReturn(StorageOperationStatus.INVALID_VALUE);\r
  68         testInstance.lock(component);\r
  69     }\r
  70 \r
  71 }\r