1e2558b580e295ac0581e0de787d77b170e4e6a0
[so.git] /
1 package org.onap.so.bpmn.infrastructure.workflow.tasks;
2
3 import static org.junit.Assert.assertEquals;
4 import static org.mockito.ArgumentMatchers.any;
5 import static org.mockito.ArgumentMatchers.anyString;
6 import static org.mockito.ArgumentMatchers.isA;
7 import static org.mockito.Mockito.doReturn;
8 import static org.mockito.Mockito.when;
9
10 import java.util.ArrayList;
11 import java.util.List;
12
13 import org.camunda.bpm.engine.delegate.DelegateExecution;
14 import org.camunda.bpm.extension.mockito.delegate.DelegateExecutionFake;
15 import org.junit.Before;
16 import org.junit.Rule;
17 import org.junit.Test;
18 import org.junit.rules.ExpectedException;
19 import org.mockito.InjectMocks;
20 import org.mockito.Spy;
21 import org.onap.so.bpmn.BaseTaskTest;
22 import org.onap.so.bpmn.servicedecomposition.entities.BuildingBlock;
23 import org.onap.so.bpmn.servicedecomposition.entities.ExecuteBuildingBlock;
24 import org.onap.so.db.request.beans.InfraActiveRequests;
25
26 public class WorkflowActionBBTasksUpdateReqDbTest extends BaseTaskTest{
27
28         protected WorkflowAction workflowAction = new WorkflowAction();
29         
30         @Spy
31         @InjectMocks
32         protected WorkflowActionBBTasks workflowActionBBTasks;
33         
34         private DelegateExecution execution;
35         
36         @Rule
37         public ExpectedException thrown = ExpectedException.none();
38         
39         @Before
40         public void before() throws Exception {
41                 execution = new DelegateExecutionFake();
42                 org.onap.aai.domain.yang.ServiceInstance servInstance = new org.onap.aai.domain.yang.ServiceInstance();
43                 servInstance.setServiceInstanceId("TEST");
44                 when(bbSetupUtils.getAAIServiceInstanceByName(anyString(), any())).thenReturn(servInstance);
45                 workflowAction.setBbInputSetupUtils(bbSetupUtils);
46                 workflowAction.setBbInputSetup(bbInputSetup);
47         }
48         
49         @Test
50         public void getUpdatedRequestTest() throws Exception{
51                 List<ExecuteBuildingBlock> flowsToExecute = new ArrayList();
52                 ExecuteBuildingBlock ebb1 = new ExecuteBuildingBlock();
53                 BuildingBlock bb1 = new BuildingBlock();
54                 bb1.setBpmnFlowName("CreateNetworkBB");
55                 flowsToExecute.add(ebb1);
56                 ebb1.setBuildingBlock(bb1);
57                 ExecuteBuildingBlock ebb2 = new ExecuteBuildingBlock();
58                 BuildingBlock bb2 = new BuildingBlock();
59                 bb2.setBpmnFlowName("ActivateNetworkBB");
60                 flowsToExecute.add(ebb2);
61                 ebb2.setBuildingBlock(bb2);
62                 String requestId = "requestId";
63                 execution.setVariable("mso-request-id", requestId);
64                 execution.setVariable("flowsToExecute", flowsToExecute);
65                 int currentSequence = 2;
66                 String expectedStatusMessage = "Execution of CreateNetworkBB has completed successfully, next invoking ActivateNetworkBB (Execution Path progress: BBs completed = 1; BBs remaining = 1).";
67                 Long expectedLong = new Long(52);
68                 InfraActiveRequests mockedRequest = new InfraActiveRequests();
69                 doReturn(mockedRequest).when(requestsDbClient).getInfraActiveRequestbyRequestId(isA(String.class));
70                 InfraActiveRequests actual = workflowActionBBTasks.getUpdatedRequest(execution, currentSequence);
71                 assertEquals(expectedStatusMessage, actual.getStatusMessage());
72                 assertEquals(expectedLong, actual.getProgress());
73         }
74 }