valid query

Author: stanislav-shchetinin

tests/integration/bulk_upsert_simple_it/bulk_upsert.cpp

@@ -1,7 +1,5 @@
 #include "bulk_upsert.h"
 
-#include <src/library/getopt/last_getopt.h>
-
 #include <filesystem>
 
 static constexpr size_t BATCH_SIZE = 1000;
@@ -144,6 +142,11 @@ TStatistic Select(TTableClient& client, const std::string& path) {
     uint64_t sumHost = 0;
     uint64_t rowCount = 0;
 
+    if (parser.ColumnsCount() != 3 || parser.RowsCount() != 1) {
+        throw TYdbErrorException(TStatus(EStatus::GENERIC_ERROR,
+        {NYql::TIssue("The number of columns should be: 3.\nThe number of rows should be: 1")}));
+    }
+
     if (parser.TryNextRow()) {
         sumApp = *parser.ColumnParser("column0").GetOptionalInt64();
         sumHost = *parser.ColumnParser("column1").GetOptionalInt64();

tests/integration/bulk_upsert_simple_it/bulk_upsert.h

@@ -2,6 +2,7 @@
 
 #include <ydb-cpp-sdk/client/driver/driver.h>
 #include <ydb-cpp-sdk/client/table/table.h>
+#include <src/library/getopt/last_getopt.h>
 
 using namespace NYdb;
 using namespace NYdb::NTable;

tests/integration/bulk_upsert_simple_it/main.cpp

@@ -36,11 +36,15 @@ TEST(Integration, BulkUpsert) {
         }
     }
 
-    auto [sumApp, sumHost, rowCount] = Select(client, path);
-
-    EXPECT_EQ(rowCount, correctRowCount);
-    EXPECT_EQ(sumApp, correctSumApp);
-    EXPECT_EQ(sumHost, correctSumHost);
+    try {
+        auto [sumApp, sumHost, rowCount] = Select(client, path);
+        EXPECT_EQ(rowCount, correctRowCount);
+        EXPECT_EQ(sumApp, correctSumApp);
+        EXPECT_EQ(sumHost, correctSumHost);
+    } catch (const TYdbErrorException& e) {
+        driver.Stop(true);
+        FAIL() << "Execution failed due to fatal error:\nStatus: " << ToString(e.Status.GetStatus()) << std::endl << e.Status.GetIssues().ToString();
+    }
 
     DropTable(client, path);
     driver.Stop(true);