简柏林 / thingskit · Commits

Commit 41796a4d891093bbcc90955d573c15229d1c20ea

Authored by ShvaykaD
Committed by Andrew Shvayka
1 parent 67e8bf8c

fix AlarmNodeTests

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