From 1c724df353d802fd6117d05afbcc625860487c88 Mon Sep 17 00:00:00 2001
From: dashevchenko <dshevchenko@thingsboard.io>
Date: Wed, 20 Mar 2024 17:59:55 +0200
Subject: [PATCH] added await to make tests more stable

---
 .../server/controller/EntityQueryControllerTest.java  | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java b/application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java
index 395f90618c..39ab60dc49 100644
--- a/application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java
+++ b/application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java
@@ -727,9 +727,14 @@ public class EntityQueryControllerTest extends AbstractControllerTest {
     }
 
     private void checkEntitiesCount(EntityCountQuery query, int expectedNumOfDevices) {
-        var count = doPost("/api/entitiesQuery/count", query, Integer.class);
-        assertThat(count).isEqualTo(expectedNumOfDevices);
-    }
+        Awaitility.await()
+                .alias("count by query")
+                .atMost(30, TimeUnit.SECONDS)
+                .until(() -> {
+                    var count = doPost("/api/entitiesQuery/count", query, Integer.class);
+                    return count == expectedNumOfDevices;
+                });
+   }
 
     private KeyFilter getEntityFieldStringEqualToKeyFilter(String keyName, String value) {
         KeyFilter tenantOwnerNameFilter = new KeyFilter();