ydb_persqueue: Fix TWriteSession_TestBrokenPolicy test

qyryq · Aug 30, 2024 · 1e96cd9 · 1e96cd9
1 parent 297d5b4
commit 1e96cd9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ydb/public/sdk/cpp/client/ydb_persqueue_public/ut/retry_policy_ut.cpp b/ydb/public/sdk/cpp/client/ydb_persqueue_public/ut/retry_policy_ut.cpp
@@ -56,10 +56,10 @@ Y_UNIT_TEST_SUITE(RetryPolicy) {
         helper.Policy->Initialize();
         helper.Policy->ExpectFatalBreakDown();
         helper.EventLoop->AllowStop();
-        auto f1 = helper.Write(false);
         helper.Setup->KickTablets();
-        helper.Write(false);
+        helper.Setup->WaitForTabletsDown();
 
+        auto f1 = helper.Write(false);
         helper.EventLoop->WaitForStop();
         UNIT_ASSERT(!f1.HasValue());
         helper.Setup = nullptr;