van-vanich
5 years ago
4 changed files with 392 additions and 598 deletions
@ -0,0 +1,227 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.ObjectMapper; |
||||
|
import com.google.common.collect.Lists; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import org.junit.runner.RunWith; |
||||
|
import org.mockito.ArgumentCaptor; |
||||
|
import org.mockito.Mock; |
||||
|
import org.mockito.junit.MockitoJUnitRunner; |
||||
|
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.Device; |
||||
|
import org.thingsboard.server.common.data.User; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.RuleChainId; |
||||
|
import org.thingsboard.server.common.data.id.RuleNodeId; |
||||
|
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.StringDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgDataType; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.dao.asset.AssetService; |
||||
|
import org.thingsboard.server.dao.attributes.AttributesService; |
||||
|
import org.thingsboard.server.dao.device.DeviceService; |
||||
|
import org.thingsboard.server.dao.timeseries.TimeseriesService; |
||||
|
import org.thingsboard.server.dao.user.UserService; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
import static org.junit.Assert.assertEquals; |
||||
|
import static org.junit.Assert.assertTrue; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.ArgumentMatchers.anyCollection; |
||||
|
import static org.mockito.ArgumentMatchers.eq; |
||||
|
import static org.mockito.ArgumentMatchers.same; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
import static org.mockito.Mockito.verify; |
||||
|
import static org.mockito.Mockito.when; |
||||
|
import static org.thingsboard.rule.engine.api.TbRelationTypes.FAILURE; |
||||
|
import static org.thingsboard.server.common.data.DataConstants.SERVER_SCOPE; |
||||
|
|
||||
|
@RunWith(MockitoJUnitRunner.class) |
||||
|
public abstract class AbstractAttributeNodeTest { |
||||
|
final CustomerId customerId = new CustomerId(Uuids.timeBased()); |
||||
|
final TenantId tenantId = new TenantId(Uuids.timeBased()); |
||||
|
final RuleChainId ruleChainId = new RuleChainId(Uuids.timeBased()); |
||||
|
final RuleNodeId ruleNodeId = new RuleNodeId(Uuids.timeBased()); |
||||
|
final String keyAttrConf = "${word}"; |
||||
|
final String valueAttrConf = "result"; |
||||
|
@Mock |
||||
|
TbContext ctx; |
||||
|
@Mock |
||||
|
AttributesService attributesService; |
||||
|
@Mock |
||||
|
TimeseriesService timeseriesService; |
||||
|
@Mock |
||||
|
UserService userService; |
||||
|
@Mock |
||||
|
AssetService assetService; |
||||
|
@Mock |
||||
|
DeviceService deviceService; |
||||
|
TbMsg msg; |
||||
|
Map<String, String> metaData; |
||||
|
TbEntityGetAttrNode node; |
||||
|
|
||||
|
void init(TbEntityGetAttrNode node) throws TbNodeException { |
||||
|
ObjectMapper mapper = JacksonUtil.OBJECT_MAPPER; |
||||
|
TbNodeConfiguration nodeConfiguration = new TbNodeConfiguration(mapper.valueToTree(getTbNodeConfig())); |
||||
|
|
||||
|
metaData = new HashMap<>(); |
||||
|
metaData.putIfAbsent("word", "temperature"); |
||||
|
|
||||
|
this.node = node; |
||||
|
this.node.init(null, nodeConfiguration); |
||||
|
} |
||||
|
|
||||
|
void errorThrownIfCannotLoadAttributes(User user) { |
||||
|
msg = TbMsg.newMsg("USER", user.getId(), new TbMsgMetaData(), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getUserService()).thenReturn(userService); |
||||
|
when(userService.findUserByIdAsync(any(), eq(user.getId()))).thenReturn(Futures.immediateFuture(user)); |
||||
|
|
||||
|
when(ctx.getAttributesService()).thenReturn(attributesService); |
||||
|
when(attributesService.find(any(), eq(getEntityId()), eq(SERVER_SCOPE), anyCollection())) |
||||
|
.thenThrow(new IllegalStateException("something wrong")); |
||||
|
|
||||
|
node.onMsg(ctx, msg); |
||||
|
final ArgumentCaptor<Throwable> captor = ArgumentCaptor.forClass(Throwable.class); |
||||
|
verify(ctx).tellFailure(same(msg), captor.capture()); |
||||
|
|
||||
|
Throwable value = captor.getValue(); |
||||
|
assertEquals("something wrong", value.getMessage()); |
||||
|
assertTrue(msg.getMetaData().getData().isEmpty()); |
||||
|
} |
||||
|
|
||||
|
void errorThrownIfCannotLoadAttributesAsync(User user) { |
||||
|
|
||||
|
msg = TbMsg.newMsg("USER", user.getId(), new TbMsgMetaData(), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getUserService()).thenReturn(userService); |
||||
|
when(userService.findUserByIdAsync(any(), eq(user.getId()))).thenReturn(Futures.immediateFuture(user)); |
||||
|
|
||||
|
when(ctx.getAttributesService()).thenReturn(attributesService); |
||||
|
when(attributesService.find(any(), eq(getEntityId()), eq(SERVER_SCOPE), anyCollection())) |
||||
|
.thenReturn(Futures.immediateFailedFuture(new IllegalStateException("something wrong"))); |
||||
|
|
||||
|
node.onMsg(ctx, msg); |
||||
|
final ArgumentCaptor<Throwable> captor = ArgumentCaptor.forClass(Throwable.class); |
||||
|
verify(ctx).tellFailure(same(msg), captor.capture()); |
||||
|
|
||||
|
Throwable value = captor.getValue(); |
||||
|
assertEquals("something wrong", value.getMessage()); |
||||
|
assertTrue(msg.getMetaData().getData().isEmpty()); |
||||
|
} |
||||
|
|
||||
|
void failedChainUsedIfCustomerCannotBeFound(User user) { |
||||
|
msg = TbMsg.newMsg("USER", user.getId(), new TbMsgMetaData(), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getUserService()).thenReturn(userService); |
||||
|
when(userService.findUserByIdAsync(any(), eq(user.getId()))).thenReturn(Futures.immediateFuture(null)); |
||||
|
|
||||
|
node.onMsg(ctx, msg); |
||||
|
verify(ctx).tellNext(msg, FAILURE); |
||||
|
assertTrue(msg.getMetaData().getData().isEmpty()); |
||||
|
} |
||||
|
|
||||
|
void entityAttributeAddedInMetadata(EntityId entityId, String type) { |
||||
|
msg = TbMsg.newMsg(type, entityId, new TbMsgMetaData(metaData), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
entityAttributeFetched(getEntityId()); |
||||
|
} |
||||
|
|
||||
|
void usersCustomerAttributesFetched(User user) { |
||||
|
msg = TbMsg.newMsg("USER", user.getId(), new TbMsgMetaData(metaData), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getUserService()).thenReturn(userService); |
||||
|
when(userService.findUserByIdAsync(any(), eq(user.getId()))).thenReturn(Futures.immediateFuture(user)); |
||||
|
|
||||
|
entityAttributeFetched(getEntityId()); |
||||
|
} |
||||
|
|
||||
|
void assetsCustomerAttributesFetched(Asset asset) { |
||||
|
msg = TbMsg.newMsg("ASSET", asset.getId(), new TbMsgMetaData(metaData), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getAssetService()).thenReturn(assetService); |
||||
|
when(assetService.findAssetByIdAsync(any(), eq(asset.getId()))).thenReturn(Futures.immediateFuture(asset)); |
||||
|
|
||||
|
entityAttributeFetched(getEntityId()); |
||||
|
} |
||||
|
|
||||
|
void deviceCustomerAttributesFetched(Device device) { |
||||
|
msg = TbMsg.newMsg("USER", device.getId(), new TbMsgMetaData(metaData), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getDeviceService()).thenReturn(deviceService); |
||||
|
when(deviceService.findDeviceByIdAsync(any(), eq(device.getId()))).thenReturn(Futures.immediateFuture(device)); |
||||
|
|
||||
|
entityAttributeFetched(getEntityId()); |
||||
|
} |
||||
|
|
||||
|
void deviceCustomerTelemetryFetched(Device device) throws TbNodeException { |
||||
|
ObjectMapper mapper = JacksonUtil.OBJECT_MAPPER; |
||||
|
TbNodeConfiguration nodeConfiguration = new TbNodeConfiguration(mapper.valueToTree(getTbNodeConfigFotTelemetry())); |
||||
|
|
||||
|
TbEntityGetAttrNode node = getEmptyNode(); |
||||
|
node.init(null, nodeConfiguration); |
||||
|
|
||||
|
msg = TbMsg.newMsg("USER", device.getId(), new TbMsgMetaData(metaData), TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId); |
||||
|
|
||||
|
when(ctx.getDeviceService()).thenReturn(deviceService); |
||||
|
when(deviceService.findDeviceByIdAsync(any(), eq(device.getId()))).thenReturn(Futures.immediateFuture(device)); |
||||
|
|
||||
|
List<TsKvEntry> timeseries = Lists.newArrayList(new BasicTsKvEntry(1L, new StringDataEntry("temperature", "highest"))); |
||||
|
|
||||
|
when(ctx.getTimeseriesService()).thenReturn(timeseriesService); |
||||
|
when(timeseriesService.findLatest(any(), eq(getEntityId()), anyCollection())) |
||||
|
.thenReturn(Futures.immediateFuture(timeseries)); |
||||
|
|
||||
|
node.onMsg(ctx, msg); |
||||
|
verify(ctx).tellSuccess(msg); |
||||
|
assertEquals(msg.getMetaData().getValue("result"), "highest"); |
||||
|
} |
||||
|
|
||||
|
void entityAttributeFetched(EntityId entityId) { |
||||
|
List<AttributeKvEntry> attributes = Lists.newArrayList(new BaseAttributeKvEntry(new StringDataEntry("temperature", "high"), 1L)); |
||||
|
|
||||
|
when(ctx.getAttributesService()).thenReturn(attributesService); |
||||
|
when(attributesService.find(any(), eq(entityId), eq(SERVER_SCOPE), anyCollection())) |
||||
|
.thenReturn(Futures.immediateFuture(attributes)); |
||||
|
|
||||
|
node.onMsg(ctx, msg); |
||||
|
verify(ctx).tellSuccess(msg); |
||||
|
assertEquals(msg.getMetaData().getValue("result"), "high"); |
||||
|
} |
||||
|
|
||||
|
protected abstract TbEntityGetAttrNode getEmptyNode(); |
||||
|
|
||||
|
abstract <T> T getTbNodeConfig(); |
||||
|
|
||||
|
abstract <T> T getTbNodeConfigFotTelemetry(); |
||||
|
|
||||
|
abstract EntityId getEntityId(); |
||||
|
} |
||||
Loading…
Reference in new issue