thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix black with script

Browse Source
This commit is contained in:
Arthur Passos 2024-08-15 12:21:06 -03:00
parent 12e6645058
commit 70e7c4e63d
1 changed files with 37 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
tests/integration/test_max_authentication_methods_per_user/test.py
View File

@ -23,8 +23,10 @@ def started_cluster():
finally:
cluster.shutdown()
expected_error = "User can not be created/updated because it exceeds the allowed quantity of authentication methods per user"
def test_create(started_cluster):
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")
def test_alter(started_cluster):
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")
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)])
query = f"{operation} USER {username} {add_operation} IDENTIFIED {identified_clauses}"
query = (
f"{operation} USER {username} {add_operation} IDENTIFIED {identified_clauses}"
)
return query
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"
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)
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")
def test_alter_default_setting(started_cluster):
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)
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)
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")