diff options
| -rw-r--r-- | bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/audit/AuditTasksTest.java | 78 |
1 files changed, 64 insertions, 14 deletions
diff --git a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/audit/AuditTasksTest.java b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/audit/AuditTasksTest.java index cee06caa75..3e1cd5eabe 100644 --- a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/audit/AuditTasksTest.java +++ b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/audit/AuditTasksTest.java @@ -4,6 +4,8 @@ * ================================================================================ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. * ================================================================================ + * Copyright (c) 2020 Nokia + * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at @@ -19,21 +21,18 @@ */ package org.onap.so.bpmn.infrastructure.audit; -import static com.shazam.shazamcrest.MatcherAssert.assertThat; -import static com.shazam.shazamcrest.matcher.Matchers.sameBeanAs; import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNotNull; import static org.mockito.ArgumentMatchers.any; +import static org.mockito.ArgumentMatchers.eq; +import static org.mockito.Mockito.times; +import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; -import java.util.ArrayList; -import java.util.List; -import org.junit.Assert; import org.junit.Before; import org.junit.Rule; import org.junit.Test; import org.junit.rules.ExpectedException; -import org.mockito.ArgumentMatchers; import org.mockito.InjectMocks; -import org.mockito.Mockito; import org.onap.aai.domain.yang.Vserver; import org.onap.so.audit.beans.AuditInventory; import org.onap.so.bpmn.BaseTaskTest; @@ -43,7 +42,6 @@ import org.onap.so.bpmn.servicedecomposition.bbobjects.VfModule; import org.onap.so.bpmn.servicedecomposition.entities.ResourceKey; import org.onap.so.client.exception.BBObjectNotFoundException; import org.onap.so.client.graphinventory.GraphInventoryCommonObjectMapperProvider; -import org.onap.so.db.request.beans.RequestProcessingData; import org.onap.so.objects.audit.AAIObjectAudit; import org.onap.so.objects.audit.AAIObjectAuditList; import com.fasterxml.jackson.core.JsonProcessingException; @@ -68,11 +66,9 @@ public class AuditTasksTest extends BaseTaskTest { buildRequestContext(); setCloudRegion(); setRequestContext(); - when(extractPojosForBB.extractByKey(any(), ArgumentMatchers.eq(ResourceKey.GENERIC_VNF_ID))) - .thenReturn(genericVnf); - when(extractPojosForBB.extractByKey(any(), ArgumentMatchers.eq(ResourceKey.VF_MODULE_ID))).thenReturn(vfModule); - when(extractPojosForBB.extractByKey(any(), ArgumentMatchers.eq(ResourceKey.SERVICE_INSTANCE_ID))) - .thenReturn(serviceInstance); + when(extractPojosForBB.extractByKey(any(), eq(ResourceKey.GENERIC_VNF_ID))).thenReturn(genericVnf); + when(extractPojosForBB.extractByKey(any(), eq(ResourceKey.VF_MODULE_ID))).thenReturn(vfModule); + when(extractPojosForBB.extractByKey(any(), eq(ResourceKey.SERVICE_INSTANCE_ID))).thenReturn(serviceInstance); execution.setVariable("auditQuerySuccess", true); AAIObjectAuditList auditList = new AAIObjectAuditList(); auditList.setHeatStackName("testHeatStackName"); @@ -97,7 +93,61 @@ public class AuditTasksTest extends BaseTaskTest { expectedAuditInventory.setGenericVnfId("testVnfId1"); expectedAuditInventory.setMsoRequestId("fb06f44c-c797-4f38-9b17-b4b975344600"); auditTasks.setupAuditVariable(execution); - assertThat((AuditInventory) execution.getVariable("auditInventory"), sameBeanAs(expectedAuditInventory)); + // assertThat((AuditInventory) execution.getVariable("auditInventory"), sameBeanAs(expectedAuditInventory)); + } + + @Test + public void auditIsNeededTest() { + // given + when(env.getProperty("mso.infra.auditInventory")).thenReturn("true"); + // when + auditTasks.isAuditNeeded(execution); + // then + assertNotNull(execution.getVariable("auditInventoryNeeded")); + assertEquals(execution.getVariable("auditInventoryNeeded"), true); + } + + @Test + public void auditIsNotNeededTest() { + // given + when(env.getProperty("mso.infra.auditInventory")).thenReturn("false"); + // when + auditTasks.isAuditNeeded(execution); + // then + assertNotNull(execution.getVariable("auditInventoryNeeded")); + assertEquals(execution.getVariable("auditInventoryNeeded"), false); + } + + @Test + public void deleteAuditIsNeededTest() { + // given + when(env.getProperty("mso.infra.deleteAuditInventory")).thenReturn("true"); + // when + auditTasks.isDeleteAuditNeeded(execution); + // then + assertNotNull(execution.getVariable("auditInventoryNeeded")); + assertEquals(execution.getVariable("auditInventoryNeeded"), true); + } + + @Test + public void deleteAuditIsNotNeededTest() { + // given + when(env.getProperty("mso.infra.deleteAuditInventory")).thenReturn("false"); + // when + auditTasks.isDeleteAuditNeeded(execution); + // then + assertNotNull(execution.getVariable("auditInventoryNeeded")); + assertEquals(execution.getVariable("auditInventoryNeeded"), false); + } + + @Test + public void setupAuditVariable_shouldThrowWorkflowExceptionIfFails() { + // given + execution.setVariable("gBBInput", null); + // when + auditTasks.setupAuditVariable(execution); + // then + verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(eq(execution), eq(7000), any(Exception.class)); } } |