1 files changed, 224 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/auditing/impl/distribution/AuditDistributionEventFuncTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/auditing/impl/distribution/AuditDistributionEventFuncTest.java
new file mode 100644
index 0000000000..4c9c63343f
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/auditing/impl/distribution/AuditDistributionEventFuncTest.java
@@ -0,0 +1,224 @@
+package org.openecomp.sdc.be.auditing.impl.distribution;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Captor;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.openecomp.sdc.be.auditing.api.AuditEventFactory;
+import org.openecomp.sdc.be.auditing.impl.AuditGetUebClusterEventFactory;
+import org.openecomp.sdc.be.auditing.impl.AuditingManager;
+import org.openecomp.sdc.be.config.Configuration;
+import org.openecomp.sdc.be.dao.api.ActionStatus;
+import org.openecomp.sdc.be.dao.cassandra.AuditCassandraDao;
+import org.openecomp.sdc.be.dao.cassandra.CassandraOperationStatus;
+import org.openecomp.sdc.be.dao.impl.AuditingDao;
+import org.openecomp.sdc.be.resources.data.auditing.*;
+import org.openecomp.sdc.be.resources.data.auditing.model.*;
+import org.openecomp.sdc.common.util.ThreadLocalsHolder;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.*;
+
+@RunWith(MockitoJUnitRunner.class)
+public class AuditDistributionEventFuncTest {
+
+    private AuditingManager auditingManager;
+
+    @Mock
+    private static AuditCassandraDao cassandraDao;
+    @Mock
+    private static AuditingDao auditingDao;
+    @Mock
+    private static Configuration.ElasticSearchConfig esConfig;
+
+    @Captor
+    private ArgumentCaptor<AuditingGenericEvent> eventCaptor;
+
+    @Before
+    public void setUp() {
+        init(esConfig);
+        auditingManager = new AuditingManager(auditingDao, cassandraDao);
+        ThreadLocalsHolder.setUuid(REQUEST_ID);
+    }
+
+    @Test
+    public void testNotifyEvent() {
+        AuditEventFactory factory = new AuditDistributionNotificationEventFactory(
+                CommonAuditData.newBuilder()
+                    .description(DESCRIPTION)
+                    .status(STATUS_OK)
+                    .requestId(REQUEST_ID)
+                    .serviceInstanceId(SERVICE_INSTANCE_ID)
+                    .build(),
+                new ResourceCommonInfo(RESOURCE_NAME,RESOURCE_TYPE),
+                ResourceVersionInfo.newBuilder()
+                    .state(CURRENT_STATE)
+                    .version(CURRENT_VERSION)
+                    .build(),
+                DIST_ID, user, TOPIC_NAME,
+                new OperationalEnvAuditData(OP_ENV_ID, VNF_WORKLOAD_CONTEXT, TENANT_CONTEXT));
+
+        when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.DISTRIBUTION_NOTIFY.getAuditingEsType())))
+                .thenReturn(ActionStatus.OK);
+        when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
+
+        assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DISTRIB_NOTIFICATION_LOG_STR);
+        verifyNotifyEvent();
+    }
+
+    @Test
+    public void testStatusEvent() {
+        AuditEventFactory factory = new AuditDistributionStatusEventFactory(
+                CommonAuditData.newBuilder()
+                        .description(DESCRIPTION)
+                        .status(STATUS_OK)
+                        .requestId(REQUEST_ID)
+                        .serviceInstanceId(SERVICE_INSTANCE_ID)
+                        .build(),
+                new DistributionData(DIST_CONSUMER_ID, DIST_RESOURCE_URL),
+                DIST_ID, TOPIC_NAME, DIST_STATUS_TIME);
+
+        when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.DISTRIBUTION_STATUS.getAuditingEsType())))
+                .thenReturn(ActionStatus.OK);
+        when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
+
+        assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DIST_STATUS_LOG_STR);
+        verifyStatusEvent();
+    }
+
+    @Test
+    public void testDownloadEvent() {
+        AuditEventFactory factory = new AuditDistributionDownloadEventFactory(
+                CommonAuditData.newBuilder()
+                        .description(DESCRIPTION)
+                        .status(STATUS_OK)
+                        .requestId(REQUEST_ID)
+                        .serviceInstanceId(SERVICE_INSTANCE_ID)
+                        .build(),
+                new DistributionData(DIST_CONSUMER_ID, DIST_RESOURCE_URL));
+
+        when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.DISTRIBUTION_ARTIFACT_DOWNLOAD.getAuditingEsType())))
+                .thenReturn(ActionStatus.OK);
+        when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
+
+        assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DIST_DOWNLOAD_LOG_STR);
+        verifyDownloadsEvent();
+    }
+
+    @Test
+    public void testDeployEvent() {
+        AuditEventFactory factory = new AuditDistributionDeployEventFactory(
+                CommonAuditData.newBuilder()
+                        .description(DESCRIPTION)
+                        .status(STATUS_OK)
+                        .requestId(REQUEST_ID)
+                        .serviceInstanceId(SERVICE_INSTANCE_ID)
+                        .build(),
+                new ResourceCommonInfo(RESOURCE_NAME,RESOURCE_TYPE),
+                DIST_ID, user, CURRENT_VERSION);
+
+        when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.DISTRIBUTION_DEPLOY.getAuditingEsType())))
+                .thenReturn(ActionStatus.OK);
+        when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
+
+        assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DISTRIB_DEPLOY_LOG_STR);
+        verifyDeployEvent();
+    }
+
+    @Test
+    public void testGetUebClusterEvent() {
+        AuditEventFactory factory = new AuditGetUebClusterEventFactory(
+                CommonAuditData.newBuilder()
+                        .description(DESCRIPTION)
+                        .status(STATUS_OK)
+                        .requestId(REQUEST_ID)
+                        .serviceInstanceId(SERVICE_INSTANCE_ID)
+                        .build(),
+                DIST_CONSUMER_ID);
+
+        when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.GET_UEB_CLUSTER.getAuditingEsType())))
+                .thenReturn(ActionStatus.OK);
+        when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
+
+        assertThat(auditingManager.auditEvent(factory)).contains(EXPECTED_GET_UEB_CLUSTER_LOG_STR);
+        verifyGetUebClusterEvent();
+    }
+
+    private void verifyNotifyEvent() {
+        verify(cassandraDao).saveRecord(eventCaptor.capture());
+        DistributionNotificationEvent storedEvent = (DistributionNotificationEvent) eventCaptor.getValue();
+        assertThat(storedEvent.getCurrState()).isEqualTo(CURRENT_STATE);
+        assertThat(storedEvent.getCurrVersion()).isEqualTo(CURRENT_VERSION);
+        assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
+        assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
+        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
+        assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
+        assertThat(storedEvent.getAction()).isEqualTo(AuditingActionEnum.DISTRIBUTION_NOTIFY.getName());
+        assertThat(storedEvent.getDid()).isEqualTo(DIST_ID);
+        assertThat(storedEvent.getModifier()).isEqualTo(USER_UID);
+        assertThat(storedEvent.getResourceName()).isEqualTo(RESOURCE_NAME);
+        assertThat(storedEvent.getResourceType()).isEqualTo(RESOURCE_TYPE);
+        assertThat(storedEvent.getTopicName()).isEqualTo(TOPIC_NAME);
+        assertThat(storedEvent.getVnfWorkloadContext()).isEqualTo(VNF_WORKLOAD_CONTEXT);
+        assertThat(storedEvent.getEnvId()).isEqualTo(OP_ENV_ID);
+        assertThat(storedEvent.getTenant()).isEqualTo(TENANT_CONTEXT);
+    }
+
+    private void verifyStatusEvent() {
+        verify(cassandraDao).saveRecord(eventCaptor.capture());
+        DistributionStatusEvent storedEvent = (DistributionStatusEvent) eventCaptor.getValue();
+        assertThat(storedEvent.getConsumerId()).isEqualTo(DIST_CONSUMER_ID);
+        assertThat(storedEvent.getDid()).isEqualTo(DIST_ID);
+        assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
+        assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
+        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
+        assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
+        assertThat(storedEvent.getAction()).isEqualTo(AuditingActionEnum.DISTRIBUTION_STATUS.getName());
+        assertThat(storedEvent.getStatusTime()).isEqualTo(DIST_STATUS_TIME);
+        assertThat(storedEvent.getResoureURL()).isEqualTo(DIST_RESOURCE_URL);
+        assertThat(storedEvent.getTopicName()).isEqualTo(TOPIC_NAME);
+    }
+
+    private void verifyDownloadsEvent() {
+        verify(cassandraDao).saveRecord(eventCaptor.capture());
+        DistributionDownloadEvent storedEvent = (DistributionDownloadEvent) eventCaptor.getValue();
+        assertThat(storedEvent.getConsumerId()).isEqualTo(DIST_CONSUMER_ID);
+        assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
+        assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
+        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
+        assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
+        assertThat(storedEvent.getResourceUrl()).isEqualTo(DIST_RESOURCE_URL);
+    }
+
+    private void verifyDeployEvent() {
+        verify(cassandraDao).saveRecord(eventCaptor.capture());
+        DistributionDeployEvent storedEvent = (DistributionDeployEvent) eventCaptor.getValue();
+        assertThat(storedEvent.getCurrVersion()).isEqualTo(CURRENT_VERSION);
+        assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
+        assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
+        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
+        assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
+        assertThat(storedEvent.getDid()).isEqualTo(DIST_ID);
+        assertThat(storedEvent.getModifier()).isEqualTo(USER_UID);
+        assertThat(storedEvent.getResourceName()).isEqualTo(RESOURCE_NAME);
+        assertThat(storedEvent.getResourceType()).isEqualTo(RESOURCE_TYPE);
+    }
+
+    private void verifyGetUebClusterEvent() {
+        verify(cassandraDao).saveRecord(eventCaptor.capture());
+        AuditingGetUebClusterEvent storedEvent = (AuditingGetUebClusterEvent) eventCaptor.getValue();
+        assertThat(storedEvent.getConsumerId()).isEqualTo(DIST_CONSUMER_ID);
+        assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
+        assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
+        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
+        assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
+     }
+
+}