Skip to content

Commit 82f1d86

Browse files
authored
Merge 61f1392 into c3cc29f
2 parents c3cc29f + 61f1392 commit 82f1d86

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

ydb/core/testlib/test_pq_client.h

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -500,9 +500,18 @@ class TFlatMsgBusPQClient : public NFlatTests::TFlatMsgBusClient {
500500
public:
501501
void RunYqlSchemeQuery(TString query, bool expectSuccess = true) {
502502
auto tableClient = NYdb::NTable::TTableClient(*Driver);
503-
auto result = tableClient.RetryOperationSync([&](NYdb::NTable::TSession session) {
504-
return session.ExecuteSchemeQuery(query).GetValueSync();
505-
});
503+
504+
NYdb::TStatus result(NYdb::EStatus::SUCCESS, NYql::TIssues());
505+
for (size_t i = 0; i < 10; ++i) {
506+
result = tableClient.RetryOperationSync([&](NYdb::NTable::TSession session) {
507+
return session.ExecuteSchemeQuery(query).GetValueSync();
508+
});
509+
if (!expectSuccess || result.IsSuccess()) {
510+
break;
511+
}
512+
Sleep(TDuration::Seconds(1));
513+
}
514+
506515
if (expectSuccess) {
507516
UNIT_ASSERT_C(result.IsSuccess(), result.GetIssues().ToString());
508517
} else {

0 commit comments

Comments
 (0)