2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017-2019 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.so.adapters.audit;
23 import static org.hamcrest.CoreMatchers.isA;
24 import static org.junit.Assert.assertEquals;
25 import static org.junit.Assert.assertNotNull;
26 import static org.mockito.Mockito.doReturn;
29 import java.io.IOException;
30 import java.util.Arrays;
31 import java.util.Collection;
33 import java.util.Optional;
35 import org.camunda.bpm.client.task.ExternalTask;
36 import org.camunda.bpm.client.task.ExternalTaskService;
37 import org.hamcrest.Matchers;
38 import org.junit.Before;
39 import org.junit.Test;
40 import org.mockito.ArgumentCaptor;
41 import org.mockito.InjectMocks;
42 import org.mockito.Mock;
43 import org.mockito.Mockito;
44 import org.mockito.MockitoAnnotations;
45 import org.onap.aai.domain.yang.LInterface;
46 import org.onap.so.audit.beans.AuditInventory;
47 import org.springframework.core.env.Environment;
48 import com.fasterxml.jackson.core.JsonParseException;
49 import com.fasterxml.jackson.databind.JsonMappingException;
50 import com.fasterxml.jackson.databind.ObjectMapper;
52 public class AuditStackServiceDataTest extends AuditStackServiceData {
55 private AuditStackServiceData auditStackService = new AuditStackServiceData();
58 private HeatStackAudit heatStackAuditMock;
61 private Environment mockEnv;
64 private ExternalTask mockExternalTask;
67 private ExternalTaskService mockExternalTaskService;
69 private ObjectMapper objectMapper = new ObjectMapper();
71 private AuditInventory auditInventory = new AuditInventory();
73 Optional<AAIObjectAuditList> auditListOptSuccess;
75 Optional<AAIObjectAuditList> auditListOptFailure;
78 public void setup() throws JsonParseException, JsonMappingException, IOException {
79 auditInventory.setCloudOwner("cloudOwner");
80 auditInventory.setCloudRegion("cloudRegion");
81 auditInventory.setTenantId("tenantId");
82 auditInventory.setHeatStackName("stackName");
83 MockitoAnnotations.initMocks(this);
85 AAIObjectAuditList auditListSuccess = objectMapper.readValue(new File("src/test/resources/ExpectedVServerFound.json"), AAIObjectAuditList.class);
86 auditListOptSuccess = Optional.of(auditListSuccess);
88 AAIObjectAuditList auditListFailure = objectMapper.readValue(new File("src/test/resources/Vserver2_Found_VServer1_Not_Found.json"), AAIObjectAuditList.class);
89 auditListOptFailure = Optional.of(auditListFailure);
90 String[] retrySequence = new String[8];
91 retrySequence[0] = "1";
92 retrySequence[1] = "1";
93 retrySequence[2] = "2";
94 retrySequence[3] = "3";
95 retrySequence[4] = "5";
96 retrySequence[5] = "8";
97 retrySequence[6] = "13";
98 retrySequence[7] = "20";
99 doReturn(auditInventory).when(mockExternalTask).getVariable("auditInventory");
100 doReturn("6000").when(mockEnv).getProperty("mso.workflow.topics.retryMultiplier","6000");
101 doReturn(retrySequence).when(mockEnv).getProperty("mso.workflow.topics.retrySequence",String[].class);
102 doReturn("aasdfasdf").when(mockExternalTask).getId();
106 public void execute_external_task_audit_success_Test() {
107 doReturn(auditListOptSuccess).when(heatStackAuditMock).auditHeatStack("cloudRegion", "cloudOwner", "tenantId", "stackName");
108 auditStackService.executeExternalTask(mockExternalTask, mockExternalTaskService);
109 ArgumentCaptor<Map> captor = ArgumentCaptor.forClass(Map.class);
110 ArgumentCaptor<ExternalTask> taskCaptor = ArgumentCaptor.forClass(ExternalTask.class);
111 Mockito.verify(mockExternalTaskService).complete(taskCaptor.capture(),captor.capture());
112 Map actualMap = captor.getValue();
113 assertEquals(true,actualMap.get("auditIsSuccessful"));
114 assertNotNull(actualMap.get("auditInventoryResult"));
118 public void execute_external_task_audit_first_failure_Test() {
119 doReturn(auditListOptFailure).when(heatStackAuditMock).auditHeatStack("cloudRegion", "cloudOwner", "tenantId", "stackName");
120 doReturn(null).when(mockExternalTask).getRetries();
121 auditStackService.executeExternalTask(mockExternalTask, mockExternalTaskService);
122 Mockito.verify(mockExternalTaskService).handleFailure(mockExternalTask,
123 "Unable to find all VServers and L-Interaces in A&AI",
124 "Unable to find all VServers and L-Interaces in A&AI", 8, 10000L);
128 public void execute_external_task_audit_intermediate_failure_Test() {
129 doReturn(auditListOptFailure).when(heatStackAuditMock).auditHeatStack("cloudRegion", "cloudOwner", "tenantId", "stackName");
130 doReturn(6).when(mockExternalTask).getRetries();
131 auditStackService.executeExternalTask(mockExternalTask, mockExternalTaskService);
132 Mockito.verify(mockExternalTaskService).handleFailure(mockExternalTask,
133 "Unable to find all VServers and L-Interaces in A&AI",
134 "Unable to find all VServers and L-Interaces in A&AI", 5, 12000L);
139 public void execute_external_task_audit_final_failure_Test() {
140 doReturn(auditListOptFailure).when(heatStackAuditMock).auditHeatStack("cloudRegion", "cloudOwner", "tenantId", "stackName");
141 doReturn(1).when(mockExternalTask).getRetries();
142 auditStackService.executeExternalTask(mockExternalTask, mockExternalTaskService);
143 ArgumentCaptor<Map> captor = ArgumentCaptor.forClass(Map.class);
144 ArgumentCaptor<ExternalTask> taskCaptor = ArgumentCaptor.forClass(ExternalTask.class);
145 Mockito.verify(mockExternalTaskService).complete(taskCaptor.capture(),captor.capture());
146 Map actualMap = captor.getValue();
147 assertEquals(false,actualMap.get("auditIsSuccessful"));
148 assertNotNull(actualMap.get("auditInventoryResult"));
152 public void retry_sequence_calculation_Test() {
153 long firstRetry = auditStackService.calculateRetryDelay(8);
154 assertEquals(6000L, firstRetry);
155 long secondRetry = auditStackService.calculateRetryDelay(7);
156 assertEquals(6000L, secondRetry);
157 long thirdRetry = auditStackService.calculateRetryDelay(6);
158 assertEquals(12000L, thirdRetry);
159 long fourthRetry = auditStackService.calculateRetryDelay(5);
160 assertEquals(18000L, fourthRetry);
161 long fifthRetry = auditStackService.calculateRetryDelay(4);
162 assertEquals(30000L, fifthRetry);
163 long sixRetry = auditStackService.calculateRetryDelay(3);
164 assertEquals(48000L, sixRetry);
165 long seventhRetry = auditStackService.calculateRetryDelay(2);
166 assertEquals(78000L, seventhRetry);
167 long eigthRetry = auditStackService.calculateRetryDelay(1);
168 assertEquals(120000L, eigthRetry);
172 public void retry_sequence_Test() {
173 long firstRetry = auditStackService.calculateRetryDelay(8);
174 assertEquals(6000L, firstRetry);
175 long secondRetry = auditStackService.calculateRetryDelay(7);
176 assertEquals(6000L, secondRetry);
177 long thirdRetry = auditStackService.calculateRetryDelay(6);
178 assertEquals(12000L, thirdRetry);
179 long fourthRetry = auditStackService.calculateRetryDelay(5);
180 assertEquals(18000L, fourthRetry);
181 long fifthRetry = auditStackService.calculateRetryDelay(4);
182 assertEquals(30000L, fifthRetry);
183 long sixRetry = auditStackService.calculateRetryDelay(3);
184 assertEquals(48000L, sixRetry);
185 long seventhRetry = auditStackService.calculateRetryDelay(2);
186 assertEquals(78000L, seventhRetry);
187 long eigthRetry = auditStackService.calculateRetryDelay(1);
188 assertEquals(120000L, eigthRetry);
193 public void determineAuditResult_Test() throws Exception{
194 boolean actual = auditStackService.didAuditFail(auditListOptSuccess);
195 assertEquals(false, actual);
199 public void determineAuditResult_Failure_Test() throws Exception{
200 boolean actual = auditStackService.didAuditFail(auditListOptFailure);
201 assertEquals(true, actual);