mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-28 02:21:59 +00:00
02b5d9a56d
In case of stdin is not attached (EOF), then expect will also close stdout, fix this by openning stdout again. Signed-off-by: Azat Khuzhin <a.khuzhin@semrush.com>
141 lines
3.4 KiB
Bash
Executable File
141 lines
3.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Tags: long
|
|
|
|
CURDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
# shellcheck source=../shell_config.sh
|
|
. "$CURDIR"/../shell_config.sh
|
|
|
|
SCRIPT_PATH="$CURDIR/$CLICKHOUSE_TEST_UNIQUE_NAME.generated-expect"
|
|
|
|
# NOTE: database = $CLICKHOUSE_DATABASE is superfluous
|
|
|
|
function test_completion_word()
|
|
{
|
|
local w=$1 && shift
|
|
|
|
local w_len=${#w}
|
|
local compword_begin=${w:0:$((w_len-3))}
|
|
local compword_end=${w:$((w_len-3))}
|
|
|
|
# NOTE:
|
|
# - here and below you should escape variables of the expect.
|
|
# - you should not use "expect <<..." since in this case timeout/eof will
|
|
# not work (I guess due to attached stdin)
|
|
cat > "$SCRIPT_PATH" << EOF
|
|
# NOTE: log will be appended
|
|
exp_internal -f $CLICKHOUSE_TMP/$(basename "${BASH_SOURCE[0]}").debuglog 0
|
|
|
|
# NOTE: when expect have EOF on stdin it also closes stdout, so let's reopen it
|
|
# again for logging
|
|
set stdout_channel [open "/dev/stdout" w]
|
|
|
|
log_user 0
|
|
set timeout 60
|
|
match_max 100000
|
|
expect_after {
|
|
# Do not ignore eof from expect
|
|
eof { exp_continue }
|
|
# A default timeout action is to do nothing, change it to fail
|
|
timeout { exit 1 }
|
|
}
|
|
|
|
spawn bash -c "$*"
|
|
expect ":) "
|
|
|
|
# Make a query
|
|
send -- "SET $compword_begin"
|
|
expect "SET $compword_begin"
|
|
|
|
# Wait for suggestions to load, they are loaded in background
|
|
set is_done 0
|
|
set timeout 1
|
|
while {\$is_done == 0} {
|
|
send -- "\\t"
|
|
expect {
|
|
"$compword_begin$compword_end" {
|
|
puts \$stdout_channel "$compword_begin$compword_end: OK"
|
|
set is_done 1
|
|
}
|
|
default {
|
|
sleep 1
|
|
}
|
|
}
|
|
}
|
|
|
|
close \$stdout_channel
|
|
|
|
send -- "\\3\\4"
|
|
expect eof
|
|
EOF
|
|
|
|
# NOTE: run expect under timeout since there is while loop that is not
|
|
# limited with timeout.
|
|
#
|
|
# NOTE: cat is required to serialize stdout for expect (without this pipe
|
|
# it will reopen the file again, and the output will be mixed).
|
|
timeout 2m expect -f "$SCRIPT_PATH" | cat
|
|
}
|
|
|
|
# last 3 bytes will be completed,
|
|
# so take this in mind when you will update the list.
|
|
client_compwords_positive=(
|
|
# system.functions
|
|
concatAssumeInjective
|
|
# system.table_engines
|
|
ReplacingMergeTree
|
|
# system.formats
|
|
JSONEachRow
|
|
# system.table_functions
|
|
clusterAllReplicas
|
|
# system.data_type_families
|
|
SimpleAggregateFunction
|
|
# system.merge_tree_settings
|
|
write_ahead_log_interval_ms_to_fsync
|
|
# system.settings
|
|
max_concurrent_queries_for_all_users
|
|
# system.clusters
|
|
test_shard_localhost
|
|
# system.macros
|
|
default_path_test
|
|
# system.storage_policies, egh not uniq
|
|
default
|
|
# system.aggregate_function_combinators
|
|
uniqCombined64ForEach
|
|
|
|
# FIXME: one may add separate case for suggestion_limit
|
|
# system.databases
|
|
system
|
|
# system.tables
|
|
aggregate_function_combinators
|
|
# system.columns
|
|
primary_key_bytes_in_memory_allocated
|
|
# system.dictionaries
|
|
# FIXME: none
|
|
)
|
|
|
|
local_compwords_positive=(
|
|
# system.functions
|
|
concatAssumeInjective
|
|
# system.table_engines
|
|
ReplacingMergeTree
|
|
# system.formats
|
|
JSONEachRow
|
|
# system.table_functions
|
|
clusterAllReplicas
|
|
# system.data_type_families
|
|
SimpleAggregateFunction
|
|
)
|
|
|
|
echo "# clickhouse-client"
|
|
for w in "${client_compwords_positive[@]}"; do
|
|
test_completion_word "$w" "$CLICKHOUSE_CLIENT"
|
|
done
|
|
echo "# clickhouse-local"
|
|
for w in "${local_compwords_positive[@]}"; do
|
|
test_completion_word "$w" "$CLICKHOUSE_LOCAL"
|
|
done
|
|
|
|
rm -f "${SCRIPT_PATH:?}"
|
|
|
|
exit 0
|