mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-19 16:20:50 +00:00
Black formatting python test.
This commit is contained in:
parent
ddf2e07fd0
commit
56f3030b17
@ -11,6 +11,7 @@ node = cluster.add_instance(
|
||||
macros={"shard": 0, "replica": 1},
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def start_cluster():
|
||||
try:
|
||||
@ -19,33 +20,42 @@ def start_cluster():
|
||||
finally:
|
||||
cluster.shutdown()
|
||||
|
||||
|
||||
def test_insert_stops_when_disk_full(start_cluster):
|
||||
min_free_bytes = 3 * 1024 * 1024 # 3 MiB
|
||||
|
||||
node.query(f"""
|
||||
node.query(
|
||||
f"""
|
||||
CREATE TABLE test_table (
|
||||
id UInt32,
|
||||
data String
|
||||
) ENGINE = MergeTree()
|
||||
ORDER BY id
|
||||
SETTINGS storage_policy = 'only_disk1', min_free_disk_bytes_to_throw_insert = {min_free_bytes}
|
||||
""")
|
||||
"""
|
||||
)
|
||||
|
||||
count = 0
|
||||
|
||||
# Insert data to fill up disk
|
||||
try:
|
||||
for _ in range(100000):
|
||||
node.query("INSERT INTO test_table SELECT number, repeat('a', 1000 * 1000) FROM numbers(1)")
|
||||
node.query(
|
||||
"INSERT INTO test_table SELECT number, repeat('a', 1000 * 1000) FROM numbers(1)"
|
||||
)
|
||||
count += 1
|
||||
except QueryRuntimeException as e:
|
||||
assert "Could not perform insert" in str(e)
|
||||
assert "free bytes in disk space" in str(e)
|
||||
|
||||
free_space = int(node.query("SELECT free_space FROM system.disks WHERE name = 'disk1'").strip())
|
||||
assert free_space <= min_free_bytes, f"Free space ({free_space}) is less than min_free_bytes ({min_free_bytes})"
|
||||
free_space = int(
|
||||
node.query("SELECT free_space FROM system.disks WHERE name = 'disk1'").strip()
|
||||
)
|
||||
assert (
|
||||
free_space <= min_free_bytes
|
||||
), f"Free space ({free_space}) is less than min_free_bytes ({min_free_bytes})"
|
||||
|
||||
rows = int(node.query("SELECT count() from test_table").strip())
|
||||
assert rows == count
|
||||
|
||||
node.query("DROP TABLE test_table")
|
||||
node.query("DROP TABLE test_table")
|
||||
|
Loading…
Reference in New Issue
Block a user