mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 08:32:02 +00:00
110 lines
4.6 KiB
Python
110 lines
4.6 KiB
Python
import pytest
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
instance = cluster.add_instance("instance", with_zookeeper=True)
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def start_cluster():
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
@pytest.mark.parametrize("inject_failpoint", [1, 0])
|
|
def test_insert_over_http_exception(start_cluster, inject_failpoint):
|
|
instance.query("DROP TABLE IF EXISTS tt SYNC")
|
|
instance.query(
|
|
"CREATE TABLE tt (KeyID UInt32) Engine = ReplicatedMergeTree('/test_insert_exception_over_http/tt', 'r1') ORDER BY (KeyID)"
|
|
)
|
|
if inject_failpoint > 0:
|
|
instance.query(
|
|
"SYSTEM ENABLE FAILPOINT execute_query_calling_empty_set_result_func_on_exception"
|
|
)
|
|
|
|
log_comment = f"{inject_failpoint}_02988_66a57d6f-d1cc-4693-8bf4-206848edab87"
|
|
assert True == instance.http_query_and_get_error(
|
|
f"insert into tt settings insert_keeper_max_retries=0, insert_keeper_fault_injection_probability=1.0, log_comment='{log_comment}' values (1), (2), (3), (4), (5)",
|
|
method="POST",
|
|
).startswith("500 Internal Server Error")
|
|
|
|
assert "0\n" == instance.query("select count() from tt")
|
|
|
|
instance.query("SYSTEM FLUSH LOGS")
|
|
|
|
assert "1\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday() and type = 'QueryStart'"
|
|
)
|
|
assert "1\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday() and type = 'ExceptionWhileProcessing'"
|
|
)
|
|
assert "0\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday() and type != 'QueryStart' and type != 'ExceptionWhileProcessing'"
|
|
)
|
|
|
|
instance.query("DROP TABLE tt SYNC")
|
|
|
|
|
|
def test_insert_over_http_invalid_statement(start_cluster):
|
|
http_status = 400
|
|
log_comment = f"{http_status}_02988_66a57d6f-d1cc-4693-8bf4-206848edab87"
|
|
assert True == instance.http_query_and_get_error(
|
|
f"insert into settings log_comment='{log_comment}' values (1), (2), (3), (4), (5)",
|
|
method="POST",
|
|
).startswith(f"{http_status}")
|
|
|
|
instance.query("SYSTEM FLUSH LOGS")
|
|
|
|
assert f"0\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday()"
|
|
)
|
|
|
|
|
|
def test_insert_over_http_unknown_table(start_cluster):
|
|
http_status = 404
|
|
log_comment = f"{http_status}_02988_66a57d6f-d1cc-4693-8bf4-206848edab87"
|
|
assert True == instance.http_query_and_get_error(
|
|
f"insert into unknown_table settings log_comment='{log_comment}' values (1), (2), (3), (4), (5)",
|
|
method="POST",
|
|
).startswith(f"{http_status}")
|
|
|
|
instance.query("SYSTEM FLUSH LOGS")
|
|
|
|
assert f"1\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday() and type = 'ExceptionBeforeStart'"
|
|
)
|
|
assert f"0\n" == instance.query(
|
|
f"select count() from system.query_log where log_comment ='{log_comment}' and current_database = currentDatabase() and event_date >= yesterday() and type != 'ExceptionBeforeStart'"
|
|
)
|
|
|
|
|
|
def test_insert_over_http_ok(start_cluster):
|
|
instance.query("DROP TABLE IF EXISTS tt SYNC")
|
|
instance.query(
|
|
"CREATE TABLE tt (KeyID UInt32) Engine = ReplicatedMergeTree('/test_insert_exception_over_http/tt', 'r1') ORDER BY (KeyID)"
|
|
)
|
|
|
|
_, error = instance.http_query_and_get_answer_with_error(
|
|
"insert into tt settings log_comment='02988_66a57d6f-d1cc-4693-8bf4-206848edab87' values (1), (2), (3), (4), (5)",
|
|
method="POST",
|
|
)
|
|
assert error == None
|
|
|
|
assert "5\n" == instance.query("select count() from tt")
|
|
|
|
instance.query("SYSTEM FLUSH LOGS")
|
|
|
|
assert "1\n" == instance.query(
|
|
"select count() from system.query_log where log_comment ='02988_66a57d6f-d1cc-4693-8bf4-206848edab87' and current_database = currentDatabase() and event_date >= yesterday() and type = 'QueryStart'"
|
|
)
|
|
|
|
assert "1\n" == instance.query(
|
|
"select count() from system.query_log where log_comment ='02988_66a57d6f-d1cc-4693-8bf4-206848edab87' and current_database = currentDatabase() and event_date >= yesterday() and type = 'QueryFinish'"
|
|
)
|
|
|
|
instance.query("DROP TABLE tt SYNC")
|