Commit b78d9ca18959f4b9719813545bdf594f6b50cd57
Committed by
Andrew Shvayka
Andrew Shvayka
1 parent
702a0913
fix AlarmNodeTests
Showing
1 changed file
with
8 additions
and
0 deletions
| @@ -34,6 +34,7 @@ import org.thingsboard.rule.engine.api.ScriptEngine; | @@ -34,6 +34,7 @@ import org.thingsboard.rule.engine.api.ScriptEngine; | ||
| 34 | import org.thingsboard.rule.engine.api.TbContext; | 34 | import org.thingsboard.rule.engine.api.TbContext; |
| 35 | import org.thingsboard.rule.engine.api.TbNodeConfiguration; | 35 | import org.thingsboard.rule.engine.api.TbNodeConfiguration; |
| 36 | import org.thingsboard.rule.engine.api.TbNodeException; | 36 | import org.thingsboard.rule.engine.api.TbNodeException; |
| 37 | +import org.thingsboard.server.common.data.DataConstants; | ||
| 37 | import org.thingsboard.server.common.data.alarm.Alarm; | 38 | import org.thingsboard.server.common.data.alarm.Alarm; |
| 38 | import org.thingsboard.server.common.data.id.AlarmId; | 39 | import org.thingsboard.server.common.data.id.AlarmId; |
| 39 | import org.thingsboard.server.common.data.id.DeviceId; | 40 | 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); |
| @@ -297,6 +301,8 @@ public class TbAlarmNodeTest { | @@ -297,6 +301,8 @@ public class TbAlarmNodeTest { | ||
| 297 | 301 | ||
| 298 | node.onMsg(ctx, msg); | 302 | node.onMsg(ctx, msg); |
| 299 | 303 | ||
| 304 | + verify(ctx).enqueue(any(), successCaptor.capture(), failureCaptor.capture()); | ||
| 305 | + successCaptor.getValue().run(); | ||
| 300 | verify(ctx).tellNext(any(), eq("Cleared")); | 306 | verify(ctx).tellNext(any(), eq("Cleared")); |
| 301 | 307 | ||
| 302 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); | 308 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); |
| @@ -345,6 +351,8 @@ public class TbAlarmNodeTest { | @@ -345,6 +351,8 @@ public class TbAlarmNodeTest { | ||
| 345 | 351 | ||
| 346 | node.onMsg(ctx, msg); | 352 | node.onMsg(ctx, msg); |
| 347 | 353 | ||
| 354 | + verify(ctx).enqueue(any(), successCaptor.capture(), failureCaptor.capture()); | ||
| 355 | + successCaptor.getValue().run(); | ||
| 348 | verify(ctx).tellNext(any(), eq("Cleared")); | 356 | verify(ctx).tellNext(any(), eq("Cleared")); |
| 349 | 357 | ||
| 350 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); | 358 | ArgumentCaptor<TbMsg> msgCaptor = ArgumentCaptor.forClass(TbMsg.class); |