mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 00:30:49 +00:00
fix black with script
This commit is contained in:
parent
12e6645058
commit
70e7c4e63d
@ -23,8 +23,10 @@ def started_cluster():
|
|||||||
finally:
|
finally:
|
||||||
cluster.shutdown()
|
cluster.shutdown()
|
||||||
|
|
||||||
|
|
||||||
expected_error = "User can not be created/updated because it exceeds the allowed quantity of authentication methods per user"
|
expected_error = "User can not be created/updated because it exceeds the allowed quantity of authentication methods per user"
|
||||||
|
|
||||||
|
|
||||||
def test_create(started_cluster):
|
def test_create(started_cluster):
|
||||||
|
|
||||||
assert expected_error in limited_node.query_and_get_error(
|
assert expected_error in limited_node.query_and_get_error(
|
||||||
@ -37,6 +39,7 @@ def test_create(started_cluster):
|
|||||||
|
|
||||||
limited_node.query("DROP USER u_max_authentication_methods")
|
limited_node.query("DROP USER u_max_authentication_methods")
|
||||||
|
|
||||||
|
|
||||||
def test_alter(started_cluster):
|
def test_alter(started_cluster):
|
||||||
limited_node.query("CREATE USER u_max_authentication_methods IDENTIFIED BY '1'")
|
limited_node.query("CREATE USER u_max_authentication_methods IDENTIFIED BY '1'")
|
||||||
|
|
||||||
@ -59,42 +62,65 @@ def test_alter(started_cluster):
|
|||||||
limited_node.query("DROP USER u_max_authentication_methods")
|
limited_node.query("DROP USER u_max_authentication_methods")
|
||||||
|
|
||||||
|
|
||||||
def get_query_with_multiple_identified_with(operation, username, identified_with_count, add_operation = ""):
|
def get_query_with_multiple_identified_with(
|
||||||
|
operation, username, identified_with_count, add_operation=""
|
||||||
|
):
|
||||||
identified_clauses = ", ".join([f"BY '1'" for _ in range(identified_with_count)])
|
identified_clauses = ", ".join([f"BY '1'" for _ in range(identified_with_count)])
|
||||||
query = f"{operation} USER {username} {add_operation} IDENTIFIED {identified_clauses}"
|
query = (
|
||||||
|
f"{operation} USER {username} {add_operation} IDENTIFIED {identified_clauses}"
|
||||||
|
)
|
||||||
return query
|
return query
|
||||||
|
|
||||||
|
|
||||||
def test_create_default_setting(started_cluster):
|
def test_create_default_setting(started_cluster):
|
||||||
expected_error = "User can not be created/updated because it exceeds the allowed quantity of authentication methods per user"
|
expected_error = "User can not be created/updated because it exceeds the allowed quantity of authentication methods per user"
|
||||||
|
|
||||||
query_exceeds = get_query_with_multiple_identified_with("CREATE", "u_max_authentication_methods", 257)
|
query_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"CREATE", "u_max_authentication_methods", 257
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error in default_node.query_and_get_error(query_exceeds)
|
assert expected_error in default_node.query_and_get_error(query_exceeds)
|
||||||
|
|
||||||
query_not_exceeds = get_query_with_multiple_identified_with("CREATE", "u_max_authentication_methods", 256)
|
query_not_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"CREATE", "u_max_authentication_methods", 256
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error not in default_node.query_and_get_answer_with_error(query_not_exceeds)
|
assert expected_error not in default_node.query_and_get_answer_with_error(
|
||||||
|
query_not_exceeds
|
||||||
|
)
|
||||||
|
|
||||||
default_node.query("DROP USER u_max_authentication_methods")
|
default_node.query("DROP USER u_max_authentication_methods")
|
||||||
|
|
||||||
|
|
||||||
def test_alter_default_setting(started_cluster):
|
def test_alter_default_setting(started_cluster):
|
||||||
default_node.query("CREATE USER u_max_authentication_methods IDENTIFIED BY '1'")
|
default_node.query("CREATE USER u_max_authentication_methods IDENTIFIED BY '1'")
|
||||||
|
|
||||||
query_add_exceeds = get_query_with_multiple_identified_with("ALTER", "u_max_authentication_methods", 256, "ADD")
|
query_add_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"ALTER", "u_max_authentication_methods", 256, "ADD"
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error in default_node.query_and_get_error(query_add_exceeds)
|
assert expected_error in default_node.query_and_get_error(query_add_exceeds)
|
||||||
|
|
||||||
query_replace_exceeds = get_query_with_multiple_identified_with("ALTER", "u_max_authentication_methods", 257)
|
query_replace_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"ALTER", "u_max_authentication_methods", 257
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error in default_node.query_and_get_error(query_replace_exceeds)
|
assert expected_error in default_node.query_and_get_error(query_replace_exceeds)
|
||||||
|
|
||||||
query_add_not_exceeds = get_query_with_multiple_identified_with("ALTER", "u_max_authentication_methods", 1, "ADD")
|
query_add_not_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"ALTER", "u_max_authentication_methods", 1, "ADD"
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error not in default_node.query_and_get_answer_with_error(query_add_not_exceeds)
|
assert expected_error not in default_node.query_and_get_answer_with_error(
|
||||||
|
query_add_not_exceeds
|
||||||
|
)
|
||||||
|
|
||||||
query_replace_not_exceeds = get_query_with_multiple_identified_with("ALTER", "u_max_authentication_methods", 2)
|
query_replace_not_exceeds = get_query_with_multiple_identified_with(
|
||||||
|
"ALTER", "u_max_authentication_methods", 2
|
||||||
|
)
|
||||||
|
|
||||||
assert expected_error not in default_node.query_and_get_answer_with_error(query_replace_not_exceeds)
|
assert expected_error not in default_node.query_and_get_answer_with_error(
|
||||||
|
query_replace_not_exceeds
|
||||||
|
)
|
||||||
|
|
||||||
default_node.query("DROP USER u_max_authentication_methods")
|
default_node.query("DROP USER u_max_authentication_methods")
|
||||||
|
Loading…
Reference in New Issue
Block a user