Source multishard repro

1 files changed, 28 insertions, 0 deletions

diff --git a/ydb/core/kqp/ut/perf/kqp_query_perf_ut.cpp b/ydb/core/kqp/ut/perf/kqp_query_perf_ut.cpp
index bf4f7d5e39..a34b71bff0 100644
--- a/ydb/core/kqp/ut/perf/kqp_query_perf_ut.cpp
+++ b/ydb/core/kqp/ut/perf/kqp_query_perf_ut.cpp
@@ -777,6 +777,34 @@ Y_UNIT_TEST_SUITE(KqpQueryPerf) {
             .ExpectedDeletes = 3,
         });
     }
+
+    Y_UNIT_TEST_TWIN(MultiRead, SourceRead) {
+        TKikimrSettings settings;
+        NKikimrConfig::TAppConfig appConfig;
+        appConfig.MutableTableServiceConfig()->SetEnableKqpDataQuerySourceRead(SourceRead);
+        settings.SetAppConfig(appConfig);
+
+        TKikimrRunner kikimr(settings);
+        auto db = kikimr.GetTableClient();
+        auto session = db.CreateSession().GetValueSync().GetSession();
+
+        {
+            auto settings = NYdb::NTable::TExecDataQuerySettings().CollectQueryStats(ECollectQueryStatsMode::Full);
+            auto result = session.ExecuteDataQuery(R"(
+                SELECT * FROM `/Root/KeyValueLargePartition` WHERE Key > 101;
+                SELECT * FROM `/Root/KeyValueLargePartition` Where Key < 201;
+            )", TTxControl::BeginTx().CommitTx(), settings).GetValueSync();
+            UNIT_ASSERT_C(result.IsSuccess(), result.GetIssues().ToString());
+            size_t partitionsCount = 0;
+            auto& stats = NYdb::TProtoAccessor::GetProto(*result.GetStats());
+            for (auto& phase : stats.query_phases()) {
+                for (auto& read : phase.table_access()) {
+                    partitionsCount += read.partitions_count();
+                }
+            }
+            UNIT_ASSERT_VALUES_EQUAL(partitionsCount, SourceRead ? 2 : 1);
+        }
+    }
 }
 
 } // namespace NKikimr::NKqp