Fix vhost test flakiness

ydb-platform · Feb 3, 2025 · d382677 · d382677
1 parent 251f1d6
commit d382677
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/cloud/blockstore/vhost-server/server_ut.cpp b/cloud/blockstore/vhost-server/server_ut.cpp
@@ -290,7 +290,21 @@ class TServerTest
         TSimpleStats prevStats;
         TCompleteStats stats;
         for (int i = 0; i != 5; ++i) {
+            // Save critical events from previous attempt.
+            auto critEvents = std::move(stats.CriticalEvents);
+
             stats = Server->GetStats(prevStats);
+
+            // Combine critical events from previous and current attempt.
+            if (critEvents) {
+                for (auto& critEvent: stats.CriticalEvents) {
+                    critEvents.push_back(std::move(critEvent));
+                }
+                stats.CriticalEvents = std::move(critEvents);
+            }
+
+            // Check that the current attempt to get statistics has brought
+            // everything we need.
             if (func(stats)) {
                 break;
             }