Checking that total rows is OK

Author: Enmk

ut/client_ut.cpp

@@ -118,9 +118,10 @@ TEST_P(ClientCase, LowCardinality) {
     block.AppendColumn("lc", lc);
     client_->Insert("test_clickhouse_cpp.low_cardinality", block);
 
+    size_t total_rows = 0;
     client_->Select("SELECT lc FROM test_clickhouse_cpp.low_cardinality",
-        [&data](const Block& block)
-        {
+        [&total_rows, &data](const Block& block) {
+            total_rows += block.GetRowCount();
             if (block.GetRowCount() == 0) {
                 return;
             }
@@ -134,6 +135,8 @@ TEST_P(ClientCase, LowCardinality) {
             }
         }
     );
+
+    ASSERT_EQ(total_rows, data.size());
 }
 
 TEST_P(ClientCase, Generic) {

ut/performance_tests.cpp

@@ -197,10 +197,12 @@ TYPED_TEST_P(ColumnPerformanceTest, InsertAndSelect) {
     }
 
     {
+        size_t total_rows = 0;
         Timer timer;
         Timer::DurationType inner_loop_duration{0};
-        client.Select("SELECT " + column_name +  " FROM " + table_name, [&inner_loop_duration](const Block & block) {
+        client.Select("SELECT " + column_name +  " FROM " + table_name, [&total_rows, &inner_loop_duration](const Block & block) {
             Timer timer;
+            total_rows += block.GetRowCount();
             if (block.GetRowCount() == 0) {
                 return;
             }
@@ -214,6 +216,8 @@ TYPED_TEST_P(ColumnPerformanceTest, InsertAndSelect) {
 
         auto elapsed = timer.Elapsed() - inner_loop_duration;
         std::cerr << "SELECT:\t" << elapsed << std::endl;
+
+        ASSERT_EQ(total_rows, ITEMS_COUNT);
     }
 }