From 20cbca87de1e4068113205c3641ceb90178009db Mon Sep 17 00:00:00 2001
From: Azat Khuzhin <a3at.mail@gmail.com>
Date: Thu, 8 Jul 2021 00:48:15 +0300
Subject: [PATCH] Fix 01791_dist_INSERT_block_structure_mismatch flakiness

Add SYSTEM STOP DISTRIBUTED SENDS to force messages from SYSTEM FLUSH
DISTRIBUTED query context.
---
 .../01791_dist_INSERT_block_structure_mismatch.reference        | 2 ++
 .../0_stateless/01791_dist_INSERT_block_structure_mismatch.sh   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.reference b/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.reference
index 3bba1ac23c0..9f376fb3e4f 100644
--- a/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.reference
+++ b/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.reference
@@ -1,5 +1,7 @@
 <Warning> DistributedBlockOutputStream: Structure does not match (remote: n Int8 Int8(size = 0), local: n UInt64 UInt64(size = 1)), implicit conversion will be done.
 <Warning> DistributedBlockOutputStream: Structure does not match (remote: n Int8 Int8(size = 0), local: n UInt64 UInt64(size = 1)), implicit conversion will be done.
+<Warning> default.dist_01683.DirectoryMonitor: Structure does not match (remote: n Int8 Int8(size = 0), local: n UInt64 UInt64(size = 0)), implicit conversion will be done
+<Warning> default.dist_01683.DirectoryMonitor: Structure does not match (remote: n Int8 Int8(size = 0), local: n UInt64 UInt64(size = 0)), implicit conversion will be done
 1
 1
 2
diff --git a/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.sh b/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.sh
index e989696da03..1a96aad3f13 100755
--- a/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.sh
+++ b/tests/queries/0_stateless/01791_dist_INSERT_block_structure_mismatch.sh
@@ -18,6 +18,8 @@ $CLICKHOUSE_CLIENT --prefer_localhost_replica=0 -nm -q "
     INSERT INTO dist_01683 VALUES (1),(2);
 
     SET insert_distributed_sync=0;
+    -- force log messages from the 'SYSTEM FLUSH DISTRIBUTED' context
+    SYSTEM STOP DISTRIBUTED SENDS dist_01683;
     INSERT INTO dist_01683 VALUES (1),(2);
     SYSTEM FLUSH DISTRIBUTED dist_01683;