gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.be.auditing.impl.category;
   2
   3 import static org.assertj.core.api.Assertions.assertThat;
   4 import static org.mockito.ArgumentMatchers.any;
   5 import static org.mockito.ArgumentMatchers.anyMap;
   6 import static org.mockito.ArgumentMatchers.eq;
   7 import static org.mockito.Mockito.verify;
   8 import static org.mockito.Mockito.when;
   9 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.CATEGORY;
  10 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.DESCRIPTION;
  11 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.EXPECTED_ADD_CATEGORY_LOG_STR;
  12 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.GROUPING_NAME;
  13 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.MODIFIER_UID;
  14 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.REQUEST_ID;
  15 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.RESOURCE_TYPE;
  16 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.SERVICE_INSTANCE_ID;
  17 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.STATUS_OK;
  18 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.SUB_CATEGORY;
  19 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.init;
  20 import static org.openecomp.sdc.be.auditing.impl.AuditTestUtils.modifier;
  21
  22 import java.util.EnumMap;
  23
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.junit.runner.RunWith;
  27 import org.mockito.ArgumentCaptor;
  28 import org.mockito.Captor;
  29 import org.mockito.Mock;
  30 import org.mockito.junit.MockitoJUnitRunner;
  31 import org.openecomp.sdc.be.auditing.api.AuditEventFactory;
  32 import org.openecomp.sdc.be.auditing.impl.AuditingManager;
  33 import org.openecomp.sdc.be.config.Configuration;
  34 import org.openecomp.sdc.be.dao.api.ActionStatus;
  35 import org.openecomp.sdc.be.dao.cassandra.AuditCassandraDao;
  36 import org.openecomp.sdc.be.dao.cassandra.CassandraOperationStatus;
  37 import org.openecomp.sdc.be.dao.impl.AuditingDao;
  38 import org.openecomp.sdc.be.resources.data.auditing.AuditingActionEnum;
  39 import org.openecomp.sdc.be.resources.data.auditing.AuditingGenericEvent;
  40 import org.openecomp.sdc.be.resources.data.auditing.CategoryEvent;
  41 import org.openecomp.sdc.be.resources.data.auditing.model.CommonAuditData;
  42 import org.openecomp.sdc.common.datastructure.AuditingFieldsKeysEnum;
  43 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
  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 }