简柏林 / thingskit · Commits

Commit b78d9ca18959f4b9719813545bdf594f6b50cd57

Authored by ShvaykaD
Committed by Andrew Shvayka
1 parent 702a0913

fix AlarmNodeTests

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