Skip to content

Commit

Permalink
Automatic style fix
Browse files Browse the repository at this point in the history
  • Loading branch information
robot-clickhouse committed Jul 7, 2023
1 parent 5408a0e commit a9d4e71
Showing 1 changed file with 17 additions and 11 deletions.
28 changes: 17 additions & 11 deletions tests/integration/test_multi_access_storage_role_management/test.py
Expand Up @@ -31,7 +31,9 @@ def test_role_from_different_storages():
node.query("CREATE ROLE default_role")
node.query("GRANT SELECT ON system.* TO default_role")

assert node.query("SHOW GRANTS FOR default_role") == TSV(["GRANT SELECT ON system.* TO default_role"])
assert node.query("SHOW GRANTS FOR default_role") == TSV(
["GRANT SELECT ON system.* TO default_role"]
)
assert node.query("SHOW ROLES") == TSV(["default_role"])

node.copy_file_to_container(
Expand All @@ -49,23 +51,25 @@ def test_role_from_different_storages():
)

# Role from users.xml will have priority
assert node.query("SHOW GRANTS FOR default_role") == TSV(["GRANT ALL ON *.* TO default_role WITH GRANT OPTION"])
assert node.query("SHOW GRANTS FOR default_role") == TSV(
["GRANT ALL ON *.* TO default_role WITH GRANT OPTION"]
)

assert node.query("SHOW GRANTS FOR default_role AT local directory STORAGE") == TSV(
[
"GRANT SELECT ON system.* TO default_role"
]
["GRANT SELECT ON system.* TO default_role"]
)

assert node.query("SHOW GRANTS FOR default_role AT users xml STORAGE") == TSV(
[
"GRANT ALL ON *.* TO default_role WITH GRANT OPTION"
]
["GRANT ALL ON *.* TO default_role WITH GRANT OPTION"]
)

node.query("GRANT default_role TO test_user")
assert node.query("SELECT granted_role_id FROM system.role_grants WHERE user_name = 'test_user'") == TSV(
["62bedbf3-7fb1-94cb-3a35-e479693223b3"] # roles from users.xml have deterministic ids
assert node.query(
"SELECT granted_role_id FROM system.role_grants WHERE user_name = 'test_user'"
) == TSV(
[
"62bedbf3-7fb1-94cb-3a35-e479693223b3"
] # roles from users.xml have deterministic ids
)
node.query("REVOKE ALL FROM test_user")

Expand All @@ -76,4 +80,6 @@ def test_role_from_different_storages():
node.query("SELECT * from mydb.table1", user="test_user")

node.query("DROP ROLE default_role AT local directory STORAGE")
assert node.query("SELECT granted_role_id FROM system.role_grants WHERE user_name = 'test_user'") == TSV([])
assert node.query(
"SELECT granted_role_id FROM system.role_grants WHERE user_name = 'test_user'"
) == TSV([])

0 comments on commit a9d4e71

Please sign in to comment.