tests

Author: den818

clickhouse/block.cpp

@@ -71,6 +71,11 @@ const BlockInfo& Block::Info() const {
     return info_;
 }
 
+/// Set block info
+void Block::SetInfo(BlockInfo info) {
+    info_ = std::move(info);
+}
+
 /// Count of rows in the block.
 size_t Block::GetRowCount() const {
     return rows_;

clickhouse/block.h

@@ -73,6 +73,9 @@ class Block {
 
     const BlockInfo& Info() const;
 
+    /// Set block info
+    void SetInfo(BlockInfo info);
+
     /// Count of rows in the block.
     size_t GetRowCount() const;
 

clickhouse/client.cpp

@@ -485,7 +485,7 @@ bool Client::Impl::ReadBlock(InputStream& input, Block* block) {
             return false;
         }
 
-        // TODO use data
+        block->SetInfo(std::move(info));
     }
 
     uint64_t num_columns = 0;

ut/client_ut.cpp

@@ -1010,6 +1010,23 @@ TEST_P(ClientCase, RoundtripArrayTString) {
     EXPECT_TRUE(CompareRecursive(*array, *result_typed));
 }
 
+TEST_P(ClientCase, WriteInfo) {
+    Block block;
+    createTableWithOneColumn<ColumnString>(block);
+
+    std::optional<Progress> received_progress;
+    Query query("INSERT INTO " + table_name + " (*) VALUES (\'Foo\'), (\'Bar\')" );
+    query.OnProgress([&](const Progress& progress) {
+            received_progress = progress;
+        });
+    client_->Execute(query);
+
+    EXPECT_TRUE(received_progress.has_value());
+    // server for some reason sent "rows" instead "written_rows"
+    EXPECT_GT(received_progress->rows + received_progress->written_rows , 0ul);
+    EXPECT_GT(received_progress->bytes + received_progress->written_bytes, 0ul);
+}
+
 const auto LocalHostEndpoint = ClientOptions()
         .SetHost(           getEnvOrDefault("CLICKHOUSE_HOST",     "localhost"))
         .SetPort(   getEnvOrDefault<size_t>("CLICKHOUSE_PORT",     "9000"))