catalog-be/src/test/java/org/openecomp/sdc/be/auditing/impl/distribution/AuditDistributionEngineFuncTest.java

   1 package org.openecomp.sdc.be.auditing.impl.distribution;
   2
   3 import org.junit.Before;
   4 import org.junit.Test;
   5 import org.junit.runner.RunWith;
   6 import org.mockito.ArgumentCaptor;
   7 import org.mockito.Captor;
   8 import org.mockito.Mock;
   9 import org.mockito.junit.MockitoJUnitRunner;
  10 import org.openecomp.sdc.be.auditing.api.AuditEventFactory;
  11 import org.openecomp.sdc.be.auditing.impl.AuditingManager;
  12 import org.openecomp.sdc.be.config.Configuration;
  13 import org.openecomp.sdc.be.dao.api.ActionStatus;
  14 import org.openecomp.sdc.be.dao.cassandra.AuditCassandraDao;
  15 import org.openecomp.sdc.be.dao.cassandra.CassandraOperationStatus;
  16 import org.openecomp.sdc.be.dao.impl.AuditingDao;
  17 import org.openecomp.sdc.be.resources.data.auditing.AuditingActionEnum;
  18 import org.openecomp.sdc.be.resources.data.auditing.AuditingGenericEvent;
  19 import org.openecomp.sdc.be.resources.data.auditing.DistributionEngineEvent;
  20 import org.openecomp.sdc.be.resources.data.auditing.model.CommonAuditData;
  21 import org.openecomp.sdc.be.resources.data.auditing.model.DistributionTopicData;
  22 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
  23
  24 import static org.assertj.core.api.Assertions.assertThat;
  25 import static org.mockito.ArgumentMatchers.any;
  26 import static org.mockito.ArgumentMatchers.eq;
  27 import static org.mockito.Mockito.verify;
  28 import static org.mockito.Mockito.when;
  29 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.*;
  30
  31 @RunWith(MockitoJUnitRunner.class)
  32 public class AuditDistributionEngineFuncTest {
  33     private AuditingManager auditingManager;
  34
  35     @Mock
  36     private static AuditCassandraDao cassandraDao;
  37     @Mock
  38     private static AuditingDao auditingDao;
  39     @Mock
  40     private static Configuration.ElasticSearchConfig esConfig;
  41
  42     @Captor
  43     private ArgumentCaptor<DistributionEngineEvent> eventCaptor;
  44
  45     @Before
  46     public void setUp() {
  47         init(esConfig);
  48         auditingManager = new AuditingManager(auditingDao, cassandraDao);
  49         ThreadLocalsHolder.setUuid(REQUEST_ID);
  50     }
  51
  52     @Test
  53     public void testAddKeyEvent() {
  54         AuditEventFactory factory = new AuditAddRemoveKeyDistributionEngineEventFactory(
  55                 AuditingActionEnum.ADD_KEY_TO_TOPIC_ACL,
  56                 CommonAuditData.newBuilder()
  57                         .description(DESCRIPTION)
  58                         .status(STATUS_OK)
  59                         .requestId(REQUEST_ID)
  60                         .serviceInstanceId(SERVICE_INSTANCE_ID)
  61                         .build(),
  62                 DistributionTopicData.newBuilder()
  63                         .statusTopic(DIST_STATUS_TOPIC)
  64                         .notificationTopic(DIST_NOTIFY_TOPIC)
  65                         .build(),
  66                 DIST_API_KEY, DIST_ENV_NAME, DIST_ROLE);
  67
  68         when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.ADD_KEY_TO_TOPIC_ACL.getAuditingEsType())))
  69                 .thenReturn(ActionStatus.OK);
  70         when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
  71
  72         assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DIST_ADD_KEY_ENGINE_LOG_STR);
  73         verifyEvent(AuditingActionEnum.ADD_KEY_TO_TOPIC_ACL.getName());
  74     }
  75
  76     @Test
  77     public void testCreateTopicEvent() {
  78         AuditEventFactory factory = new AuditCreateTopicDistributionEngineEventFactory(
  79                 CommonAuditData.newBuilder()
  80                         .description(DESCRIPTION)
  81                         .status(STATUS_OK)
  82                         .requestId(REQUEST_ID)
  83                         .serviceInstanceId(SERVICE_INSTANCE_ID)
  84                         .build(),
  85                 DistributionTopicData.newBuilder()
  86                         .statusTopic(DIST_STATUS_TOPIC)
  87                         .notificationTopic(DIST_NOTIFY_TOPIC)
  88                         .build(),
  89                 DIST_API_KEY, DIST_ENV_NAME, DIST_ROLE);
  90
  91         when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.CREATE_DISTRIBUTION_TOPIC.getAuditingEsType())))
  92                 .thenReturn(ActionStatus.OK);
  93         when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
  94
  95         assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DIST_CREATE_TOPIC_ENGINE_LOG_STR);
  96         verifyEvent(AuditingActionEnum.CREATE_DISTRIBUTION_TOPIC.getName());
  97     }
  98
  99     @Test
 100     public void testRegisterEvent() {
 101         AuditEventFactory factory = new AuditRegUnregDistributionEngineEventFactory(
 102                 AuditingActionEnum.DISTRIBUTION_REGISTER,
 103                 CommonAuditData.newBuilder()
 104                         .description(DESCRIPTION)
 105                         .status(STATUS_OK)
 106                         .requestId(REQUEST_ID)
 107                         .serviceInstanceId(SERVICE_INSTANCE_ID)
 108                         .build(),
 109                 DistributionTopicData.newBuilder()
 110                         .statusTopic(DIST_STATUS_TOPIC)
 111                         .notificationTopic(DIST_NOTIFY_TOPIC)
 112                         .build(),
 113                 DIST_CONSUMER_ID, DIST_API_KEY, DIST_ENV_NAME);
 114
 115         when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.DISTRIBUTION_REGISTER.getAuditingEsType())))
 116                 .thenReturn(ActionStatus.OK);
 117         when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
 118
 119         assertThat(auditingManager.auditEvent(factory)).isEqualTo(EXPECTED_DIST_REG_ENGINE_LOG_STR);
 120         verifyEvent(AuditingActionEnum.DISTRIBUTION_REGISTER.getName());
 121     }
 122
 123     private void verifyEvent(String action) {
 124         verify(cassandraDao).saveRecord(eventCaptor.capture());
 125         DistributionEngineEvent storedEvent = eventCaptor.getValue();
 126         assertThat(storedEvent.getDnotifTopic()).isEqualTo(DIST_NOTIFY_TOPIC);
 127         assertThat(storedEvent.getDstatusTopic()).isEqualTo(DIST_STATUS_TOPIC);
 128         assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
 129         assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
 130         assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
 131         assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
 132         assertThat(storedEvent.getAction()).isEqualTo(action);
 133         assertThat(storedEvent.getEnvironmentName()).isEqualTo(DIST_ENV_NAME);
 134         assertThat(storedEvent.getApiKey()).isEqualTo(DIST_API_KEY);
 135         if (!action.equals(AuditingActionEnum.CREATE_DISTRIBUTION_TOPIC.getName()) &&
 136             !action.equals(AuditingActionEnum.ADD_KEY_TO_TOPIC_ACL.getName()) &&
 137             !action.equals(AuditingActionEnum.REMOVE_KEY_FROM_TOPIC_ACL.getName())) {
 138             assertThat(storedEvent.getConsumerId()).isEqualTo(DIST_CONSUMER_ID);
 139         }
 140         if (!action.equals(AuditingActionEnum.DISTRIBUTION_REGISTER.getName()) &&
 141                 !action.equals(AuditingActionEnum.DISTRIBUTION_UN_REGISTER.getName())) {
 142             assertThat(storedEvent.getRole()).isEqualTo(DIST_ROLE);
 143         }
 144     }
 145
 146 }