gerrit.onap Code Review - so.git/blob

   1 package org.onap.so.adapters.tasks.orchestration;
   2
   3 import org.camunda.bpm.client.task.ExternalTask;
   4 import org.camunda.bpm.client.task.ExternalTaskService;
   5 import org.junit.Test;
   6 import org.junit.runner.RunWith;
   7 import org.mockito.InjectMocks;
   8 import org.mockito.Mock;
   9 import org.mockito.Mockito;
  10 import static org.mockito.Mockito.any;
  11 import static org.mockito.Mockito.eq;
  12 import java.io.IOException;
  13 import java.nio.file.Files;
  14 import java.nio.file.Paths;
  15 import org.mockito.junit.MockitoJUnitRunner;
  16 import org.onap.so.adapters.tasks.orchestration.PollService;
  17 import org.onap.so.adapters.vnf.VnfAdapterUtils;
  18 import org.onap.so.logging.tasks.AuditMDCSetup;
  19 import org.onap.so.openstack.exceptions.MsoException;
  20 import org.onap.so.openstack.utils.MsoHeatUtils;
  21 import com.woorea.openstack.heat.model.Stack;
  22
  23 @RunWith(MockitoJUnitRunner.class)
  24 public class PollServiceTest {
  25
  26     private String RESOURCE_PATH = "src/test/resources/__files/";
  27
  28     @Mock
  29     private ExternalTask mockExternalTask;
  30
  31     @Mock
  32     private ExternalTaskService mockExternalTaskService;
  33
  34     @Mock
  35     private MsoHeatUtils msoHeatUtils;
  36
  37     @Mock
  38     private VnfAdapterUtils vnfAdapterUtils;
  39
  40     @Mock
  41     private AuditMDCSetup mdcSetup;
  42
  43     @InjectMocks
  44     private PollService pollService;
  45
  46     @Test
  47     public void testExecuteExternalTask() throws MsoException, IOException {
  48         String xmlString =
  49                 new String(Files.readAllBytes(Paths.get(RESOURCE_PATH + "/vnfAdapterTaskRequestCreate.xml")));
  50
  51         Mockito.when(vnfAdapterUtils.isMulticloudMode(Mockito.any(), Mockito.any())).thenReturn(false);
  52         Mockito.when(mockExternalTask.getVariable("openstackAdapterTaskRequest")).thenReturn(xmlString);
  53         Mockito.when(mockExternalTask.getVariable("PollRollbackStatus")).thenReturn(false);
  54         Mockito.when(mockExternalTask.getVariable("stackId")).thenReturn("stackId/stack123");
  55         Mockito.when(msoHeatUtils.pollStackForStatus(eq(1), any(Stack.class), eq("CREATE_IN_PROGRESS"), eq("regionOne"),
  56                 eq("0422ffb57ba042c0800a29dc85ca70f8"), eq(false))).thenReturn(new Stack());
  57         // Mockito.doNothing().when(msoHeatUtils).postProcessStackCreate(Mockito.any(), Mockito.any(), Mockito.any(),
  58         // Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any());
  59
  60         pollService.executeExternalTask(mockExternalTask, mockExternalTaskService);
  61
  62         Mockito.verify(msoHeatUtils).pollStackForStatus(eq(1), any(Stack.class), eq("CREATE_IN_PROGRESS"),
  63                 eq("regionOne"), eq("0422ffb57ba042c0800a29dc85ca70f8"), eq(false));
  64
  65     }
  66
  67     @Test
  68     public void testExecuteExternalTask_rollback() throws MsoException, IOException {
  69         String xmlString =
  70                 new String(Files.readAllBytes(Paths.get(RESOURCE_PATH + "/vnfAdapterTaskRequestCreate.xml")));
  71
  72         Mockito.when(mockExternalTask.getVariable("openstackAdapterTaskRequest")).thenReturn(xmlString);
  73         Mockito.when(mockExternalTask.getVariable("PollRollbackStatus")).thenReturn(true);
  74         Mockito.when(mockExternalTask.getVariable("stackId")).thenReturn("stackId/stack123");
  75         Mockito.when(msoHeatUtils.pollStackForStatus(eq(1), any(), eq("DELETE_IN_PROGRESS"), eq("regionOne"),
  76                 eq("0422ffb57ba042c0800a29dc85ca70f8"), eq(true))).thenReturn(new Stack());
  77         Mockito.doNothing().when(msoHeatUtils).postProcessStackDelete(Mockito.any());
  78
  79
  80         pollService.executeExternalTask(mockExternalTask, mockExternalTaskService);
  81
  82         Mockito.verify(msoHeatUtils).pollStackForStatus(eq(1), any(), eq("DELETE_IN_PROGRESS"), eq("regionOne"),
  83                 eq("0422ffb57ba042c0800a29dc85ca70f8"), eq(true));
  84
  85     }
  86
  87 }