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 "DESC file('${USER_FILES_PATH}/$1')"
    $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"

    extension_without_dot=$(echo $1 | sed -e 's/\.//g')
    FILE_PREFIX="02661_read_from_archive_${CLICKHOUSE_DATABASE}_$extension_without_dot"

    touch ${FILE_PREFIX}_data0.csv
    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}_data0.csv ${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_EXTRACT_TABLE_STRUCTURE" && 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_EXTRACT_TABLE_STRUCTURE" && echo "OK" || echo "FAIL"

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

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

# vi: ft=bash