ClickHouse/tests/queries/0_stateless/02661_read_from_archive.lib

#!/usr/bin/env bash
# Tags: no-fasttest

CUR_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
# shellcheck source=../shell_config.sh
. "$CUR_DIR"/../shell_config.sh

function read_archive_file() {
    $CLICKHOUSE_LOCAL --query "SELECT * FROM file('${user_files_path}/$1') ORDER BY 1, 2"
    $CLICKHOUSE_CLIENT --query "SELECT * FROM file('${user_files_path}/$1') ORDER BY 1, 2"
    $CLICKHOUSE_CLIENT --query "CREATE TABLE 02661_archive_table Engine=File('CSV', '${user_files_path}/$1')"
    $CLICKHOUSE_CLIENT --query "SELECT * FROM 02661_archive_table ORDER BY 1, 2"
    $CLICKHOUSE_CLIENT --query "DROP TABLE 02661_archive_table"
}

function run_archive_test() {
    $CLICKHOUSE_CLIENT --query "DROP TABLE IF EXISTS 02661_archive_table"

    FILE_PREFIX="${CLICKHOUSE_TEST_UNIQUE_NAME}_$1_"

    user_files_path=$(clickhouse-client --query "select _path,_file from file('nonexist.txt', 'CSV', 'val1 char')" 2>&1 | grep Exception | awk '{gsub("/nonexist.txt","",$9); print $9}')

    echo -e "1,2\n3,4" > ${FILE_PREFIX}_data1.csv
    echo -e "5,6\n7,8" > ${FILE_PREFIX}_data2.csv
    echo -e "9,10\n11,12" > ${FILE_PREFIX}_data3.csv

    eval "$2 ${user_files_path}/${FILE_PREFIX}_archive1.$1 ${FILE_PREFIX}_data1.csv ${FILE_PREFIX}_data2.csv > /dev/null"
    eval "$2 ${user_files_path}/${FILE_PREFIX}_archive2.$1 ${FILE_PREFIX}_data1.csv ${FILE_PREFIX}_data3.csv > /dev/null"
    eval "$2 ${user_files_path}/${FILE_PREFIX}_archive3.$1 ${FILE_PREFIX}_data2.csv ${FILE_PREFIX}_data3.csv > /dev/null"

    echo "archive1 data1.csv"
    read_archive_file "${FILE_PREFIX}_archive1.$1 :: ${FILE_PREFIX}_data1.csv"
    echo "archive{1..2} data1.csv"
    read_archive_file "${FILE_PREFIX}_archive{1..2}.$1 :: ${FILE_PREFIX}_data1.csv"
    echo "archive{1,2} data{1,3}.csv"
    read_archive_file "${FILE_PREFIX}_archive{1,2}.$1 :: ${FILE_PREFIX}_data{1,3}.csv"
    echo "archive3 data*.csv"
    read_archive_file "${FILE_PREFIX}_archive3.$1 :: ${FILE_PREFIX}_data*.csv"
    echo "archive* *.csv"
    read_archive_file "${FILE_PREFIX}_archive*.$1 :: *.csv"
    echo "archive* {2..3}.csv"
    read_archive_file "${FILE_PREFIX}_archive*.$1 :: ${FILE_PREFIX}_data{2..3}.csv"

    $CLICKHOUSE_LOCAL --query "SELECT * FROM file('${user_files_path}/${FILE_PREFIX}_archive1.$1::nonexistent.csv')" 2>&1 | grep -q "CANNOT_UNPACK_ARCHIVE" && echo "OK" || echo "FAIL"
    $CLICKHOUSE_LOCAL --query "SELECT * FROM file('${user_files_path}/${FILE_PREFIX}_archive3.$1::{2..3}.csv')" 2>&1 | grep -q "CANNOT_UNPACK_ARCHIVE" && echo "OK" || echo "FAIL"

    rm ${user_files_path}/${FILE_PREFIX}_archive{1..3}.$1

    rm ${FILE_PREFIX}_data{1..3}.csv
}