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

try to fix black

Browse Source
This commit is contained in:
Arthur Passos 2024-06-27 08:52:20 -03:00
parent 678700f137
commit ec0d426a6f
4 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/integration/test_backup_restore_new/test.py
View File

@ -1212,7 +1212,10 @@ def test_system_users_required_privileges():
instance.query("GRANT SELECT ON test.* TO u2 WITH GRANT OPTION")
instance.query(f"RESTORE ALL FROM {backup_name}", user="u2")
assert instance.query("SHOW CREATE USER u1") == "CREATE USER u1 IDENTIFIED WITH no_password DEFAULT ROLE r1\n"
assert (
instance.query("SHOW CREATE USER u1")
== "CREATE USER u1 IDENTIFIED WITH no_password DEFAULT ROLE r1\n"
)
assert instance.query("SHOW GRANTS FOR u1") == TSV(
["GRANT SELECT ON test.* TO u1", "GRANT r1 TO u1"]
)

3
tests/integration/test_backup_restore_on_cluster/test.py
View File

@ -769,7 +769,8 @@ def test_system_users():
)
assert (
node1.query("SHOW CREATE USER u1") == "CREATE USER u1 IDENTIFIED WITH no_password SETTINGS custom_a = 123\n"
node1.query("SHOW CREATE USER u1")
== "CREATE USER u1 IDENTIFIED WITH no_password SETTINGS custom_a = 123\n"
)
assert node1.query("SHOW GRANTS FOR u1") == "GRANT SELECT ON default.tbl TO u1\n"

6
tests/integration/test_enabling_access_management/test.py
View File

@ -20,10 +20,12 @@ def started_cluster():
def test_enabling_access_management():
instance.query("CREATE USER Alex", user="default")
assert (
instance.query("SHOW CREATE USER Alex", user="default") == "CREATE USER Alex IDENTIFIED WITH no_password\n"
instance.query("SHOW CREATE USER Alex", user="default")
== "CREATE USER Alex IDENTIFIED WITH no_password\n"
)
assert (
instance.query("SHOW CREATE USER Alex", user="readonly") == "CREATE USER Alex IDENTIFIED WITH no_password\n"
instance.query("SHOW CREATE USER Alex", user="readonly")
== "CREATE USER Alex IDENTIFIED WITH no_password\n"
)
assert "Not enough privileges" in instance.query_and_get_error(
"SHOW CREATE USER Alex", user="xyz"

11
tests/integration/test_grant_and_revoke/test.py
View File

@ -144,7 +144,8 @@ def test_allowed_grantees():
instance.query("ALTER USER A GRANTEES ANY EXCEPT B")
assert (
instance.query("SHOW CREATE USER A") == "CREATE USER A IDENTIFIED WITH no_password GRANTEES ANY EXCEPT B\n"
instance.query("SHOW CREATE USER A")
== "CREATE USER A IDENTIFIED WITH no_password GRANTEES ANY EXCEPT B\n"
)
expected_error = "user `B` is not allowed as grantee"
assert expected_error in instance.query_and_get_error(
@ -157,7 +158,10 @@ def test_allowed_grantees():
instance.query("REVOKE SELECT ON test.table FROM B", user="A")
instance.query("ALTER USER A GRANTEES ANY")
assert instance.query("SHOW CREATE USER A") == "CREATE USER A IDENTIFIED WITH no_password\n"
assert (
instance.query("SHOW CREATE USER A")
== "CREATE USER A IDENTIFIED WITH no_password\n"
)
instance.query("GRANT SELECT ON test.table TO B", user="A")
assert instance.query("SELECT * FROM test.table", user="B") == "1\t5\n2\t10\n"
@ -169,7 +173,8 @@ def test_allowed_grantees():
instance.query("CREATE USER C GRANTEES ANY EXCEPT C")
assert (
instance.query("SHOW CREATE USER C") == "CREATE USER C IDENTIFIED WITH no_password GRANTEES ANY EXCEPT C\n"
instance.query("SHOW CREATE USER C")
== "CREATE USER C IDENTIFIED WITH no_password GRANTEES ANY EXCEPT C\n"
)
instance.query("GRANT SELECT ON test.table TO C WITH GRANT OPTION")
assert instance.query("SELECT * FROM test.table", user="C") == "1\t5\n2\t10\n"