Commit 41796a4d891093bbcc90955d573c15229d1c20ea
Committed by
Andrew Shvayka
Andrew Shvayka
1 parent
67e8bf8c
fix AlarmNodeTests
Showing
1 changed file
with
8 additions
and
0 deletions
| @@ -35,6 +35,7 @@ import org.thingsboard.rule.engine.api.ScriptEngine; | @@ -35,6 +35,7 @@ import org.thingsboard.rule.engine.api.ScriptEngine; | ||
| 35 | import org.thingsboard.rule.engine.api.TbContext; | 35 | import org.thingsboard.rule.engine.api.TbContext; |
| 36 | import org.thingsboard.rule.engine.api.TbNodeConfiguration; | 36 | import org.thingsboard.rule.engine.api.TbNodeConfiguration; |
| 37 | import org.thingsboard.rule.engine.api.TbNodeException; | 37 | import org.thingsboard.rule.engine.api.TbNodeException; |
| 38 | +import org.thingsboard.server.common.data.DataConstants; | ||
| 38 | import org.thingsboard.server.common.data.alarm.Alarm; | 39 | import org.thingsboard.server.common.data.alarm.Alarm; |
| 39 | import org.thingsboard.server.common.data.id.AlarmId; | 40 | import org.thingsboard.server.common.data.id.AlarmId; |
| 40 | import org.thingsboard.server.common.data.id.DeviceId; | 41 | import org.thingsboard.server.common.data.id.DeviceId; |
| @@ -249,6 +250,9 @@ public class TbAlarmNodeTest { | @@ -249,6 +250,9 @@ public class TbAlarmNodeTest { | ||
| 249 | 250 | ||
| 250 | node.onMsg(ctx, msg); | 251 | node.onMsg(ctx, msg); |
| 251 | 252 | ||
| 253 | + verify(ctx).enqueue(any(), successCaptor.capture(), failureCaptor.capture()); | ||
| 254 | + successCaptor.getValue().run(); | ||
| 255 | + | ||
| 252 | verify(ctx).tellNext(any(), eq("Updated")); | 256 | verify(ctx).tellNext(any(), eq("Updated")); |
| 253 | 257 | ||
| 254 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); | 258 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); |
| @@ -298,6 +302,8 @@ public class TbAlarmNodeTest { | @@ -298,6 +302,8 @@ public class TbAlarmNodeTest { | ||
| 298 | 302 | ||
| 299 | node.onMsg(ctx, msg); | 303 | node.onMsg(ctx, msg); |
| 300 | 304 | ||
| 305 | + verify(ctx).enqueue(any(), successCaptor.capture(), failureCaptor.capture()); | ||
| 306 | + successCaptor.getValue().run(); | ||
| 301 | verify(ctx).tellNext(any(), eq("Cleared")); | 307 | verify(ctx).tellNext(any(), eq("Cleared")); |
| 302 | 308 | ||
| 303 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); | 309 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); |
| @@ -346,6 +352,8 @@ public class TbAlarmNodeTest { | @@ -346,6 +352,8 @@ public class TbAlarmNodeTest { | ||
| 346 | 352 | ||
| 347 | node.onMsg(ctx, msg); | 353 | node.onMsg(ctx, msg); |
| 348 | 354 | ||
| 355 | + verify(ctx).enqueue(any(), successCaptor.capture(), failureCaptor.capture()); | ||
| 356 | + successCaptor.getValue().run(); | ||
| 349 | verify(ctx).tellNext(any(), eq("Cleared")); | 357 | verify(ctx).tellNext(any(), eq("Cleared")); |
| 350 | 358 | ||
| 351 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); | 359 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); |