add test for TbAbstractGetAttributesNode
This commit is contained in:
Yuriy Lytvynchuk
parent
4c9e52bce9
commit
4c277daadd
@ -93,7 +93,7 @@ public abstract class TbAbstractGetAttributesNode<C extends TbGetAttributesNodeC
|
|||||||
JsonNode msgDataNode = JacksonUtil.toJsonNode(msg.getData());
|
JsonNode msgDataNode = JacksonUtil.toJsonNode(msg.getData());
|
||||||
if (fetchToData) {
|
if (fetchToData) {
|
||||||
if (!msgDataNode.isObject()) {
|
if (!msgDataNode.isObject()) {
|
||||||
ctx.tellFailure(msg, new IllegalArgumentException("Msg body is not object!"));
|
ctx.tellFailure(msg, new IllegalArgumentException("Msg body is not an object!"));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -0,0 +1,296 @@
|
|||||||
|
/**
|
||||||
|
* 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.metadata;
|
||||||
|
|
||||||
|
import com.datastax.oss.driver.api.core.uuid.Uuids;
|
||||||
|
import com.fasterxml.jackson.databind.JsonNode;
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
||||||
|
import com.google.common.util.concurrent.Futures;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.mockito.ArgumentCaptor;
|
||||||
|
import org.mockito.Mock;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.mockito.junit.MockitoJUnitRunner;
|
||||||
|
import org.thingsboard.common.util.AbstractListeningExecutor;
|
||||||
|
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.common.data.DataConstants;
|
||||||
|
import org.thingsboard.server.common.data.id.DeviceId;
|
||||||
|
import org.thingsboard.server.common.data.id.EntityId;
|
||||||
|
import org.thingsboard.server.common.data.id.TenantId;
|
||||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry;
|
||||||
|
import org.thingsboard.server.common.data.kv.BaseAttributeKvEntry;
|
||||||
|
import org.thingsboard.server.common.data.kv.BasicTsKvEntry;
|
||||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry;
|
||||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry;
|
||||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry;
|
||||||
|
import org.thingsboard.server.common.msg.TbMsg;
|
||||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData;
|
||||||
|
import org.thingsboard.server.dao.attributes.AttributesService;
|
||||||
|
import org.thingsboard.server.dao.timeseries.TimeseriesService;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import static org.mockito.ArgumentMatchers.any;
|
||||||
|
import static org.mockito.Mockito.lenient;
|
||||||
|
import static org.mockito.Mockito.never;
|
||||||
|
import static org.mockito.Mockito.times;
|
||||||
|
|
||||||
|
@RunWith(MockitoJUnitRunner.class)
|
||||||
|
public class TbAbstractGetAttributesNodeTest {
|
||||||
|
|
||||||
|
final ObjectMapper mapper = new ObjectMapper();
|
||||||
|
|
||||||
|
private EntityId originator = new DeviceId(Uuids.timeBased());
|
||||||
|
private TenantId tenantId = TenantId.fromUUID(Uuids.timeBased());
|
||||||
|
|
||||||
|
@Mock
|
||||||
|
private TbContext ctx;
|
||||||
|
@Mock
|
||||||
|
private AttributesService attributesService;
|
||||||
|
@Mock
|
||||||
|
private TimeseriesService tsService;
|
||||||
|
private AbstractListeningExecutor dbExecutor;
|
||||||
|
|
||||||
|
private List<String> clientAttributes;
|
||||||
|
private List<String> serverAttributes;
|
||||||
|
private List<String> sharedAttributes;
|
||||||
|
private List<String> tsKeys;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void before() throws TbNodeException {
|
||||||
|
dbExecutor = new AbstractListeningExecutor() {
|
||||||
|
@Override
|
||||||
|
protected int getThreadPollSize() {
|
||||||
|
return 3;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
dbExecutor.init();
|
||||||
|
|
||||||
|
Mockito.reset(ctx);
|
||||||
|
Mockito.reset(attributesService);
|
||||||
|
Mockito.reset(tsService);
|
||||||
|
|
||||||
|
Mockito.reset(ctx);
|
||||||
|
Mockito.reset(attributesService);
|
||||||
|
Mockito.reset(tsService);
|
||||||
|
|
||||||
|
lenient().when(ctx.getAttributesService()).thenReturn(attributesService);
|
||||||
|
lenient().when(ctx.getTimeseriesService()).thenReturn(tsService);
|
||||||
|
lenient().when(ctx.getTenantId()).thenReturn(tenantId);
|
||||||
|
lenient().when(ctx.getDbCallbackExecutor()).thenReturn(dbExecutor);
|
||||||
|
|
||||||
|
clientAttributes = getAttributeNames("client");
|
||||||
|
serverAttributes = getAttributeNames("server");
|
||||||
|
sharedAttributes = getAttributeNames("shared");
|
||||||
|
tsKeys = List.of("temperature", "humidity", "unknown");
|
||||||
|
|
||||||
|
Mockito.when(attributesService.find(tenantId, originator, DataConstants.CLIENT_SCOPE, clientAttributes))
|
||||||
|
.thenReturn(Futures.immediateFuture(getListAttributeKvEntry(clientAttributes)));
|
||||||
|
|
||||||
|
|
||||||
|
Mockito.when(attributesService.find(tenantId, originator, DataConstants.SERVER_SCOPE, serverAttributes))
|
||||||
|
.thenReturn(Futures.immediateFuture(getListAttributeKvEntry(serverAttributes)));
|
||||||
|
|
||||||
|
|
||||||
|
Mockito.when(attributesService.find(tenantId, originator, DataConstants.SHARED_SCOPE, sharedAttributes))
|
||||||
|
.thenReturn(Futures.immediateFuture(getListAttributeKvEntry(sharedAttributes)));
|
||||||
|
|
||||||
|
Mockito.when(tsService.findLatest(tenantId, originator, tsKeys))
|
||||||
|
.thenReturn(Futures.immediateFuture(getListTelemetryKvEntry(tsKeys)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void after() {
|
||||||
|
dbExecutor.destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void fetchToMetadata_whenOnMsg_then_success() throws Exception {
|
||||||
|
TbGetAttributesNode node = initNode(false, false, false);
|
||||||
|
TbMsg msg = getTbMsg(originator);
|
||||||
|
node.onMsg(ctx, msg);
|
||||||
|
|
||||||
|
TbMsg resultMsg = checkMsg();
|
||||||
|
TbMsgMetaData msgMetaData = resultMsg.getMetaData();
|
||||||
|
|
||||||
|
//check attributes
|
||||||
|
checkAttributes(clientAttributes, "cs_", false, msgMetaData, null);
|
||||||
|
checkAttributes(serverAttributes, "ss_", false, msgMetaData, null);
|
||||||
|
checkAttributes(sharedAttributes, "shared_", false, msgMetaData, null);
|
||||||
|
|
||||||
|
//check timeseries
|
||||||
|
checkTs(tsKeys, false, false, msgMetaData, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void fetchToData_whenOnMsg_then_success() throws Exception {
|
||||||
|
TbGetAttributesNode node = initNode(true, true, false);
|
||||||
|
TbMsg msg = getTbMsg(originator);
|
||||||
|
node.onMsg(ctx, msg);
|
||||||
|
|
||||||
|
TbMsg resultMsg = checkMsg();
|
||||||
|
JsonNode msgData = JacksonUtil.toJsonNode(resultMsg.getData());
|
||||||
|
|
||||||
|
//check attributes
|
||||||
|
checkAttributes(clientAttributes, "cs_", true, null, msgData);
|
||||||
|
checkAttributes(serverAttributes, "ss_", true, null, msgData);
|
||||||
|
checkAttributes(sharedAttributes, "shared_", true, null, msgData);
|
||||||
|
|
||||||
|
//check timeseries with ts
|
||||||
|
checkTs(tsKeys, true, true, null, msgData);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void fetchToData_whenOnMsg_then_failure() throws Exception {
|
||||||
|
TbGetAttributesNode node = initNode(true, true, true);
|
||||||
|
TbMsg msg = getTbMsg(originator);
|
||||||
|
node.onMsg(ctx, msg);
|
||||||
|
|
||||||
|
ArgumentCaptor<TbMsg> newMsgCaptor = ArgumentCaptor.forClass(TbMsg.class);
|
||||||
|
ArgumentCaptor<Exception> exceptionCaptor = ArgumentCaptor.forClass(Exception.class);
|
||||||
|
Mockito.verify(ctx, never()).tellSuccess(any());
|
||||||
|
Mockito.verify(ctx, Mockito.timeout(5000)).tellFailure(newMsgCaptor.capture(), exceptionCaptor.capture());
|
||||||
|
|
||||||
|
Assert.assertSame(newMsgCaptor.getValue(), msg);
|
||||||
|
Assert.assertNotNull(exceptionCaptor.getValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void fetchToData_whenOnMsg_then_data_not_object_failure() throws Exception {
|
||||||
|
TbGetAttributesNode node = initNode(true, true, true);
|
||||||
|
TbMsg msg = TbMsg.newMsg("TEST", originator, new TbMsgMetaData(), "[]");
|
||||||
|
node.onMsg(ctx, msg);
|
||||||
|
|
||||||
|
ArgumentCaptor<TbMsg> newMsgCaptor = ArgumentCaptor.forClass(TbMsg.class);
|
||||||
|
ArgumentCaptor<Exception> exceptionCaptor = ArgumentCaptor.forClass(Exception.class);
|
||||||
|
Mockito.verify(ctx, never()).tellSuccess(any());
|
||||||
|
Mockito.verify(ctx, Mockito.timeout(5000)).tellFailure(newMsgCaptor.capture(), exceptionCaptor.capture());
|
||||||
|
|
||||||
|
Assert.assertSame(newMsgCaptor.getValue(), msg);
|
||||||
|
Assert.assertNotNull(exceptionCaptor.getValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
private TbMsg checkMsg() {
|
||||||
|
ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class);
|
||||||
|
Mockito.verify(ctx, Mockito.timeout(5000)).tellSuccess(msgCaptor.capture());
|
||||||
|
|
||||||
|
TbMsg resultMsg = msgCaptor.getValue();
|
||||||
|
Assert.assertNotNull(resultMsg);
|
||||||
|
Assert.assertNotNull(resultMsg.getMetaData());
|
||||||
|
Assert.assertNotNull(resultMsg.getData());
|
||||||
|
return resultMsg;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void checkAttributes(List<String> attributes, String prefix, boolean fetchToData, TbMsgMetaData msgMetaData, JsonNode msgData) {
|
||||||
|
attributes.stream()
|
||||||
|
.filter(attribute -> !attribute.equals("unknown"))
|
||||||
|
.forEach(attribute -> {
|
||||||
|
String result;
|
||||||
|
if (fetchToData) {
|
||||||
|
result = msgData.get(prefix + attribute).asText();
|
||||||
|
} else {
|
||||||
|
result = msgMetaData.getValue(prefix + attribute);
|
||||||
|
}
|
||||||
|
Assert.assertNotNull(result);
|
||||||
|
Assert.assertEquals(attribute + "_value", result);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private void checkTs(List<String> tsKeys, boolean fetchToData, boolean getLatestValueWithTs, TbMsgMetaData msgMetaData, JsonNode msgData) {
|
||||||
|
long tsValue = 1L;
|
||||||
|
for (String key : tsKeys) {
|
||||||
|
if (key.equals("unknown")) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
String result;
|
||||||
|
if (fetchToData) {
|
||||||
|
if (getLatestValueWithTs) {
|
||||||
|
JsonNode resultTs = msgData.get(key);
|
||||||
|
Assert.assertNotNull(resultTs);
|
||||||
|
Assert.assertNotNull(resultTs.get("value"));
|
||||||
|
Assert.assertNotNull(resultTs.get("ts"));
|
||||||
|
result = resultTs.get("value").asText();
|
||||||
|
} else {
|
||||||
|
result = msgData.get(key).asText();
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
result = msgMetaData.getValue(key);
|
||||||
|
}
|
||||||
|
Assert.assertNotNull(result);
|
||||||
|
Assert.assertEquals(String.valueOf(tsValue), result);
|
||||||
|
tsValue++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private TbGetAttributesNode initNode(boolean fetchToData, boolean getLatestValueWithTs, boolean isTellFailureIfAbsent) throws TbNodeException {
|
||||||
|
TbGetAttributesNodeConfiguration config = new TbGetAttributesNodeConfiguration();
|
||||||
|
config.setClientAttributeNames(List.of("client_attr_1", "client_attr_2", "${client_attr_metadata}", "unknown"));
|
||||||
|
config.setServerAttributeNames(List.of("server_attr_1", "server_attr_2", "${server_attr_metadata}", "unknown"));
|
||||||
|
config.setSharedAttributeNames(List.of("shared_attr_1", "shared_attr_2", "$[shared_attr_data]", "unknown"));
|
||||||
|
config.setLatestTsKeyNames(List.of("temperature", "humidity", "unknown"));
|
||||||
|
config.setFetchToData(fetchToData);
|
||||||
|
config.setGetLatestValueWithTs(getLatestValueWithTs);
|
||||||
|
config.setTellFailureIfAbsent(isTellFailureIfAbsent);
|
||||||
|
TbNodeConfiguration nodeConfiguration = new TbNodeConfiguration(mapper.valueToTree(config));
|
||||||
|
TbGetAttributesNode node = new TbGetAttributesNode();
|
||||||
|
node.init(ctx, nodeConfiguration);
|
||||||
|
return node;
|
||||||
|
}
|
||||||
|
|
||||||
|
private TbMsg getTbMsg(EntityId entityId) {
|
||||||
|
ObjectNode msgData = JacksonUtil.newObjectNode();
|
||||||
|
msgData.put("shared_attr_data", "shared_attr_3");
|
||||||
|
|
||||||
|
TbMsgMetaData msgMetaData = new TbMsgMetaData();
|
||||||
|
msgMetaData.putValue("client_attr_metadata", "client_attr_3");
|
||||||
|
msgMetaData.putValue("server_attr_metadata", "server_attr_3");
|
||||||
|
|
||||||
|
return TbMsg.newMsg("TEST", entityId, msgMetaData, msgData.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<String> getAttributeNames(String prefix) {
|
||||||
|
return List.of(prefix + "_attr_1", prefix + "_attr_2", prefix + "_attr_3", "unknown");
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<AttributeKvEntry> getListAttributeKvEntry(List<String> attributes) {
|
||||||
|
List<AttributeKvEntry> attributeKvEntries = new ArrayList<>();
|
||||||
|
attributes.stream().filter(attribute -> !attribute.equals("unknown")).forEach(attribute -> attributeKvEntries.add(new BaseAttributeKvEntry(System.currentTimeMillis(), new StringDataEntry(attribute, attribute + "_value"))));
|
||||||
|
return attributeKvEntries;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<TsKvEntry> getListTelemetryKvEntry(List<String> keys) {
|
||||||
|
long value = 1L;
|
||||||
|
List<TsKvEntry> kvEntries = new ArrayList<>();
|
||||||
|
for (String key : keys) {
|
||||||
|
if (key.equals("unknown")) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
kvEntries.add(new BasicTsKvEntry(System.currentTimeMillis(), new LongDataEntry(key, value)));
|
||||||
|
value++;
|
||||||
|
}
|
||||||
|
return kvEntries;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user