fix 03221_insert_timeout_overflow_mode flaky

Author: CheSema

tests/queries/0_stateless/03221_insert_timeout_overflow_mode.reference

@@ -1,5 +1,4 @@
-TIMEOUT_EXCEEDED
-TIMEOUT_EXCEEDED
+timeout_overflow_mode throw OK
 QUERY_WAS_CANCELLED
 QUERY_WAS_CANCELLED
 QUERY_WAS_CANCELLED

tests/queries/0_stateless/03221_insert_timeout_overflow_mode.sh

@@ -6,10 +6,32 @@ CUR_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
 
 ${CLICKHOUSE_CLIENT} --query "create table dst (number UInt64) engine = MergeTree ORDER BY number;"
 
-${CLICKHOUSE_CLIENT} --query "select number + sleep(0.1) as number from system.numbers limit 10 settings max_block_size = 1 format Native" 2>/dev/null | ${CLICKHOUSE_CLIENT} --max_execution_time 0.3 --timeout_overflow_mode throw --query "insert into dst format Native" 2>&1 | grep -o "TIMEOUT_EXCEEDED"
+error=$(${CLICKHOUSE_CLIENT} --query "select number + sleep(0.1) as number from system.numbers limit 10 settings max_block_size = 1 format Native" 2>/dev/null | ${CLICKHOUSE_CLIENT} --max_execution_time 0.3 --timeout_overflow_mode throw --query "insert into dst format Native" 2>&1 ||:)
+
+function check_error()
+{
+    error="$1"
+
+    if echo "$error" | grep "TIMEOUT_EXCEEDED" > /dev/null
+    then
+        echo "timeout_overflow_mode throw OK"
+        return 0
+    fi
+
+    if echo "$error" | grep "QUERY_WAS_CANCELLED" | grep "is killed in pending state" > /dev/null
+    then
+        echo "timeout_overflow_mode throw OK"
+        return 0
+    fi
+
+    echo "Query ended in unexpected way:"
+    echo "$error"
+    return 1
+}
+
+check_error "$error"
 
 # this is the tests with PushingPipelineExecutor
 ${CLICKHOUSE_CLIENT} --query "select number + sleep(0.1) as number from system.numbers limit 10 settings max_block_size = 1 format Native" 2>/dev/null | ${CLICKHOUSE_CLIENT} --max_threads=1 --max_execution_time 0.3 --timeout_overflow_mode break --query "insert into dst format Native" 2>&1 | grep -o "QUERY_WAS_CANCELLED"
-
 # this is the tests with PushingPipelineExecutorPushingAsyncPipelineExecutor
 ${CLICKHOUSE_CLIENT} --query "select number + sleep(0.1) as number from system.numbers limit 10 settings max_block_size = 1 format Native" 2>/dev/null | ${CLICKHOUSE_CLIENT} --max_threads=3 --max_execution_time 0.3 --timeout_overflow_mode break --query "insert into dst format Native" 2>&1 | grep -o "QUERY_WAS_CANCELLED"