gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.be.auditing.impl.category;
   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.CategoryEvent;
  20 import org.openecomp.sdc.be.resources.data.auditing.model.CommonAuditData;
  21 import org.openecomp.sdc.common.datastructure.AuditingFieldsKeysEnum;
  22 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
  23
  24 import java.util.EnumMap;
  25
  26 import static org.assertj.core.api.Assertions.assertThat;
  27 import static org.mockito.ArgumentMatchers.any;
  28 import static org.mockito.ArgumentMatchers.anyMap;
  29 import static org.mockito.ArgumentMatchers.eq;
  30 import static org.mockito.Mockito.verify;
  31 import static org.mockito.Mockito.when;
  32 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.CATEGORY;
  33 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.DESCRIPTION;
  34 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.EXPECTED_ADD_CATEGORY_LOG_STR;
  35 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.GROUPING_NAME;
  36 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.MODIFIER_UID;
  37 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.REQUEST_ID;
  38 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.RESOURCE_TYPE;
  39 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.SERVICE_INSTANCE_ID;
  40 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.STATUS_OK;
  41 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.SUB_CATEGORY;
  42 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.init;
  43 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.modifier;
  44
  45 @RunWith(MockitoJUnitRunner.class)
  46 public class AuditCategoryEventFuncTest {
  47     private AuditingManager auditingManager;
  48
  49     @Mock
  50     private static AuditCassandraDao cassandraDao;
  51     @Mock
  52     private static AuditingDao auditingDao;
  53     @Mock
  54     private static Configuration.ElasticSearchConfig esConfig;
  55
  56     @Captor
  57     private ArgumentCaptor<AuditingGenericEvent> eventCaptor;
  58
  59     @Before
  60     public void setUp() {
  61         init(esConfig);
  62         auditingManager = new AuditingManager(auditingDao, cassandraDao);
  63         ThreadLocalsHolder.setUuid(REQUEST_ID);
  64     }
  65
  66     @Test
  67     public void testNewAddCategoryEvent() {
  68         AuditEventFactory builder = new AuditAddCategoryEventFactory(
  69                 CommonAuditData.newBuilder()
  70                         .description(DESCRIPTION)
  71                         .status(STATUS_OK)
  72                         .requestId(REQUEST_ID)
  73                         .serviceInstanceId(SERVICE_INSTANCE_ID)
  74                         .build(),
  75                 modifier, CATEGORY, SUB_CATEGORY, GROUPING_NAME, RESOURCE_TYPE);
  76
  77         when(auditingDao.addRecord(any(AuditingGenericEvent.class), eq(AuditingActionEnum.ADD_CATEGORY.getAuditingEsType())))
  78                 .thenReturn(ActionStatus.OK);
  79         when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
  80
  81         assertThat(auditingManager.auditEvent(builder)).isEqualTo(EXPECTED_ADD_CATEGORY_LOG_STR);
  82         verifyCategoryEvent(AuditingActionEnum.ADD_CATEGORY.getName());
  83     }
  84
  85     @Test
  86     public void testOldAddEcompUserCredEvent() {
  87         when(auditingDao.addRecord(anyMap(), eq(AuditingActionEnum.ADD_CATEGORY.getAuditingEsType())))
  88                 .thenReturn(ActionStatus.OK);
  89         when(cassandraDao.saveRecord(any(AuditingGenericEvent.class))).thenReturn(CassandraOperationStatus.OK);
  90
  91         assertThat(auditingManager.auditEvent(fillMap(AuditingActionEnum.ADD_CATEGORY))).isEqualTo(EXPECTED_ADD_CATEGORY_LOG_STR);
  92         verifyCategoryEvent(AuditingActionEnum.ADD_CATEGORY.getName());
  93
  94     }
  95
  96     private EnumMap<AuditingFieldsKeysEnum, Object> fillMap(AuditingActionEnum action) {
  97         EnumMap<AuditingFieldsKeysEnum, Object> auditingFields = new EnumMap<>(AuditingFieldsKeysEnum.class);
  98         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_ACTION, action.getName());
  99         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_MODIFIER_NAME, modifier.getFirstName() + " " + modifier.getLastName());
 100         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_MODIFIER_UID, modifier.getUserId());
 101         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_STATUS, STATUS_OK);
 102         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_DESC, DESCRIPTION);
 103         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_CATEGORY_NAME, CATEGORY);
 104         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_SUB_CATEGORY_NAME, SUB_CATEGORY);
 105         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_GROUPING_NAME, GROUPING_NAME);
 106         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_RESOURCE_TYPE, RESOURCE_TYPE);
 107         auditingFields.put(AuditingFieldsKeysEnum.AUDIT_SERVICE_INSTANCE_ID, SERVICE_INSTANCE_ID);
 108
 109         return auditingFields;
 110     }
 111
 112     private void verifyCategoryEvent(String action) {
 113         verify(cassandraDao).saveRecord(eventCaptor.capture());
 114         CategoryEvent storedEvent = (CategoryEvent) eventCaptor.getValue();
 115         assertThat(storedEvent.getModifier()).isEqualTo(MODIFIER_UID);
 116         assertThat(storedEvent.getDesc()).isEqualTo(DESCRIPTION);
 117         assertThat(storedEvent.getStatus()).isEqualTo(STATUS_OK);
 118 //        assertThat(storedEvent.getRequestId()).isEqualTo(REQUEST_ID);
 119         assertThat(storedEvent.getServiceInstanceId()).isEqualTo(SERVICE_INSTANCE_ID);
 120         assertThat(storedEvent.getAction()).isEqualTo(action);
 121         assertThat(storedEvent.getCategoryName()).isEqualTo(CATEGORY);
 122         assertThat(storedEvent.getSubCategoryName()).isEqualTo(SUB_CATEGORY);
 123         assertThat(storedEvent.getGroupingName()).isEqualTo(GROUPING_NAME);
 124         assertThat(storedEvent.getResourceType()).isEqualTo(RESOURCE_TYPE);
 125
 126     }
 127 }