File tree Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Original file line number Diff line number Diff line change @@ -500,9 +500,18 @@ class TFlatMsgBusPQClient : public NFlatTests::TFlatMsgBusClient {
500500public:
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 {
You can’t perform that action at this time.
0 commit comments