Volodymyr Babak
4 years ago
30 changed files with 654 additions and 124 deletions
@ -0,0 +1,176 @@ |
|||||
|
/** |
||||
|
* 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.rule.engine.profile; |
||||
|
|
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.mockito.ArgumentCaptor; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.rule.engine.api.RuleEngineAlarmService; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.server.common.data.DataConstants; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmCondition; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmConditionFilter; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmConditionKeyType; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmRule; |
||||
|
import org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm; |
||||
|
import org.thingsboard.server.common.data.device.profile.DeviceProfileData; |
||||
|
import org.thingsboard.server.common.data.device.profile.SimpleAlarmConditionSpec; |
||||
|
import org.thingsboard.server.common.data.id.AlarmId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.query.BooleanFilterPredicate; |
||||
|
import org.thingsboard.server.common.data.query.EntityKeyValueType; |
||||
|
import org.thingsboard.server.common.data.query.FilterPredicateValue; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.common.msg.session.SessionMsgType; |
||||
|
import org.thingsboard.server.dao.attributes.AttributesService; |
||||
|
import org.thingsboard.server.dao.device.DeviceService; |
||||
|
|
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.TreeMap; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.ArgumentMatchers.anyCollection; |
||||
|
import static org.mockito.ArgumentMatchers.anyString; |
||||
|
import static org.mockito.ArgumentMatchers.eq; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
import static org.mockito.Mockito.never; |
||||
|
import static org.mockito.Mockito.reset; |
||||
|
import static org.mockito.Mockito.verify; |
||||
|
import static org.mockito.Mockito.when; |
||||
|
|
||||
|
public class DeviceStateTest { |
||||
|
|
||||
|
private TbContext ctx; |
||||
|
|
||||
|
@Before |
||||
|
public void beforeEach() { |
||||
|
ctx = mock(TbContext.class); |
||||
|
|
||||
|
when(ctx.getDeviceService()).thenReturn(mock(DeviceService.class)); |
||||
|
|
||||
|
AttributesService attributesService = mock(AttributesService.class); |
||||
|
when(attributesService.find(any(), any(), any(), anyCollection())).thenReturn(Futures.immediateFuture(Collections.emptyList())); |
||||
|
when(ctx.getAttributesService()).thenReturn(attributesService); |
||||
|
|
||||
|
RuleEngineAlarmService alarmService = mock(RuleEngineAlarmService.class); |
||||
|
when(alarmService.findLatestByOriginatorAndType(any(), any(), any())).thenReturn(Futures.immediateFuture(null)); |
||||
|
when(alarmService.createOrUpdateAlarm(any())).thenAnswer(invocationOnMock -> { |
||||
|
Alarm alarm = invocationOnMock.getArgument(0); |
||||
|
alarm.setId(new AlarmId(UUID.randomUUID())); |
||||
|
return alarm; |
||||
|
}); |
||||
|
when(ctx.getAlarmService()).thenReturn(alarmService); |
||||
|
|
||||
|
when(ctx.newMsg(any(), any(), any(), any(), any(), any())).thenAnswer(invocationOnMock -> { |
||||
|
String data = invocationOnMock.getArgument(invocationOnMock.getArguments().length - 1); |
||||
|
return TbMsg.newMsg(null, null, new TbMsgMetaData(), data); |
||||
|
}); |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void whenAttributeIsDeleted_thenUnneededAlarmRulesAreNotReevaluated() throws Exception { |
||||
|
|
||||
|
DeviceProfileAlarm alarmConfig = createAlarmConfigWithBoolAttrCondition("enabled", false); |
||||
|
DeviceId deviceId = new DeviceId(UUID.randomUUID()); |
||||
|
DeviceState deviceState = createDeviceState(deviceId, alarmConfig); |
||||
|
|
||||
|
TbMsg attributeUpdateMsg = TbMsg.newMsg(SessionMsgType.POST_ATTRIBUTES_REQUEST.name(), |
||||
|
deviceId, new TbMsgMetaData(), "{ \"enabled\": false }"); |
||||
|
|
||||
|
deviceState.process(ctx, attributeUpdateMsg); |
||||
|
|
||||
|
ArgumentCaptor<TbMsg> resultMsgCaptor = ArgumentCaptor.forClass(TbMsg.class); |
||||
|
verify(ctx).enqueueForTellNext(resultMsgCaptor.capture(), eq("Alarm Created")); |
||||
|
Alarm alarm = JacksonUtil.fromString(resultMsgCaptor.getValue().getData(), Alarm.class); |
||||
|
|
||||
|
deviceState.process(ctx, TbMsg.newMsg(DataConstants.ALARM_CLEAR, deviceId, new TbMsgMetaData(), JacksonUtil.toString(alarm))); |
||||
|
reset(ctx); |
||||
|
|
||||
|
String deletedAttributes = "{ \"attributes\": [ \"other\" ] }"; |
||||
|
deviceState.process(ctx, TbMsg.newMsg(DataConstants.ATTRIBUTES_DELETED, deviceId, new TbMsgMetaData(), deletedAttributes)); |
||||
|
verify(ctx, never()).enqueueForTellNext(any(), anyString()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void whenDeletingClearedAlarm_thenNoError() throws Exception { |
||||
|
DeviceProfileAlarm alarmConfig = createAlarmConfigWithBoolAttrCondition("enabled", false); |
||||
|
DeviceId deviceId = new DeviceId(UUID.randomUUID()); |
||||
|
DeviceState deviceState = createDeviceState(deviceId, alarmConfig); |
||||
|
|
||||
|
TbMsg attributeUpdateMsg = TbMsg.newMsg(SessionMsgType.POST_ATTRIBUTES_REQUEST.name(), |
||||
|
deviceId, new TbMsgMetaData(), "{ \"enabled\": false }"); |
||||
|
|
||||
|
deviceState.process(ctx, attributeUpdateMsg); |
||||
|
ArgumentCaptor<TbMsg> resultMsgCaptor = ArgumentCaptor.forClass(TbMsg.class); |
||||
|
verify(ctx).enqueueForTellNext(resultMsgCaptor.capture(), eq("Alarm Created")); |
||||
|
Alarm alarm = JacksonUtil.fromString(resultMsgCaptor.getValue().getData(), Alarm.class); |
||||
|
|
||||
|
deviceState.process(ctx, TbMsg.newMsg(DataConstants.ALARM_CLEAR, deviceId, new TbMsgMetaData(), JacksonUtil.toString(alarm))); |
||||
|
|
||||
|
TbMsg alarmDeleteNotification = TbMsg.newMsg(DataConstants.ALARM_DELETE, deviceId, new TbMsgMetaData(), JacksonUtil.toString(alarm)); |
||||
|
assertDoesNotThrow(() -> { |
||||
|
deviceState.process(ctx, alarmDeleteNotification); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
private DeviceState createDeviceState(DeviceId deviceId, DeviceProfileAlarm... alarmConfigs) { |
||||
|
DeviceProfile deviceProfile = new DeviceProfile(); |
||||
|
DeviceProfileData profileData = new DeviceProfileData(); |
||||
|
profileData.setAlarms(List.of(alarmConfigs)); |
||||
|
deviceProfile.setProfileData(profileData); |
||||
|
|
||||
|
ProfileState profileState = new ProfileState(deviceProfile); |
||||
|
return new DeviceState(ctx, new TbDeviceProfileNodeConfiguration(), |
||||
|
deviceId, profileState, null); |
||||
|
} |
||||
|
|
||||
|
private DeviceProfileAlarm createAlarmConfigWithBoolAttrCondition(String key, boolean value) { |
||||
|
|
||||
|
AlarmConditionFilter condition = new AlarmConditionFilter(); |
||||
|
condition.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, key)); |
||||
|
condition.setValueType(EntityKeyValueType.BOOLEAN); |
||||
|
BooleanFilterPredicate predicate = new BooleanFilterPredicate(); |
||||
|
predicate.setOperation(BooleanFilterPredicate.BooleanOperation.EQUAL); |
||||
|
predicate.setValue(new FilterPredicateValue<>(value)); |
||||
|
condition.setPredicate(predicate); |
||||
|
|
||||
|
DeviceProfileAlarm alarmConfig = new DeviceProfileAlarm(); |
||||
|
alarmConfig.setId("MyAlarmID"); |
||||
|
alarmConfig.setAlarmType("MyAlarm"); |
||||
|
AlarmRule alarmRule = new AlarmRule(); |
||||
|
AlarmCondition alarmCondition = new AlarmCondition(); |
||||
|
alarmCondition.setSpec(new SimpleAlarmConditionSpec()); |
||||
|
alarmCondition.setCondition(List.of(condition)); |
||||
|
alarmRule.setCondition(alarmCondition); |
||||
|
alarmConfig.setCreateRules(new TreeMap<>(Map.of(AlarmSeverity.CRITICAL, alarmRule))); |
||||
|
|
||||
|
return alarmConfig; |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue