Browse Source

fix unit test: KIKIMR-19028

gvit 1 year ago
parent
commit
b0b14da8be

+ 1 - 1
ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp

@@ -85,7 +85,7 @@ void InsertAllColumnsAndCheckSelectAll(TKikimrRunner* runner) {
     UNIT_ASSERT(*parser.ColumnParser(17).GetOptionalJsonDocument().Get() == TString("[14]"));
     UNIT_ASSERT(*parser.ColumnParser(18).GetOptionalDyNumber().Get() == TString(".1515e2"));
     streamPart = it.ReadNext().GetValueSync();
-    UNIT_ASSERT_C(streamPart.EOS(), streamPart.GetIssues().ToString());
+    UNIT_ASSERT_C(streamPart.EOS() || streamPart.HasQueryStats(), streamPart.GetIssues().ToString());
 }
 
 }

+ 1 - 1
ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp

@@ -91,7 +91,7 @@ void InsertAllColumnsAndCheckSelectAll(TKikimrRunner* runner) {
     UNIT_ASSERT(*parser.ColumnParser(17).GetOptionalJsonDocument().Get() == TString("[14]"));
     UNIT_ASSERT(*parser.ColumnParser(18).GetOptionalDyNumber().Get() == TString(".1515e2"));
     streamPart = it.ReadNext().GetValueSync();
-    UNIT_ASSERT_C(streamPart.EOS(), streamPart.GetIssues().ToString());
+    UNIT_ASSERT_C(streamPart.EOS() || streamPart.HasQueryStats(), streamPart.GetIssues().ToString());
 }
 
 }