Add retries to test_access_control_on_cluster

This commit is contained in:
Yatsishin Ilya 2021-02-09 18:37:33 +03:00
parent be27578294
commit eac13d9d9d

View File

@ -18,22 +18,22 @@ def started_cluster():
def test_access_control_on_cluster():
ch1.query("CREATE USER Alex ON CLUSTER 'cluster'")
ch1.query_with_retry("CREATE USER Alex ON CLUSTER 'cluster'", retry_count=3)
assert ch1.query("SHOW CREATE USER Alex") == "CREATE USER Alex\n"
assert ch2.query("SHOW CREATE USER Alex") == "CREATE USER Alex\n"
assert ch3.query("SHOW CREATE USER Alex") == "CREATE USER Alex\n"
ch2.query("GRANT ON CLUSTER 'cluster' SELECT ON *.* TO Alex")
ch2.query_with_retry("GRANT ON CLUSTER 'cluster' SELECT ON *.* TO Alex", retry_count=3)
assert ch1.query("SHOW GRANTS FOR Alex") == "GRANT SELECT ON *.* TO Alex\n"
assert ch2.query("SHOW GRANTS FOR Alex") == "GRANT SELECT ON *.* TO Alex\n"
assert ch3.query("SHOW GRANTS FOR Alex") == "GRANT SELECT ON *.* TO Alex\n"
ch3.query("REVOKE ON CLUSTER 'cluster' SELECT ON *.* FROM Alex")
ch3.query_with_retry("REVOKE ON CLUSTER 'cluster' SELECT ON *.* FROM Alex", retry_count=3)
assert ch1.query("SHOW GRANTS FOR Alex") == ""
assert ch2.query("SHOW GRANTS FOR Alex") == ""
assert ch3.query("SHOW GRANTS FOR Alex") == ""
ch2.query("DROP USER Alex ON CLUSTER 'cluster'")
ch2.query_with_retry("DROP USER Alex ON CLUSTER 'cluster'", retry_count=3)
assert "There is no user `Alex`" in ch1.query_and_get_error("SHOW CREATE USER Alex")
assert "There is no user `Alex`" in ch2.query_and_get_error("SHOW CREATE USER Alex")
assert "There is no user `Alex`" in ch3.query_and_get_error("SHOW CREATE USER Alex")