Dmytro Skarzhynets
3 years ago
9 changed files with 797 additions and 39 deletions
@ -0,0 +1,178 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 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.action; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.rule.engine.api.RuleNode; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.rule.engine.api.TbNode; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeConfiguration; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeException; |
||||
|
import org.thingsboard.rule.engine.api.util.TbNodeUtils; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.data.plugin.ComponentType; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
import org.thingsboard.server.queue.TbQueueCallback; |
||||
|
import org.thingsboard.server.queue.TbQueueMsgMetadata; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RuleNode( |
||||
|
type = ComponentType.ACTION, |
||||
|
name = "device state", |
||||
|
nodeDescription = "Triggers device connectivity events", |
||||
|
nodeDetails = "If incoming message originator is a device," + |
||||
|
" registers configured event for that device in the Device State Service," + |
||||
|
" which sends appropriate message to the Rule Engine. " + |
||||
|
"Incoming message is forwarded using the <code>Success</code> chain," + |
||||
|
" unless an unexpected error occurs during message processing" + |
||||
|
" then incoming message is forwarded using the <code>Failure</code> chain." + |
||||
|
"<br>" + |
||||
|
"Supported device connectivity events are:" + |
||||
|
"<ul>" + |
||||
|
"<li>Connect event</li>" + |
||||
|
"<li>Disconnect event</li>" + |
||||
|
"<li>Activity event</li>" + |
||||
|
"<li>Inactivity event</li>" + |
||||
|
"</ul>" + |
||||
|
"This node is particularly useful when device isn't using transports to receive data," + |
||||
|
" such as when fetching data from external API or computing new data within the rule chain.", |
||||
|
configClazz = TbDeviceStateNodeConfiguration.class, |
||||
|
uiResources = {"static/rulenode/rulenode-core-config.js"}, |
||||
|
configDirective = "tbActionNodeDeviceStateConfig" |
||||
|
) |
||||
|
public class TbDeviceStateNode implements TbNode { |
||||
|
|
||||
|
private static final TbQueueCallback EMPTY_CALLBACK = new TbQueueCallback() { |
||||
|
|
||||
|
@Override |
||||
|
public void onSuccess(TbQueueMsgMetadata metadata) { |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
} |
||||
|
|
||||
|
}; |
||||
|
|
||||
|
private final Map<TbMsgType, ConnectivityEvent> SUPPORTED_EVENTS = Map.of( |
||||
|
TbMsgType.CONNECT_EVENT, this::sendDeviceConnectMsg, |
||||
|
TbMsgType.ACTIVITY_EVENT, this::sendDeviceActivityMsg, |
||||
|
TbMsgType.DISCONNECT_EVENT, this::sendDeviceDisconnectMsg, |
||||
|
TbMsgType.INACTIVITY_EVENT, this::sendDeviceInactivityMsg |
||||
|
); |
||||
|
|
||||
|
private interface ConnectivityEvent { |
||||
|
|
||||
|
void sendEvent(TbContext ctx, TbMsg msg); |
||||
|
|
||||
|
} |
||||
|
|
||||
|
private TbMsgType event; |
||||
|
|
||||
|
@Override |
||||
|
public void init(TbContext ctx, TbNodeConfiguration configuration) throws TbNodeException { |
||||
|
TbMsgType event = TbNodeUtils.convert(configuration, TbDeviceStateNodeConfiguration.class).getEvent(); |
||||
|
if (event == null) { |
||||
|
throw new TbNodeException("Event cannot be null!", true); |
||||
|
} |
||||
|
if (!SUPPORTED_EVENTS.containsKey(event)) { |
||||
|
throw new TbNodeException("Unsupported event: " + event, true); |
||||
|
} |
||||
|
this.event = event; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onMsg(TbContext ctx, TbMsg msg) { |
||||
|
var originator = msg.getOriginator(); |
||||
|
if (!ctx.isLocalEntity(originator)) { |
||||
|
log.warn("[{}][device-state-node] Received message from non-local entity [{}]!", ctx.getSelfId(), originator); |
||||
|
return; |
||||
|
} |
||||
|
if (!EntityType.DEVICE.equals(originator.getEntityType())) { |
||||
|
ctx.tellSuccess(msg); |
||||
|
return; |
||||
|
} |
||||
|
SUPPORTED_EVENTS.get(event).sendEvent(ctx, msg); |
||||
|
ctx.tellSuccess(msg); |
||||
|
} |
||||
|
|
||||
|
private void sendDeviceConnectMsg(TbContext ctx, TbMsg msg) { |
||||
|
var tenantUuid = ctx.getTenantId().getId(); |
||||
|
var deviceUuid = msg.getOriginator().getId(); |
||||
|
var deviceConnectMsg = TransportProtos.DeviceConnectProto.newBuilder() |
||||
|
.setTenantIdMSB(tenantUuid.getMostSignificantBits()) |
||||
|
.setTenantIdLSB(tenantUuid.getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(deviceUuid.getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(deviceUuid.getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
var toCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceConnectMsg(deviceConnectMsg) |
||||
|
.build(); |
||||
|
ctx.getClusterService().pushMsgToCore(ctx.getTenantId(), msg.getOriginator(), toCoreMsg, EMPTY_CALLBACK); |
||||
|
} |
||||
|
|
||||
|
private void sendDeviceActivityMsg(TbContext ctx, TbMsg msg) { |
||||
|
var tenantUuid = ctx.getTenantId().getId(); |
||||
|
var deviceUuid = msg.getOriginator().getId(); |
||||
|
var deviceActivityMsg = TransportProtos.DeviceActivityProto.newBuilder() |
||||
|
.setTenantIdMSB(tenantUuid.getMostSignificantBits()) |
||||
|
.setTenantIdLSB(tenantUuid.getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(deviceUuid.getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(deviceUuid.getLeastSignificantBits()) |
||||
|
.setLastActivityTime(System.currentTimeMillis()) |
||||
|
.build(); |
||||
|
var toCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceActivityMsg(deviceActivityMsg) |
||||
|
.build(); |
||||
|
ctx.getClusterService().pushMsgToCore(ctx.getTenantId(), msg.getOriginator(), toCoreMsg, EMPTY_CALLBACK); |
||||
|
} |
||||
|
|
||||
|
private void sendDeviceDisconnectMsg(TbContext ctx, TbMsg msg) { |
||||
|
var tenantUuid = ctx.getTenantId().getId(); |
||||
|
var deviceUuid = msg.getOriginator().getId(); |
||||
|
var deviceDisconnectMsg = TransportProtos.DeviceDisconnectProto.newBuilder() |
||||
|
.setTenantIdMSB(tenantUuid.getMostSignificantBits()) |
||||
|
.setTenantIdLSB(tenantUuid.getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(deviceUuid.getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(deviceUuid.getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
var toCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceDisconnectMsg(deviceDisconnectMsg) |
||||
|
.build(); |
||||
|
ctx.getClusterService().pushMsgToCore(ctx.getTenantId(), msg.getOriginator(), toCoreMsg, EMPTY_CALLBACK); |
||||
|
} |
||||
|
|
||||
|
private void sendDeviceInactivityMsg(TbContext ctx, TbMsg msg) { |
||||
|
var tenantUuid = ctx.getTenantId().getId(); |
||||
|
var deviceUuid = msg.getOriginator().getId(); |
||||
|
var deviceInactivityMsg = TransportProtos.DeviceInactivityProto.newBuilder() |
||||
|
.setTenantIdMSB(tenantUuid.getMostSignificantBits()) |
||||
|
.setTenantIdLSB(tenantUuid.getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(deviceUuid.getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(deviceUuid.getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
var toCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceInactivityMsg(deviceInactivityMsg) |
||||
|
.build(); |
||||
|
ctx.getClusterService().pushMsgToCore(ctx.getTenantId(), msg.getOriginator(), toCoreMsg, EMPTY_CALLBACK); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,34 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 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.action; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.rule.engine.api.NodeConfiguration; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
|
||||
|
@Data |
||||
|
public class TbDeviceStateNodeConfiguration implements NodeConfiguration<TbDeviceStateNodeConfiguration> { |
||||
|
|
||||
|
private TbMsgType event; |
||||
|
|
||||
|
@Override |
||||
|
public TbDeviceStateNodeConfiguration defaultConfiguration() { |
||||
|
var config = new TbDeviceStateNodeConfiguration(); |
||||
|
config.setEvent(TbMsgType.ACTIVITY_EVENT); |
||||
|
return config; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,313 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 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.action; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeAll; |
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.junit.jupiter.api.extension.ExtendWith; |
||||
|
import org.mockito.ArgumentCaptor; |
||||
|
import org.mockito.Mock; |
||||
|
import org.mockito.junit.jupiter.MockitoExtension; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeConfiguration; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeException; |
||||
|
import org.thingsboard.server.cluster.TbClusterService; |
||||
|
import org.thingsboard.server.common.data.Device; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.ArgumentMatchers.eq; |
||||
|
import static org.mockito.BDDMockito.given; |
||||
|
import static org.mockito.BDDMockito.then; |
||||
|
import static org.mockito.Mockito.only; |
||||
|
import static org.mockito.Mockito.times; |
||||
|
|
||||
|
@ExtendWith(MockitoExtension.class) |
||||
|
public class TbDeviceStateNodeTest { |
||||
|
|
||||
|
private static TenantId DUMMY_TENANT_ID; |
||||
|
private static TbMsg DUMMY_MSG; |
||||
|
private static DeviceId DUMMY_MSG_ORIGINATOR; |
||||
|
@Mock |
||||
|
private TbContext ctxMock; |
||||
|
@Mock |
||||
|
private TbClusterService tbClusterServiceMock; |
||||
|
private TbDeviceStateNode node; |
||||
|
private TbDeviceStateNodeConfiguration config; |
||||
|
|
||||
|
@BeforeAll |
||||
|
public static void init() { |
||||
|
DUMMY_TENANT_ID = TenantId.fromUUID(UUID.randomUUID()); |
||||
|
|
||||
|
var device = new Device(); |
||||
|
device.setTenantId(DUMMY_TENANT_ID); |
||||
|
device.setId(new DeviceId(UUID.randomUUID())); |
||||
|
device.setName("My humidity sensor"); |
||||
|
device.setType("Humidity sensor"); |
||||
|
device.setDeviceProfileId(new DeviceProfileId(UUID.randomUUID())); |
||||
|
var metaData = new TbMsgMetaData(); |
||||
|
metaData.putValue("deviceName", device.getName()); |
||||
|
metaData.putValue("deviceType", device.getType()); |
||||
|
metaData.putValue("ts", String.valueOf(System.currentTimeMillis())); |
||||
|
var data = JacksonUtil.newObjectNode(); |
||||
|
data.put("humidity", 58.3); |
||||
|
DUMMY_MSG = TbMsg.newMsg( |
||||
|
TbMsgType.POST_TELEMETRY_REQUEST, device.getId(), metaData, JacksonUtil.toString(data) |
||||
|
); |
||||
|
DUMMY_MSG_ORIGINATOR = device.getId(); |
||||
|
} |
||||
|
|
||||
|
@BeforeEach |
||||
|
public void setUp() { |
||||
|
node = new TbDeviceStateNode(); |
||||
|
config = new TbDeviceStateNodeConfiguration(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenDefaultConfiguration_whenInvoked_thenCorrectValuesAreSet() { |
||||
|
// GIVEN-WHEN
|
||||
|
config = config.defaultConfiguration(); |
||||
|
|
||||
|
// THEN
|
||||
|
assertThat(config.getEvent()).isEqualTo(TbMsgType.ACTIVITY_EVENT); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenNullEventInConfig_whenInit_thenThrowsUnrecoverableTbNodeException() { |
||||
|
// GIVEN
|
||||
|
config.setEvent(null); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
|
||||
|
// WHEN-THEN
|
||||
|
assertThatThrownBy(() -> node.init(ctxMock, nodeConfig)) |
||||
|
.isInstanceOf(TbNodeException.class) |
||||
|
.hasMessage("Event cannot be null!") |
||||
|
.matches(e -> ((TbNodeException) e).isUnrecoverable()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenUnsupportedEventInConfig_whenInit_thenThrowsUnrecoverableTbNodeException() { |
||||
|
// GIVEN
|
||||
|
var unsupportedEvent = TbMsgType.TO_SERVER_RPC_REQUEST; |
||||
|
config.setEvent(unsupportedEvent); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
|
||||
|
// WHEN-THEN
|
||||
|
assertThatThrownBy(() -> node.init(ctxMock, nodeConfig)) |
||||
|
.isInstanceOf(TbNodeException.class) |
||||
|
.hasMessage("Unsupported event: " + unsupportedEvent) |
||||
|
.matches(e -> ((TbNodeException) e).isUnrecoverable()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenNonDeviceOriginator_whenOnMsg_thenTellsSuccessAndNoActivityActionsTriggered() { |
||||
|
// GIVEN
|
||||
|
var msg = TbMsg.newMsg(TbMsgType.ENTITY_CREATED, new AssetId(UUID.randomUUID()), new TbMsgMetaData(), "{}"); |
||||
|
|
||||
|
// WHEN
|
||||
|
node.onMsg(ctxMock, msg); |
||||
|
|
||||
|
// THEN
|
||||
|
then(ctxMock).should(only()).tellSuccess(msg); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenNonLocalOriginator_whenOnMsg_thenTellsSuccessAndNoActivityActionsTriggered() { |
||||
|
// GIVEN
|
||||
|
given(ctxMock.isLocalEntity(DUMMY_MSG.getOriginator())).willReturn(false); |
||||
|
|
||||
|
// WHEN
|
||||
|
node.onMsg(ctxMock, DUMMY_MSG); |
||||
|
|
||||
|
// THEN
|
||||
|
then(ctxMock).should(times(1)).isLocalEntity(DUMMY_MSG_ORIGINATOR); |
||||
|
then(ctxMock).should(times(1)).tellSuccess(DUMMY_MSG); |
||||
|
then(ctxMock).shouldHaveNoMoreInteractions(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenConnectEventInConfig_whenOnMsg_thenOnDeviceConnectCalledAndTellsSuccess() throws TbNodeException { |
||||
|
// GIVEN
|
||||
|
config.setEvent(TbMsgType.CONNECT_EVENT); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
node.init(ctxMock, nodeConfig); |
||||
|
given(ctxMock.getTenantId()).willReturn(DUMMY_TENANT_ID); |
||||
|
given(ctxMock.getClusterService()).willReturn(tbClusterServiceMock); |
||||
|
given(ctxMock.isLocalEntity(DUMMY_MSG.getOriginator())).willReturn(true); |
||||
|
|
||||
|
// WHEN
|
||||
|
node.onMsg(ctxMock, DUMMY_MSG); |
||||
|
|
||||
|
// THEN
|
||||
|
var protoCaptor = ArgumentCaptor.forClass(TransportProtos.ToCoreMsg.class); |
||||
|
then(tbClusterServiceMock).should(times(1)) |
||||
|
.pushMsgToCore(eq(DUMMY_TENANT_ID), eq(DUMMY_MSG_ORIGINATOR), protoCaptor.capture(), any()); |
||||
|
|
||||
|
TransportProtos.DeviceConnectProto expectedDeviceConnectMsg = TransportProtos.DeviceConnectProto.newBuilder() |
||||
|
.setTenantIdMSB(DUMMY_TENANT_ID.getId().getMostSignificantBits()) |
||||
|
.setTenantIdLSB(DUMMY_TENANT_ID.getId().getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(DUMMY_MSG_ORIGINATOR.getId().getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(DUMMY_MSG_ORIGINATOR.getId().getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
TransportProtos.ToCoreMsg expectedToCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceConnectMsg(expectedDeviceConnectMsg) |
||||
|
.build(); |
||||
|
assertThat(expectedToCoreMsg).isEqualTo(protoCaptor.getValue()); |
||||
|
|
||||
|
then(ctxMock).should(times(1)).tellSuccess(DUMMY_MSG); |
||||
|
then(tbClusterServiceMock).shouldHaveNoMoreInteractions(); |
||||
|
then(ctxMock).shouldHaveNoMoreInteractions(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenActivityEventInConfig_whenOnMsg_thenOnDeviceActivityCalledWithCorrectTimeAndTellsSuccess() |
||||
|
throws TbNodeException { |
||||
|
// GIVEN
|
||||
|
config.setEvent(TbMsgType.ACTIVITY_EVENT); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
node.init(ctxMock, nodeConfig); |
||||
|
given(ctxMock.getTenantId()).willReturn(DUMMY_TENANT_ID); |
||||
|
given(ctxMock.getClusterService()).willReturn(tbClusterServiceMock); |
||||
|
given(ctxMock.isLocalEntity(DUMMY_MSG.getOriginator())).willReturn(true); |
||||
|
|
||||
|
|
||||
|
// WHEN
|
||||
|
long timeBeforeCall = System.currentTimeMillis(); |
||||
|
node.onMsg(ctxMock, DUMMY_MSG); |
||||
|
long timeAfterCall = System.currentTimeMillis(); |
||||
|
|
||||
|
// THEN
|
||||
|
var protoCaptor = ArgumentCaptor.forClass(TransportProtos.ToCoreMsg.class); |
||||
|
then(tbClusterServiceMock).should(times(1)) |
||||
|
.pushMsgToCore(eq(DUMMY_TENANT_ID), eq(DUMMY_MSG_ORIGINATOR), protoCaptor.capture(), any()); |
||||
|
|
||||
|
TransportProtos.ToCoreMsg actualToCoreMsg = protoCaptor.getValue(); |
||||
|
long actualLastActivityTime = actualToCoreMsg.getDeviceActivityMsg().getLastActivityTime(); |
||||
|
|
||||
|
assertThat(actualLastActivityTime).isGreaterThanOrEqualTo(timeBeforeCall); |
||||
|
assertThat(actualLastActivityTime).isLessThanOrEqualTo(timeAfterCall); |
||||
|
|
||||
|
TransportProtos.DeviceActivityProto updatedActivityMsg = actualToCoreMsg.getDeviceActivityMsg().toBuilder() |
||||
|
.setLastActivityTime(123L) |
||||
|
.build(); |
||||
|
TransportProtos.ToCoreMsg updatedToCoreMsg = actualToCoreMsg.toBuilder() |
||||
|
.setDeviceActivityMsg(updatedActivityMsg) |
||||
|
.build(); |
||||
|
|
||||
|
TransportProtos.DeviceActivityProto expectedDeviceActivityMsg = TransportProtos.DeviceActivityProto.newBuilder() |
||||
|
.setTenantIdMSB(DUMMY_TENANT_ID.getId().getMostSignificantBits()) |
||||
|
.setTenantIdLSB(DUMMY_TENANT_ID.getId().getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(DUMMY_MSG_ORIGINATOR.getId().getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(DUMMY_MSG_ORIGINATOR.getId().getLeastSignificantBits()) |
||||
|
.setLastActivityTime(123L) |
||||
|
.build(); |
||||
|
TransportProtos.ToCoreMsg expectedToCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceActivityMsg(expectedDeviceActivityMsg) |
||||
|
.build(); |
||||
|
|
||||
|
assertThat(updatedToCoreMsg).isEqualTo(expectedToCoreMsg); |
||||
|
|
||||
|
then(ctxMock).should(times(1)).tellSuccess(DUMMY_MSG); |
||||
|
then(tbClusterServiceMock).shouldHaveNoMoreInteractions(); |
||||
|
then(ctxMock).shouldHaveNoMoreInteractions(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenInactivityEventInConfig_whenOnMsg_thenOnDeviceInactivityCalledAndTellsSuccess() |
||||
|
throws TbNodeException { |
||||
|
// GIVEN
|
||||
|
config.setEvent(TbMsgType.INACTIVITY_EVENT); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
node.init(ctxMock, nodeConfig); |
||||
|
given(ctxMock.getTenantId()).willReturn(DUMMY_TENANT_ID); |
||||
|
given(ctxMock.getClusterService()).willReturn(tbClusterServiceMock); |
||||
|
given(ctxMock.isLocalEntity(DUMMY_MSG.getOriginator())).willReturn(true); |
||||
|
|
||||
|
// WHEN
|
||||
|
node.onMsg(ctxMock, DUMMY_MSG); |
||||
|
|
||||
|
// THEN
|
||||
|
var protoCaptor = ArgumentCaptor.forClass(TransportProtos.ToCoreMsg.class); |
||||
|
then(tbClusterServiceMock).should(times(1)) |
||||
|
.pushMsgToCore(eq(DUMMY_TENANT_ID), eq(DUMMY_MSG_ORIGINATOR), protoCaptor.capture(), any()); |
||||
|
|
||||
|
TransportProtos.DeviceInactivityProto expectedDeviceInactivityMsg = |
||||
|
TransportProtos.DeviceInactivityProto.newBuilder() |
||||
|
.setTenantIdMSB(DUMMY_TENANT_ID.getId().getMostSignificantBits()) |
||||
|
.setTenantIdLSB(DUMMY_TENANT_ID.getId().getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(DUMMY_MSG_ORIGINATOR.getId().getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(DUMMY_MSG_ORIGINATOR.getId().getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
TransportProtos.ToCoreMsg expectedToCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceInactivityMsg(expectedDeviceInactivityMsg) |
||||
|
.build(); |
||||
|
assertThat(expectedToCoreMsg).isEqualTo(protoCaptor.getValue()); |
||||
|
|
||||
|
then(ctxMock).should(times(1)).tellSuccess(DUMMY_MSG); |
||||
|
then(tbClusterServiceMock).shouldHaveNoMoreInteractions(); |
||||
|
then(ctxMock).shouldHaveNoMoreInteractions(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void givenDisconnectEventInConfig_whenOnMsg_thenOnDeviceDisconnectCalledAndTellsSuccess() |
||||
|
throws TbNodeException { |
||||
|
// GIVEN
|
||||
|
config.setEvent(TbMsgType.DISCONNECT_EVENT); |
||||
|
var nodeConfig = new TbNodeConfiguration(JacksonUtil.valueToTree(config)); |
||||
|
node.init(ctxMock, nodeConfig); |
||||
|
given(ctxMock.getTenantId()).willReturn(DUMMY_TENANT_ID); |
||||
|
given(ctxMock.getClusterService()).willReturn(tbClusterServiceMock); |
||||
|
given(ctxMock.isLocalEntity(DUMMY_MSG.getOriginator())).willReturn(true); |
||||
|
|
||||
|
// WHEN
|
||||
|
node.onMsg(ctxMock, DUMMY_MSG); |
||||
|
|
||||
|
// THEN
|
||||
|
var protoCaptor = ArgumentCaptor.forClass(TransportProtos.ToCoreMsg.class); |
||||
|
then(tbClusterServiceMock).should(times(1)) |
||||
|
.pushMsgToCore(eq(DUMMY_TENANT_ID), eq(DUMMY_MSG_ORIGINATOR), protoCaptor.capture(), any()); |
||||
|
|
||||
|
TransportProtos.DeviceDisconnectProto expectedDeviceDisconnectMsg = |
||||
|
TransportProtos.DeviceDisconnectProto.newBuilder() |
||||
|
.setTenantIdMSB(DUMMY_TENANT_ID.getId().getMostSignificantBits()) |
||||
|
.setTenantIdLSB(DUMMY_TENANT_ID.getId().getLeastSignificantBits()) |
||||
|
.setDeviceIdMSB(DUMMY_MSG_ORIGINATOR.getId().getMostSignificantBits()) |
||||
|
.setDeviceIdLSB(DUMMY_MSG_ORIGINATOR.getId().getLeastSignificantBits()) |
||||
|
.build(); |
||||
|
TransportProtos.ToCoreMsg expectedToCoreMsg = TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setDeviceDisconnectMsg(expectedDeviceDisconnectMsg) |
||||
|
.build(); |
||||
|
assertThat(expectedToCoreMsg).isEqualTo(protoCaptor.getValue()); |
||||
|
|
||||
|
then(ctxMock).should(times(1)).tellSuccess(DUMMY_MSG); |
||||
|
then(tbClusterServiceMock).shouldHaveNoMoreInteractions(); |
||||
|
then(ctxMock).shouldHaveNoMoreInteractions(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue