Test for export parts and export partitions

alter/table/replace_partition/common.py

@@ -9,6 +9,26 @@
 from helpers.tables import create_table_partitioned_by_column, create_table, Column
 
 
+@TestStep(Given)
+def create_partitions_for_collapsing_merge_tree(
+    self,
+    table_name,
+    number_of_values=3,
+    number_of_partitions=5,
+    number_of_parts=1,
+    node=None,
+):
+    """Insert random UInt64 values into a column and create multiple partitions based on the value of number_of_partitions."""
+    if node is None:
+        node = self.context.node
+
+    with By("Inserting random values into a column with uint64 datatype"):
+        for i in range(1, number_of_partitions + 1):
+            for parts in range(1, number_of_parts + 1):
+                node.query(
+                    f"INSERT INTO {table_name} (p, i) SELECT {random.choice([-1, 1])}, rand64() FROM numbers({number_of_values})"
+                )
+
 @TestStep(Given)
 def create_partitions_with_random_uint64(
     self,

engines/tests/replacing_merge_tree/replacing_merge_tree.py

@@ -533,7 +533,7 @@ def incorrect_data_insert_with_disabled_optimize_on_insert(self, node=None):
         ):
             node.query(f"SELECT * FROM {name} FORMAT JSONEachRow;")
 
-            with And("I optimize table"):
+            with By("optimizing a table"):
                 node.query(
                     f"OPTIMIZE TABLE {name} FINAL;",
                     message="DB::Exception:",

helpers/cluster.py

@@ -1012,6 +1012,7 @@ def query(
         ignore_exception=False,
         step=By,
         settings=None,
+        inline_settings=None,
         retry_count=5,
         messages_to_retry=None,
         retry_delay=5,
@@ -1035,6 +1036,7 @@ def query(
         :param no_check: disable exitcode and message checks, default: False
         :param step: wrapping step class, default: By
         :param settings: list of settings to be used for the query in the form [(name, value),...], default: None
+        :param inline_settings: list of inline settings to be used for the query in the form [(name, value),...], default: None
         :param retry_count: number of retries, default: 5
         :param messages_to_retry: list of messages in the query output for
                which retry should be triggered, default: MESSAGES_TO_RETRY
@@ -1057,6 +1059,7 @@ def query(
         retry_count = max(0, int(retry_count))
         retry_delay = max(0, float(retry_delay))
         settings = list(settings or [])
+        inline_settings = list(inline_settings or [])
         query_settings = list(settings)
 
         if raise_on_exception:
@@ -1077,6 +1080,13 @@ def query(
         if query_id is not None:
             query_settings += [("query_id", f"{query_id}")]
 
+        if inline_settings:
+            sql = (
+                "; ".join([f"SET {name} = {value}" for name, value in inline_settings])
+                + "; "
+                + sql
+            )
+
         client = "clickhouse client -n"
         if secure:
             client += (