1 files changed, 104 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/ArchiveBusinessLogicTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/ArchiveBusinessLogicTest.java
new file mode 100644
index 0000000000..8029e788d9
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/ArchiveBusinessLogicTest.java
@@ -0,0 +1,104 @@
+package org.openecomp.sdc.be.components.impl;
+
+import com.google.common.collect.ImmutableList;
+import fj.data.Either;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.openecomp.sdc.be.dao.api.ActionStatus;
+import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
+import org.openecomp.sdc.be.impl.ComponentsUtils;
+import org.openecomp.sdc.be.model.Component;
+import org.openecomp.sdc.be.model.ComponentParametersView;
+import org.openecomp.sdc.be.model.User;
+import org.openecomp.sdc.be.model.jsontitan.operations.ArchiveOperation;
+import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
+import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
+import org.openecomp.sdc.be.resources.data.auditing.AuditingActionEnum;
+import org.openecomp.sdc.exception.ResponseFormat;
+
+import java.util.Arrays;
+import java.util.List;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.*;
+
+@RunWith(MockitoJUnitRunner.class)
+public class ArchiveBusinessLogicTest {
+    @Mock
+    private ComponentsUtils componentsUtils;
+    @Mock
+    private ToscaOperationFacade toscaOperationFacade;
+    @Mock
+    private User user;
+    @Mock
+    private ResponseFormat responseFormat;
+    @Mock
+    private Component component;
+
+    @InjectMocks
+    private ArchiveBusinessLogic archiveBusinessLogic;
+
+    @Test
+    public void auditLastComponentVersionOnlyAndIgnorePreviousVersions() {
+        List<String> archivedCompIds = Arrays.asList("1", "2", "3", "4", "5");
+        when(toscaOperationFacade.getToscaElement(any(String.class), any(ComponentParametersView.class)))
+                .thenReturn(Either.left(component));
+        when(componentsUtils.getResponseFormat(ActionStatus.OK)).thenReturn(responseFormat);
+        when(component.getUUID())
+                .thenReturn("1")
+                .thenReturn("1")
+                .thenReturn("2")
+                .thenReturn("3")
+                .thenReturn("2");
+
+        archiveBusinessLogic.auditAction(ArchiveOperation.Action.ARCHIVE, archivedCompIds, user, ComponentTypeEnum.RESOURCE_PARAM_NAME);
+        verify(componentsUtils, times(3)).auditComponentAdmin(eq(responseFormat), eq(user), eq(component),
+                eq(AuditingActionEnum.ARCHIVE_COMPONENT), eq(ComponentTypeEnum.RESOURCE), any(String.class));
+    }
+
+
+    @Test
+    public void auditLastComponentVersionOnly() {
+        List<String> archivedCompIds = Arrays.asList("1", "2", "3", "4", "5");
+        when(toscaOperationFacade.getToscaElement(any(String.class), any(ComponentParametersView.class)))
+                .thenReturn(Either.left(component));
+        when(componentsUtils.getResponseFormat(ActionStatus.OK)).thenReturn(responseFormat);
+        when(component.getUUID())
+                .thenReturn("1")
+                .thenReturn("2")
+                .thenReturn("4")
+                .thenReturn("5")
+                .thenReturn("3");
+
+        archiveBusinessLogic.auditAction(ArchiveOperation.Action.RESTORE, archivedCompIds, user, ComponentTypeEnum.RESOURCE_PARAM_NAME);
+        verify(componentsUtils, times(5)).auditComponentAdmin(eq(responseFormat), eq(user), eq(component),
+                eq(AuditingActionEnum.RESTORE_COMPONENT), eq(ComponentTypeEnum.RESOURCE), any(String.class));
+    }
+
+
+    @Test
+    public void noAuditDoneForEmptyList() {
+        List<String> archivedCompIds = ImmutableList.of();
+        archiveBusinessLogic.auditAction(ArchiveOperation.Action.RESTORE, archivedCompIds, user, ComponentTypeEnum.RESOURCE_PARAM_NAME);
+        verify(componentsUtils, times(0)).auditComponentAdmin(any(ResponseFormat.class), any(User.class), any(Component.class),
+                any(AuditingActionEnum.class), any(ComponentTypeEnum.class), any(String.class));
+    }
+
+
+    @Test
+    public void noAuditOnErrorGetElementResponse() {
+        List<String> archivedCompIds = Arrays.asList("1", "2", "3", "4", "5");
+        when(toscaOperationFacade.getToscaElement(any(String.class), any(ComponentParametersView.class)))
+                .thenReturn(Either.right(StorageOperationStatus.NOT_FOUND));
+        archiveBusinessLogic.auditAction(ArchiveOperation.Action.RESTORE, archivedCompIds, user, ComponentTypeEnum.RESOURCE_PARAM_NAME);
+        verify(componentsUtils, times(0)).auditComponentAdmin(any(ResponseFormat.class), any(User.class), any(Component.class),
+                any(AuditingActionEnum.class), any(ComponentTypeEnum.class), any(String.class));
+    }
+
+
+
+}