tests: sync lib files with private

Author: azat

tests/queries/0_stateless/mergetree_mutations.lib

@@ -7,19 +7,16 @@ function wait_for_mutation()
     local database=$3
     database=${database:="${CLICKHOUSE_DATABASE}"}
 
-    local i
-    for i in {1..100}
+    for _ in {1..300}
     do
         sleep 0.1
         if [[ $(${CLICKHOUSE_CLIENT} --query="SELECT min(is_done) FROM system.mutations WHERE database='$database' AND table='$table' AND mutation_id='$mutation_id'") -eq 1 ]]; then
-            break
+            return
         fi
-
-        if [[ $i -eq 100 ]]; then
-            echo "Timed out while waiting for mutation to execute!"
-        fi
-
     done
+
+    echo "Timed out while waiting for mutation to execute!"
+    ${CLICKHOUSE_CLIENT} -q "SELECT * FROM system.mutations WHERE database='$database' AND table like '$table' AND mutation_id='$mutation_id' AND is_done=0"
 }
 
 function wait_for_all_mutations()
@@ -28,20 +25,16 @@ function wait_for_all_mutations()
     local database=$2
     database=${database:="${CLICKHOUSE_DATABASE}"}
 
-    local i
-    for i in {1..200}
+    for _ in {1..600}
     do
         if [[ $(${CLICKHOUSE_CLIENT} --query="SELECT coalesce(minOrNull(is_done), 1) FROM system.mutations WHERE database='$database' AND table like '$table'") -eq 1 ]]; then
-            break
+            return
         fi
-
-        if [[ $i -eq 200 ]]; then
-            echo "Timed out while waiting for mutation to execute!"
-            ${CLICKHOUSE_CLIENT} -q "SELECT * FROM system.mutations WHERE database='$database' AND table like '$table' AND is_done=0"
-        fi
-
         sleep 0.3
     done
+
+    echo "Timed out while waiting for mutation to execute!"
+    ${CLICKHOUSE_CLIENT} -q "SELECT * FROM system.mutations WHERE database='$database' AND table like '$table' AND is_done=0"
 }
 
 # vi: ft=bash

tests/queries/0_stateless/replication.lib

@@ -17,7 +17,7 @@ function try_sync_replicas()
             WHERE (database = currentDatabase()) AND (table LIKE '$table_name_prefix%')
         ))")
     local tables_arr
-    readarray -t tables_arr < <(${CLICKHOUSE_CLIENT} -q "SELECT name FROM system.tables WHERE database=currentDatabase() AND name like '$table_name_prefix%' AND engine like '%Replicated%'")
+    readarray -t tables_arr < <(${CLICKHOUSE_CLIENT} -q "SELECT name FROM system.tables WHERE database=currentDatabase() AND name like '$table_name_prefix%' AND (engine like '%Replicated%' or engine like '%Shared%')")
 
     local t
     for t in "${tables_arr[@]}"