From 47c7d2282593e3d6200124b70eb96d5d386682b0 Mon Sep 17 00:00:00 2001
From: robot-clickhouse <robot-clickhouse@users.noreply.github.com>
Date: Fri, 16 Aug 2024 18:12:24 +0000
Subject: [PATCH] Backport #68409 to 24.8: Try to fix test
 03221_mutation_analyzer_skip_part

---
 .../03221_mutate_profile_events.sh            | 53 +++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100755 tests/queries/0_stateless/03221_mutate_profile_events.sh

diff --git a/tests/queries/0_stateless/03221_mutate_profile_events.sh b/tests/queries/0_stateless/03221_mutate_profile_events.sh
new file mode 100755
index 00000000000..3758db905e0
--- /dev/null
+++ b/tests/queries/0_stateless/03221_mutate_profile_events.sh
@@ -0,0 +1,53 @@
+#!/usr/bin/env bash
+# Tags: no-random-settings, no-random-merge-tree-settings
+
+CUR_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+# shellcheck source=../shell_config.sh
+. "$CUR_DIR"/../shell_config.sh
+
+${CLICKHOUSE_CLIENT} --query "
+    DROP TABLE IF EXISTS t_mutate_profile_events;
+
+    CREATE TABLE t_mutate_profile_events (key UInt64, id UInt64, v1 UInt64, v2 UInt64)
+    ENGINE = MergeTree ORDER BY id PARTITION BY key
+    SETTINGS min_bytes_for_wide_part = 0;
+
+    INSERT INTO t_mutate_profile_events SELECT 1, number, number, number FROM numbers(10000);
+    INSERT INTO t_mutate_profile_events SELECT 2, number, number, number FROM numbers(10000);
+
+    SET mutations_sync = 2;
+
+    ALTER TABLE t_mutate_profile_events UPDATE v1 = 1000 WHERE key = 1;
+    ALTER TABLE t_mutate_profile_events DELETE WHERE key = 2 AND v2 % 10 = 0;
+"
+
+# Mutation query may return before the entry is added to part log.
+# So, we may have to retry the flush of logs until all entries are actually flushed.
+for _ in {1..10}; do
+    ${CLICKHOUSE_CLIENT} --query "SYSTEM FLUSH LOGS"
+    res=$(${CLICKHOUSE_CLIENT} --query "SELECT count() FROM system.part_log WHERE database = currentDatabase() AND table = 't_mutate_profile_events' AND event_type = 'MutatePart'")
+
+    if [[ $res -eq 4 ]]; then
+        break
+    fi
+
+    sleep 2.0
+done
+
+${CLICKHOUSE_CLIENT} --query "
+    SELECT
+        splitByChar('_', part_name)[-1] AS version,
+        sum(ProfileEvents['MutationTotalParts']),
+        sum(ProfileEvents['MutationUntouchedParts']),
+        sum(ProfileEvents['MutatedRows']),
+        sum(ProfileEvents['MutatedUncompressedBytes']),
+        sum(ProfileEvents['MutationAllPartColumns']),
+        sum(ProfileEvents['MutationSomePartColumns']),
+        sum(ProfileEvents['MutationTotalMilliseconds']) > 0,
+        sum(ProfileEvents['MutationExecuteMilliseconds']) > 0,
+    FROM system.part_log
+    WHERE database = currentDatabase() AND table = 't_mutate_profile_events' AND event_type = 'MutatePart'
+    GROUP BY version ORDER BY version;
+
+    DROP TABLE IF EXISTS t_mutate_profile_events;
+"