IrynaMatveieva
2 years ago
4 changed files with 228 additions and 46 deletions
@ -0,0 +1,141 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.dao.audit; |
||||
|
|
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.junit.jupiter.api.extension.ExtendWith; |
||||
|
import org.mockito.ArgumentCaptor; |
||||
|
import org.mockito.InjectMocks; |
||||
|
import org.mockito.Mock; |
||||
|
import org.mockito.junit.jupiter.MockitoExtension; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmComment; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmInfo; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.audit.AuditLog; |
||||
|
import org.thingsboard.server.common.data.id.AlarmId; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.id.UserId; |
||||
|
import org.thingsboard.server.dao.audit.sink.AuditLogSink; |
||||
|
import org.thingsboard.server.dao.entity.EntityService; |
||||
|
import org.thingsboard.server.dao.service.validator.AuditLogDataValidator; |
||||
|
import org.thingsboard.server.dao.sql.JpaExecutorService; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.Callable; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.ArgumentMatchers.eq; |
||||
|
import static org.mockito.BDDMockito.given; |
||||
|
import static org.mockito.BDDMockito.never; |
||||
|
import static org.mockito.BDDMockito.then; |
||||
|
|
||||
|
@ExtendWith(MockitoExtension.class) |
||||
|
public class AuditLogServiceImplTest { |
||||
|
|
||||
|
private final TenantId TENANT_ID = TenantId.fromUUID(UUID.fromString("9114e9ac-6c28-4019-a2a7-b948cb9500d5")); |
||||
|
private final CustomerId CUSTOMER_ID = new CustomerId(UUID.fromString("d15822ef-09eb-49a6-9068-21b9c8ae3356")); |
||||
|
private final UserId USER_ID = new UserId(UUID.fromString("47a2c904-3a47-4530-91bb-51068a4610a7")); |
||||
|
private final DeviceId DEVICE_ID = new DeviceId(UUID.fromString("b913c12a-9942-4cbd-9481-c42dad4831b0")); |
||||
|
|
||||
|
private final String USER_NAME = "Test User"; |
||||
|
|
||||
|
@InjectMocks |
||||
|
private AuditLogServiceImpl auditLogService; |
||||
|
@Mock |
||||
|
private EntityService entityService; |
||||
|
@Mock |
||||
|
private AuditLogLevelFilter auditLogLevelFilter; |
||||
|
@Mock |
||||
|
private AuditLogDataValidator auditLogDataValidator; |
||||
|
@Mock |
||||
|
private JpaExecutorService executor; |
||||
|
@Mock |
||||
|
private AuditLogDao auditLogDao; |
||||
|
@Mock |
||||
|
private AuditLogSink auditLogSink; |
||||
|
|
||||
|
@Test |
||||
|
public void givenEntityIsNull_whenLogEntityAction_thenShouldFetchEntityName() { |
||||
|
// GIVEN
|
||||
|
given(auditLogLevelFilter.logEnabled(any(), any())).willReturn(true); |
||||
|
|
||||
|
// WHEN
|
||||
|
auditLogService.logEntityAction(TENANT_ID, CUSTOMER_ID, USER_ID, USER_NAME, DEVICE_ID, null, ActionType.ADDED, null); |
||||
|
|
||||
|
// THEN
|
||||
|
then(entityService).should().fetchEntityName(TENANT_ID, DEVICE_ID); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenActionTypeIsAlarmActionAndEntityIsAlarm_whenLogEntityAction_thenShouldGetEntityName() throws Exception { |
||||
|
// GIVEN
|
||||
|
given(auditLogLevelFilter.logEnabled(any(), any())).willReturn(true); |
||||
|
Alarm alarm = new Alarm(new AlarmId(UUID.fromString("55f577b3-6ef5-4b99-92dc-70eb78b2a970"))); |
||||
|
alarm.setType("Test Alarm"); |
||||
|
AlarmComment comment = new AlarmComment(); |
||||
|
comment.setComment(JacksonUtil.toJsonNode("{\"comment\": \"test\"}")); |
||||
|
|
||||
|
// WHEN
|
||||
|
auditLogService.logEntityAction(TENANT_ID, CUSTOMER_ID, USER_ID, USER_NAME, alarm.getId(), alarm, ActionType.ADDED_COMMENT, null, comment); |
||||
|
|
||||
|
// THEN
|
||||
|
then(entityService).should(never()).fetchEntityName(any(), any()); |
||||
|
ArgumentCaptor<Callable> submitTask = ArgumentCaptor.forClass(Callable.class); |
||||
|
then(executor).should().submit(submitTask.capture()); |
||||
|
submitTask.getValue().call(); |
||||
|
ArgumentCaptor<AuditLog> auditLogEntry = ArgumentCaptor.forClass(AuditLog.class); |
||||
|
then(auditLogDao).should().save(eq(TENANT_ID), auditLogEntry.capture()); |
||||
|
assertThat(auditLogEntry.getValue().getEntityName()).isEqualTo("Test Alarm"); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenActionTypeIsAlarmActionAndEntityIsAlarmInfo_whenLogEntityAction_thenShouldGetEntityOriginatorName() throws Exception { |
||||
|
// GIVEN
|
||||
|
given(auditLogLevelFilter.logEnabled(any(), any())).willReturn(true); |
||||
|
AlarmInfo alarmInfo = new AlarmInfo(); |
||||
|
alarmInfo.setOriginatorName("Test Device"); |
||||
|
|
||||
|
// WHEN
|
||||
|
auditLogService.logEntityAction(TENANT_ID, CUSTOMER_ID, USER_ID, USER_NAME, DEVICE_ID, alarmInfo, ActionType.ALARM_ASSIGNED, null); |
||||
|
|
||||
|
// THEN
|
||||
|
then(entityService).should(never()).fetchEntityName(any(), any()); |
||||
|
ArgumentCaptor<Callable> submitTask = ArgumentCaptor.forClass(Callable.class); |
||||
|
then(executor).should().submit(submitTask.capture()); |
||||
|
submitTask.getValue().call(); |
||||
|
ArgumentCaptor<AuditLog> auditLogEntry = ArgumentCaptor.forClass(AuditLog.class); |
||||
|
then(auditLogDao).should().save(eq(TENANT_ID), auditLogEntry.capture()); |
||||
|
assertThat(auditLogEntry.getValue().getEntityName()).isEqualTo("Test Device"); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenActionTypeIsAlarmActionAndEntityIsAlarm_whenLogEntityAction_thenShouldFetchEntityName() { |
||||
|
// GIVEN
|
||||
|
given(auditLogLevelFilter.logEnabled(any(), any())).willReturn(true); |
||||
|
|
||||
|
// WHEN
|
||||
|
auditLogService.logEntityAction(TENANT_ID, CUSTOMER_ID, USER_ID, USER_NAME, DEVICE_ID, new Alarm(), ActionType.ALARM_DELETE, null); |
||||
|
|
||||
|
// THEN
|
||||
|
then(entityService).should().fetchEntityName(TENANT_ID, DEVICE_ID); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue