From e184a53b05d995bd40945882dc7f71d9cfa3e207 Mon Sep 17 00:00:00 2001
From: YevhenBondarenko <ybondarenko@thingsboard.io>
Date: Fri, 29 Jan 2021 17:23:54 +0200
Subject: [PATCH] fixed TbDeviceProfileNodeTest

---
 .../rule/engine/profile/TbDeviceProfileNodeTest.java          | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/profile/TbDeviceProfileNodeTest.java b/rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/profile/TbDeviceProfileNodeTest.java
index 912fee62c2..ffa9da7e8b 100644
--- a/rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/profile/TbDeviceProfileNodeTest.java
+++ b/rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/profile/TbDeviceProfileNodeTest.java
@@ -48,6 +48,7 @@ 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.common.msg.session.SessionMsgType;
+import org.thingsboard.server.dao.device.DeviceService;
 import org.thingsboard.server.dao.timeseries.TimeseriesService;
 
 import java.util.Collections;
@@ -71,6 +72,8 @@ public class TbDeviceProfileNodeTest {
     private TimeseriesService timeseriesService;
     @Mock
     private RuleEngineAlarmService alarmService;
+    @Mock
+    private DeviceService deviceService;
 
     private TenantId tenantId = new TenantId(UUID.randomUUID());
     private DeviceId deviceId = new DeviceId(UUID.randomUUID());
@@ -188,6 +191,7 @@ public class TbDeviceProfileNodeTest {
         Mockito.when(ctx.getDeviceProfileCache()).thenReturn(cache);
         Mockito.when(ctx.getTimeseriesService()).thenReturn(timeseriesService);
         Mockito.when(ctx.getAlarmService()).thenReturn(alarmService);
+        Mockito.when(ctx.getDeviceService()).thenReturn(deviceService);
         TbNodeConfiguration nodeConfiguration = new TbNodeConfiguration(mapper.createObjectNode());
         node = new TbDeviceProfileNode();
         node.init(ctx, nodeConfiguration);