diff options
| author | FloatingCrowbar <[email protected]> | 2024-10-23 13:24:10 +0300 |
|---|---|---|
| committer | GitHub <[email protected]> | 2024-10-23 13:24:10 +0300 |
| commit | 3545be80cf0b59be093cd7a1dd5da8660068653a (patch) | |
| tree | 6f4cb99a1ea0ef4d268e71ab600df25cd46bcda3 | |
| parent | e6a3afc293b017cb2f45e996ececa53f165bfac9 (diff) | |
Stabilize DR tests (#10750)
| -rw-r--r-- | ydb/core/persqueue/ut/common/pq_ut_common.cpp | 7 | ||||
| -rw-r--r-- | ydb/services/persqueue_v1/persqueue_ut.cpp | 2 |
2 files changed, 6 insertions, 3 deletions
diff --git a/ydb/core/persqueue/ut/common/pq_ut_common.cpp b/ydb/core/persqueue/ut/common/pq_ut_common.cpp index 87bdcbda765..6132f236056 100644 --- a/ydb/core/persqueue/ut/common/pq_ut_common.cpp +++ b/ydb/core/persqueue/ut/common/pq_ut_common.cpp @@ -722,11 +722,14 @@ void CmdSetOffset(const ui32 partition, const TString& user, ui64 offset, bool e TActorId CmdCreateSession(const TPQCmdSettings& settings, TTestContext& tc) { - TActorId pipeClient = tc.Runtime->ConnectToPipe(tc.BalancerTabletId, tc.Edge, 0, GetPipeConfigWithRetries()); TActorId tabletPipe = tc.Runtime->ConnectToPipe(tc.TabletId, tc.Edge, 0, GetPipeConfigWithRetries()); - TAutoPtr<IEventHandle> handle; + { + auto predicate = [&](const auto& ev) { return ev.TabletId == tc.TabletId; }; + auto ev = tc.Runtime->GrabEdgeEventIf<TEvTabletPipe::TEvClientConnected>(handle, predicate); + UNIT_ASSERT(ev); + } TEvPersQueue::TEvResponse *result; THolder<TEvPersQueue::TEvRequest> request; for (i32 retriesLeft = 2; retriesLeft > 0; --retriesLeft) { diff --git a/ydb/services/persqueue_v1/persqueue_ut.cpp b/ydb/services/persqueue_v1/persqueue_ut.cpp index d696e37ad7b..2b6a3b517db 100644 --- a/ydb/services/persqueue_v1/persqueue_ut.cpp +++ b/ydb/services/persqueue_v1/persqueue_ut.cpp @@ -1167,7 +1167,7 @@ TPersQueueV1TestServer server{{.CheckACL=true, .NodeCount=1}}; } Y_UNIT_TEST(DirectReadCleanCache) { - TPersQueueV1TestServer server; + TPersQueueV1TestServer server{{.NodeCount=1}}; SET_LOCALS; TString topicPath{"/Root/PQ/rt3.dc1--acc--topic2"}; server.Server->AnnoyingClient->CreateTopicNoLegacy(topicPath, 1); |