gerrit.onap Code Review - so.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   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.onap.so.bpmn.infrastructure.scripts
  22
  23
  24 import org.camunda.bpm.engine.delegate.BpmnError
  25 import org.camunda.bpm.engine.impl.persistence.entity.ExecutionEntity
  26 import org.junit.Assert
  27 import org.junit.Before
  28 import org.junit.Rule
  29 import org.junit.Test
  30 import org.junit.rules.ExpectedException
  31 import org.junit.runner.RunWith
  32 import org.mockito.ArgumentCaptor
  33 import org.mockito.Captor
  34 import org.mockito.MockitoAnnotations
  35 import org.mockito.runners.MockitoJUnitRunner
  36 import org.onap.aai.domain.yang.GenericVnf
  37 import org.onap.aai.domain.yang.VfModule
  38 import org.onap.aai.domain.yang.VolumeGroup
  39 import org.onap.so.bpmn.common.scripts.MsoGroovyTest
  40 import org.onap.so.bpmn.mock.FileUtil
  41 import org.onap.aaiclient.client.aai.AAIObjectType
  42 import org.onap.aaiclient.client.aai.entities.AAIResultWrapper
  43 import org.onap.aaiclient.client.aai.entities.uri.AAIResourceUri
  44 import org.onap.aaiclient.client.aai.entities.uri.AAIUriFactory
  45 import org.onap.aaiclient.client.generated.fluentbuilders.AAIFluentTypeBuilder
  46 import org.onap.aaiclient.client.generated.fluentbuilders.AAIFluentTypeBuilder.Types
  47 import org.onap.so.constants.Defaults
  48
  49 import javax.ws.rs.core.UriBuilder
  50
  51 import static org.mockito.Mockito.*
  52
  53 @RunWith(MockitoJUnitRunner.class)
  54 class UpdateVfModuleVolumeTest extends MsoGroovyTest{
  55
  56     def prefix = "UPDVfModVol_"
  57     @Captor
  58     static ArgumentCaptor<ExecutionEntity> captor = ArgumentCaptor.forClass(ExecutionEntity.class)
  59
  60     @Rule
  61     public ExpectedException thrown = ExpectedException.none()
  62
  63
  64         @Before
  65         public void init(){
  66         super.init("UpdateVfModuleVolume")
  67                 MockitoAnnotations.initMocks(this)
  68     }
  69
  70
  71     @Test
  72     void testQueryAAIForVolumeGroup() {
  73         String aicCloudRegion = "aicCloudRegionId"
  74         String volumeGroupId = "volumeGroupId"
  75         when(mockExecution.getVariable("UPDVfModVol_volumeGroupId")).thenReturn(volumeGroupId)
  76         when(mockExecution.getVariable("UPDVfModVol_aicCloudRegion")).thenReturn(aicCloudRegion)
  77
  78         UpdateVfModuleVolume obj = spy(UpdateVfModuleVolume.class)
  79         when(obj.getAAIClient()).thenReturn(client)
  80         AAIResourceUri uri = AAIUriFactory.createResourceUri(AAIFluentTypeBuilder.cloudInfrastructure().cloudRegion(Defaults.CLOUD_OWNER.toString(), aicCloudRegion).volumeGroup(volumeGroupId))
  81         VolumeGroup volumeGroup = new VolumeGroup();
  82         volumeGroup.setVolumeGroupId(volumeGroupId)
  83
  84         AAIResultWrapper wrapper = new AAIResultWrapper(FileUtil.readResourceFile("__files/aai/VolumeGroupWithTenant.json"))
  85         when(client.get(uri)).thenReturn(wrapper)
  86         obj.queryAAIForVolumeGroup(mockExecution)
  87         verify(mockExecution).setVariable("UPDVfModVol_volumeGroupHeatStackId","heatStackId")
  88     }
  89
  90     @Test
  91     void testQueryAAIForVolumeGroupNoTenant() {
  92         String aicCloudRegion = "aicCloudRegionId"
  93         String volumeGroupId = "volumeGroupId"
  94         when(mockExecution.getVariable("UPDVfModVol_volumeGroupId")).thenReturn(volumeGroupId)
  95         when(mockExecution.getVariable("UPDVfModVol_aicCloudRegion")).thenReturn(aicCloudRegion)
  96
  97         UpdateVfModuleVolume obj = spy(UpdateVfModuleVolume.class)
  98         when(obj.getAAIClient()).thenReturn(client)
  99         AAIResourceUri uri = AAIUriFactory.createResourceUri(AAIFluentTypeBuilder.cloudInfrastructure().cloudRegion(Defaults.CLOUD_OWNER.toString(), aicCloudRegion).volumeGroup(volumeGroupId))
 100         VolumeGroup volumeGroup = new VolumeGroup();
 101         volumeGroup.setVolumeGroupId(volumeGroupId)
 102
 103         AAIResultWrapper wrapper = new AAIResultWrapper(FileUtil.readResourceFile("__files/aai/VolumeGroupWithTenant.json"))
 104         when(client.get(uri)).thenThrow(Exception.class)
 105         thrown.expect(BpmnError.class)
 106         obj.queryAAIForVolumeGroup(mockExecution)
 107     }
 108 }