From b79d3d521ce2f0c38787021a18a5f6f82abaae12 Mon Sep 17 00:00:00 2001
From: Alexey Milovidov <milovidov@clickhouse.com>
Date: Sun, 23 Jun 2024 17:36:27 +0200
Subject: [PATCH] Improve flaky test to provide more diagnostics

---
 .../02210_processors_profile_log.reference    | 30 -------------------
 .../02210_processors_profile_log.sql          |  7 ++---
 2 files changed, 3 insertions(+), 34 deletions(-)

diff --git a/tests/queries/0_stateless/02210_processors_profile_log.reference b/tests/queries/0_stateless/02210_processors_profile_log.reference
index 41543d0706ae..035bd9897ad3 100644
--- a/tests/queries/0_stateless/02210_processors_profile_log.reference
+++ b/tests/queries/0_stateless/02210_processors_profile_log.reference
@@ -1,38 +1,8 @@
--- { echo }
-EXPLAIN PIPELINE SELECT sleep(1);
 (Expression)
 ExpressionTransform
   (ReadFromStorage)
   SourceFromSingleChunk 0 → 1
-SELECT sleep(1) SETTINGS log_processors_profiles=true, log_queries=1, log_queries_min_type='QUERY_FINISH';
 0
-SYSTEM FLUSH LOGS;
-WITH
-    (
-        SELECT query_id
-        FROM system.query_log
-        WHERE current_database = currentDatabase() AND Settings['log_processors_profiles']='1'
-    ) AS query_id_
-SELECT
-    name,
-    multiIf(
-        -- ExpressionTransform executes sleep(),
-        -- so IProcessor::work() will spend 1 sec.
-        name = 'ExpressionTransform', elapsed_us>=1e6,
-        -- SourceFromSingleChunk, that feed data to ExpressionTransform,
-        -- will feed first block and then wait in PortFull.
-        name = 'SourceFromSingleChunk', output_wait_elapsed_us>=1e6,
-        -- NullSource/LazyOutputFormatLazyOutputFormat are the outputs
-        -- so they cannot starts to execute before sleep(1) will be executed.
-        input_wait_elapsed_us>=1e6)
-    elapsed,
-    input_rows,
-    input_bytes,
-    output_rows,
-    output_bytes
-FROM system.processors_profile_log
-WHERE query_id = query_id_
-ORDER BY name;
 ExpressionTransform	1	1	1	1	1
 LazyOutputFormat	1	1	1	0	0
 LimitsCheckingTransform	1	1	1	1	1
diff --git a/tests/queries/0_stateless/02210_processors_profile_log.sql b/tests/queries/0_stateless/02210_processors_profile_log.sql
index a15ed26fd677..59edbb714576 100644
--- a/tests/queries/0_stateless/02210_processors_profile_log.sql
+++ b/tests/queries/0_stateless/02210_processors_profile_log.sql
@@ -1,4 +1,3 @@
--- { echo }
 EXPLAIN PIPELINE SELECT sleep(1);
 
 SELECT sleep(1) SETTINGS log_processors_profiles=true, log_queries=1, log_queries_min_type='QUERY_FINISH';
@@ -15,13 +14,13 @@ SELECT
     multiIf(
         -- ExpressionTransform executes sleep(),
         -- so IProcessor::work() will spend 1 sec.
-        name = 'ExpressionTransform', elapsed_us>=1e6,
+        name = 'ExpressionTransform', elapsed_us>=1e6 ? 1 : elapsed_us,
         -- SourceFromSingleChunk, that feed data to ExpressionTransform,
         -- will feed first block and then wait in PortFull.
-        name = 'SourceFromSingleChunk', output_wait_elapsed_us>=1e6,
+        name = 'SourceFromSingleChunk', output_wait_elapsed_us>=1e6 ? 1 : output_wait_elapsed_us,
         -- NullSource/LazyOutputFormatLazyOutputFormat are the outputs
         -- so they cannot starts to execute before sleep(1) will be executed.
-        input_wait_elapsed_us>=1e6)
+        input_wait_elapsed_us>=1e6 ? 1 : input_wait_elapsed_us)
     elapsed,
     input_rows,
     input_bytes,