aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorgvit <gvit@ydb.tech>2023-08-18 17:33:56 +0300
committergvit <gvit@ydb.tech>2023-08-18 19:51:18 +0300
commitb0b14da8be040c1b5e4285c500f646c2ab901b26 (patch)
treeed569744c74f19590fc8cf662b797487f6068ff6
parent73215359bc33e76f5b94d1832a377072bf245cfc (diff)
downloadydb-b0b14da8be040c1b5e4285c500f646c2ab901b26.tar.gz
fix unit test: KIKIMR-19028
-rw-r--r--ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp2
-rw-r--r--ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp2
2 files changed, 2 insertions, 2 deletions
diff --git a/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp b/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp
index 4209f59e66c..51e2bfbbb7b 100644
--- a/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp
+++ b/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());
}
}
diff --git a/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp b/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp
index fd834e6874b..d5abf13d62e 100644
--- a/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp
+++ b/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());
}
}