From c8b13fc3507645fb6d690d7b131708067b0b9891 Mon Sep 17 00:00:00 2001 From: IrynaMatveieva Date: Wed, 7 May 2025 11:13:36 +0300 Subject: [PATCH] added await for test --- .../DefaultTbLocalSubscriptionServiceTest.java | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/application/src/test/java/org/thingsboard/server/service/subscription/DefaultTbLocalSubscriptionServiceTest.java b/application/src/test/java/org/thingsboard/server/service/subscription/DefaultTbLocalSubscriptionServiceTest.java index 623dd6b9e2..3bb470830b 100644 --- a/application/src/test/java/org/thingsboard/server/service/subscription/DefaultTbLocalSubscriptionServiceTest.java +++ b/application/src/test/java/org/thingsboard/server/service/subscription/DefaultTbLocalSubscriptionServiceTest.java @@ -49,8 +49,10 @@ import java.util.List; import java.util.Map; import java.util.UUID; import java.util.concurrent.Executors; +import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicReference; +import static org.awaitility.Awaitility.await; import static org.junit.jupiter.api.Assertions.assertFalse; import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertTrue; @@ -163,9 +165,13 @@ public class DefaultTbLocalSubscriptionServiceTest { TsKvEntry kv = new BasicTsKvEntry(ts, new StringDataEntry(key, "42")); subscriptionService.onTimeSeriesUpdate(deviceId, List.of(kv), mock(TbCallback.class)); - TelemetrySubscriptionUpdate update = capturedUpdate.get(); - assertNotNull(update); - assertTrue(update.getLatestValues().containsKey(key)); + await().atMost(5, TimeUnit.SECONDS) + .pollInterval(5, TimeUnit.MILLISECONDS) + .untilAsserted(() -> { + TelemetrySubscriptionUpdate update = capturedUpdate.get(); + assertNotNull(update); + assertTrue(update.getLatestValues().containsKey(key)); + }); } }