Add QueryService coverage for kqp_perf_ut.(KIKIMR-16294) (#721)

Co-authored-by: Sergey Puchin <[email protected]>

Author: spuchin

ydb/core/kqp/ut/common/kqp_ut_common.cpp

@@ -542,9 +542,9 @@ void PrintQueryStats(const TDataQueryResult& result) {
     }
 }
 
-void AssertTableStats(const TDataQueryResult& result, TStringBuf table, const TExpectedTableStats& expectedStats) {
-    auto stats = NYdb::TProtoAccessor::GetProto(*result.GetStats());
-
+void AssertTableStats(const Ydb::TableStats::QueryStats& stats, TStringBuf table,
+    const TExpectedTableStats& expectedStats)
+{
     ui64 actualReads = 0;
     ui64 actualUpdates = 0;
     ui64 actualDeletes = 0;
@@ -575,6 +575,11 @@ void AssertTableStats(const TDataQueryResult& result, TStringBuf table, const TE
     }
 }
 
+void AssertTableStats(const TDataQueryResult& result, TStringBuf table, const TExpectedTableStats& expectedStats) {
+    auto stats = NYdb::TProtoAccessor::GetProto(*result.GetStats());
+    return AssertTableStats(stats, table, expectedStats);
+}
+
 TDataQueryResult ExecQueryAndTestResult(TSession& session, const TString& query, const NYdb::TParams& params,
     const TString& expectedYson)
 {

ydb/core/kqp/ut/common/kqp_ut_common.h

@@ -268,6 +268,12 @@ struct TExpectedTableStats {
     TMaybe<ui64> ExpectedDeletes;
 };
 
+void AssertTableStats(const Ydb::TableStats::QueryStats& stats, TStringBuf table,
+    const TExpectedTableStats& expectedStats);
+
+void AssertTableStats(const NYdb::NTable::TDataQueryResult& result, TStringBuf table,
+    const TExpectedTableStats& expectedStats);
+
 void AssertTableStats(const NYdb::NTable::TDataQueryResult& result, TStringBuf table,
     const TExpectedTableStats& expectedStats);