cc96326563650a6ed25d7d05922da9b67623cf2d
[so.git] / bpmn / so-bpmn-tasks / src / test / java / org / onap / so / bpmn / infrastructure / sdnc / tasks / SDNCAssignTasksTest.java
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.sdnc.tasks;
22
23 import static org.junit.Assert.assertTrue;
24 import static org.mockito.Mockito.doReturn;
25 import static org.mockito.Mockito.doThrow;
26 import static org.mockito.Mockito.times;
27 import static org.mockito.Mockito.verify;
28
29 import org.camunda.bpm.engine.delegate.BpmnError;
30 import org.junit.Before;
31 import org.junit.Test;
32 import org.onap.so.bpmn.BaseTaskTest;
33 import org.onap.so.bpmn.servicedecomposition.bbobjects.CloudRegion;
34 import org.onap.so.bpmn.servicedecomposition.bbobjects.Customer;
35 import org.onap.so.bpmn.servicedecomposition.bbobjects.GenericVnf;
36 import org.onap.so.bpmn.servicedecomposition.bbobjects.L3Network;
37 import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
38 import org.onap.so.bpmn.servicedecomposition.bbobjects.VfModule;
39 import org.onap.so.bpmn.servicedecomposition.bbobjects.VolumeGroup;
40 import org.onap.so.bpmn.servicedecomposition.generalobjects.RequestContext;
41 import org.springframework.beans.factory.annotation.Autowired;
42
43
44 public class SDNCAssignTasksTest extends BaseTaskTest{
45         @Autowired
46         private SDNCAssignTasks sdncAssignTasks;
47
48         private L3Network network;
49         private ServiceInstance serviceInstance;
50         private RequestContext requestContext;
51         private CloudRegion cloudRegion;
52         private GenericVnf genericVnf;
53         private VfModule vfModule;
54         private VolumeGroup volumeGroup;
55         private Customer customer;
56
57         @Before
58         public void before() {
59                 customer = setCustomer();
60                 serviceInstance = setServiceInstance();
61                 network = setL3Network();
62                 cloudRegion = setCloudRegion();
63                 requestContext = setRequestContext();
64                 genericVnf = setGenericVnf();
65                 vfModule = setVfModule();
66                 volumeGroup = setVolumeGroup();
67
68         }
69
70         @Test
71         public void assignServiceInstanceTest() throws Exception {
72                 doReturn("response").when(sdncServiceInstanceResources).assignServiceInstance(serviceInstance, customer, requestContext);
73
74                 sdncAssignTasks.assignServiceInstance(execution);
75
76                 verify(sdncServiceInstanceResources, times(1)).assignServiceInstance(serviceInstance, customer, requestContext);
77                 assertTrue(execution.getVariable("SDNCResponse").equals("response"));
78         }
79
80         @Test
81         public void assignServiceInstanceExceptionTest() throws Exception {
82                 expectedException.expect(BpmnError.class);
83
84                 doThrow(Exception.class).when(sdncServiceInstanceResources).assignServiceInstance(serviceInstance, customer, requestContext);
85
86                 sdncAssignTasks.assignServiceInstance(execution);
87         }
88
89         @Test
90         public void assignVnfTest() throws Exception {
91                 doReturn("response").when(sdncVnfResources).assignVnf(genericVnf, serviceInstance, customer, cloudRegion, requestContext, false);
92
93                 execution.setVariable("generalBuildingBlock", gBBInput);
94                 sdncAssignTasks.assignVnf(execution);
95
96                 verify(sdncVnfResources, times(1)).assignVnf(genericVnf, serviceInstance,customer, cloudRegion, requestContext, false);
97                 assertTrue(execution.getVariable("SDNCResponse").equals("response"));
98         }
99
100         @Test
101         public void assignVnfExceptionTest() throws Exception {
102                 expectedException.expect(BpmnError.class);
103
104                 doThrow(Exception.class).when(sdncVnfResources).assignVnf(genericVnf, serviceInstance, customer, cloudRegion, requestContext, false);
105
106                 sdncAssignTasks.assignVnf(execution);
107         }
108
109         @Test
110         public void assignVfModuleTest() throws Exception {
111                 doReturn("response").when(sdncVfModuleResources).assignVfModule(vfModule, volumeGroup, genericVnf, serviceInstance, customer, cloudRegion, requestContext);
112
113                 sdncAssignTasks.assignVfModule(execution);
114
115                 verify(sdncVfModuleResources, times(1)).assignVfModule(vfModule, volumeGroup, genericVnf, serviceInstance, customer, cloudRegion, requestContext);
116                 assertTrue(execution.getVariable("SDNCAssignResponse_" + vfModule.getVfModuleId()).equals("response"));
117         }
118
119         @Test
120         public void assignVfModuleExceptionTest() throws Exception {
121                 expectedException.expect(BpmnError.class);
122
123                 doThrow(Exception.class).when(sdncVfModuleResources).assignVfModule(vfModule, volumeGroup, genericVnf, serviceInstance, customer, cloudRegion, requestContext);
124
125                 sdncAssignTasks.assignVfModule(execution);
126         }
127
128         @Test
129         public void assignNetworkTest() throws Exception {
130                 doReturn("response").when(sdncNetworkResources).assignNetwork(network, serviceInstance, customer, requestContext, cloudRegion);
131
132                 sdncAssignTasks.assignNetwork(execution);
133
134                 verify(sdncNetworkResources, times(1)).assignNetwork(network, serviceInstance, customer, requestContext, cloudRegion);
135         }
136
137         @Test
138         public void assignNetworkExceptionTest() throws Exception {
139                 expectedException.expect(BpmnError.class);
140
141                 doThrow(Exception.class).when(sdncNetworkResources).assignNetwork(network, serviceInstance, customer, requestContext, cloudRegion);
142
143                 sdncAssignTasks.assignNetwork(execution);
144         }
145 }