nickAS21
4 years ago
9 changed files with 269 additions and 44 deletions
@ -0,0 +1,132 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.controller; |
|||
|
|||
import org.mockito.Mockito; |
|||
import org.springframework.boot.test.mock.mockito.SpyBean; |
|||
import org.thingsboard.server.cluster.TbClusterService; |
|||
import org.thingsboard.server.common.data.EntityType; |
|||
import org.thingsboard.server.common.data.HasName; |
|||
import org.thingsboard.server.common.data.audit.ActionType; |
|||
import org.thingsboard.server.common.data.id.CustomerId; |
|||
import org.thingsboard.server.common.data.id.EntityId; |
|||
import org.thingsboard.server.common.data.id.EntityIdFactory; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.id.UserId; |
|||
import org.thingsboard.server.dao.audit.AuditLogService; |
|||
import org.thingsboard.server.dao.model.ModelConstants; |
|||
|
|||
import java.util.Locale; |
|||
|
|||
import static org.mockito.Mockito.never; |
|||
import static org.mockito.Mockito.times; |
|||
import static org.thingsboard.server.service.entitiy.DefaultTbNotificationEntityService.edgeTypeByActionType; |
|||
|
|||
public abstract class AbstractNotifyEntityTest extends AbstractWebTest { |
|||
|
|||
@SpyBean |
|||
protected TbClusterService tbClusterService; |
|||
|
|||
@SpyBean |
|||
protected AuditLogService auditLogService; |
|||
|
|||
protected void testNotifyEntityOk(HasName entity, EntityId entityId, EntityId originatorId, |
|||
TenantId tenantId, CustomerId customerId, UserId userId, String userName, |
|||
ActionType actionType) { |
|||
testSendNotificationMsgToEdgeServiceOk(entityId, tenantId, actionType); |
|||
testLogEntityActionOk(entity, originatorId, tenantId, customerId, userId, userName, actionType); |
|||
testPushMsgToRuleEngineOk(entity, originatorId, tenantId, customerId, userId, userName, actionType); |
|||
Mockito.reset(tbClusterService, auditLogService); |
|||
} |
|||
|
|||
protected void testNotifyEntityDeleteOk(HasName entity, EntityId entityId, EntityId originatorId, |
|||
TenantId tenantId, CustomerId customerId, UserId userId, String userName, |
|||
ActionType actionType) { |
|||
Mockito.verify(tbClusterService, never()).sendNotificationMsgToEdgeService(Mockito.any(), |
|||
Mockito.any(), Mockito.any(entityId.getClass()), Mockito.any(), Mockito.any(), Mockito.any()); |
|||
testLogEntityActionOk(entity, originatorId, tenantId, customerId, userId, userName, actionType); |
|||
testPushMsgToRuleEngineOk(entity, originatorId, tenantId, customerId, userId, userName, actionType); |
|||
} |
|||
|
|||
private void testNotifyEntityError(EntityId entityId, HasName entity, TenantId tenantId, |
|||
UserId userId, String userName, ActionType actionType, Exception exp, |
|||
Object... additionalInfo) { |
|||
CustomerId customer_NULL_UUID = (CustomerId) EntityIdFactory.getByTypeAndUuid(EntityType.CUSTOMER, ModelConstants.NULL_UUID); |
|||
EntityId entity_NULL_UUID = EntityIdFactory.getByTypeAndUuid(EntityType.valueOf(entity.getClass().toString().substring(entity.getClass().toString().lastIndexOf(".") + 1).toUpperCase(Locale.ENGLISH)), |
|||
ModelConstants.NULL_UUID); |
|||
testNotificationMsgToEdgeServiceNever(entityId); |
|||
if (additionalInfo.length > 0) { |
|||
Mockito.verify(auditLogService, times(1)).logEntityAction(Mockito.eq(tenantId), |
|||
Mockito.eq(customer_NULL_UUID), Mockito.eq(userId), Mockito.eq(userName), |
|||
Mockito.eq(entity_NULL_UUID), Mockito.any(entity.getClass()), Mockito.eq(actionType), |
|||
Mockito.any(exp.getClass()), Mockito.eq(additionalInfo)); |
|||
} else { |
|||
Mockito.verify(auditLogService, times(1)).logEntityAction(Mockito.eq(tenantId), |
|||
Mockito.eq(customer_NULL_UUID), Mockito.eq(userId), Mockito.eq(userName), |
|||
Mockito.eq(entity_NULL_UUID), Mockito.any(entity.getClass()), Mockito.eq(actionType), |
|||
Mockito.any(exp.getClass()), Mockito.isNull()); |
|||
} |
|||
Mockito.verify(tbClusterService, never()).pushMsgToRuleEngine(Mockito.any(), Mockito.any(entityId.getClass()), |
|||
Mockito.any(), Mockito.any()); |
|||
Mockito.reset(tbClusterService, auditLogService); |
|||
} |
|||
|
|||
protected void testNotifyEntityNever(EntityId entityId, HasName entity) { |
|||
testNotificationMsgToEdgeServiceNever(entityId); |
|||
testLogEntityActionNever(entityId, entity); |
|||
testPushMsgToRuleEngineNever(entityId); |
|||
} |
|||
|
|||
protected void testNotificationMsgToEdgeServiceNever(EntityId entityId) { |
|||
Mockito.verify(tbClusterService, never()).sendNotificationMsgToEdgeService(Mockito.any(), |
|||
Mockito.any(), Mockito.any(entityId.getClass()), Mockito.any(), Mockito.any(), Mockito.any()); |
|||
} |
|||
|
|||
protected void testLogEntityActionNever(EntityId entityId, HasName entity) { |
|||
Mockito.verify(auditLogService, never()).logEntityAction(Mockito.any(), Mockito.any(), |
|||
Mockito.any(), Mockito.any(), Mockito.any(entityId.getClass()), Mockito.any(entity.getClass()), |
|||
Mockito.any(), Mockito.any()); |
|||
} |
|||
|
|||
protected void testPushMsgToRuleEngineNever(EntityId entityId) { |
|||
Mockito.verify(tbClusterService, never()).pushMsgToRuleEngine(Mockito.any(), |
|||
Mockito.any(entityId.getClass()), Mockito.any(), Mockito.any()); |
|||
} |
|||
|
|||
private void testLogEntityActionOk(HasName entity, EntityId originatorId, |
|||
TenantId tenantId, CustomerId customerId, UserId userId, String userName, |
|||
ActionType actionType) { |
|||
Mockito.verify(auditLogService, times(1)).logEntityAction(Mockito.eq(tenantId), Mockito.eq(customerId), |
|||
Mockito.eq(userId), Mockito.eq(userName), Mockito.eq(originatorId), |
|||
Mockito.eq(entity), Mockito.eq(actionType), Mockito.isNull()); |
|||
} |
|||
|
|||
private void testPushMsgToRuleEngineOk(HasName entity, EntityId originatorId, |
|||
TenantId tenantId, CustomerId customerId, UserId userId, String userName, |
|||
ActionType actionType) { |
|||
|
|||
Mockito.verify(auditLogService, times(1)).logEntityAction(Mockito.eq(tenantId), Mockito.eq(customerId), |
|||
Mockito.eq(userId), Mockito.eq(userName), Mockito.eq(originatorId), |
|||
Mockito.eq(entity), Mockito.eq(actionType), Mockito.isNull()); |
|||
} |
|||
|
|||
private void testSendNotificationMsgToEdgeServiceOk(EntityId entityId, TenantId tenantId, ActionType actionType) { |
|||
Mockito.verify(tbClusterService, times(1)).sendNotificationMsgToEdgeService(Mockito.eq(tenantId), |
|||
Mockito.isNull(), Mockito.eq(entityId), Mockito.isNull(), Mockito.isNull(), |
|||
Mockito.eq(edgeTypeByActionType(actionType))); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue