mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-05 05:52:05 +00:00
b2535d7f50
Signed-off-by: Azat Khuzhin <a.khuzhin@semrush.com>
48 lines
1.4 KiB
Python
Executable File
48 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Tags: no-replicated-database, no-parallel, no-fasttest
|
|
|
|
import os
|
|
import sys
|
|
import signal
|
|
|
|
CURDIR = os.path.dirname(os.path.realpath(__file__))
|
|
sys.path.insert(0, os.path.join(CURDIR, "helpers"))
|
|
|
|
from client import client, prompt, end_of_block
|
|
|
|
log = None
|
|
# uncomment the line below for debugging
|
|
# log=sys.stdout
|
|
|
|
with client(
|
|
name="client>",
|
|
log=log,
|
|
command=os.environ.get("CLICKHOUSE_BINARY", "clickhouse")
|
|
+ " client --storage_file_read_method=pread",
|
|
) as client1:
|
|
filename = os.environ["CLICKHOUSE_TMP"] + "/infile_progress.tsv"
|
|
|
|
client1.expect(prompt)
|
|
client1.send("DROP TABLE IF EXISTS test.infile_progress")
|
|
client1.expect(prompt)
|
|
client1.send(f"SELECT number FROM numbers(5) INTO OUTFILE '{filename}'")
|
|
client1.expect(prompt)
|
|
client1.send(
|
|
"CREATE TABLE test.infile_progress (a Int32) Engine=MergeTree order by tuple()"
|
|
)
|
|
client1.expect(prompt)
|
|
client1.send(f"INSERT INTO test.infile_progress FROM INFILE '{filename}'")
|
|
client1.expect("Progress: 5.00 rows, 10.00 B.*\\)")
|
|
client1.expect(prompt)
|
|
|
|
# send Ctrl-C
|
|
client1.send("\x03", eol="")
|
|
match = client1.expect("(%s)|([#\\$] )" % prompt)
|
|
if match.groups()[1]:
|
|
client1.send(client1.command)
|
|
client1.expect(prompt)
|
|
client1.send("DROP TABLE test.infile_progress")
|
|
client1.expect(prompt)
|
|
|
|
os.remove(filename)
|