From 6f86ca0d0825485d9735366f26b93358f5662aaf Mon Sep 17 00:00:00 2001
From: Viacheslav Klimov <viacheslavklimov11@gmail.com>
Date: Thu, 19 Feb 2026 15:22:51 +0200
Subject: [PATCH] Await save future in integration test to prevent flakiness

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 .../service/timeseries/nosql/TimeseriesServiceNoSqlTest.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dao/src/test/java/org/thingsboard/server/dao/service/timeseries/nosql/TimeseriesServiceNoSqlTest.java b/dao/src/test/java/org/thingsboard/server/dao/service/timeseries/nosql/TimeseriesServiceNoSqlTest.java
index 146542f501..aa43826734 100644
--- a/dao/src/test/java/org/thingsboard/server/dao/service/timeseries/nosql/TimeseriesServiceNoSqlTest.java
+++ b/dao/src/test/java/org/thingsboard/server/dao/service/timeseries/nosql/TimeseriesServiceNoSqlTest.java
@@ -114,7 +114,7 @@ public class TimeseriesServiceNoSqlTest extends BaseTimeseriesServiceTest {
         for (int i = 0; i < 5; i++) {
             entries.add(new BasicTsKvEntry(TimeUnit.MINUTES.toMillis(i + 1), new StringDataEntry("bigKey", value)));
         }
-        tsService.save(tenantId, deviceId, entries, 0);
+        tsService.save(tenantId, deviceId, entries, 0).get(MAX_TIMEOUT, TimeUnit.SECONDS);
 
         long originalLimit = (long) ReflectionTestUtils.getField(cassandraBaseTimeseriesDao, "maxResultSetSizeBytes");
         try {