mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-21 23:21:59 +00:00
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
import time
|
|
import pytest
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.client import QueryRuntimeException
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
ch1 = cluster.add_instance('ch1', config_dir="configs", with_zookeeper=True)
|
|
ch2 = cluster.add_instance('ch2', config_dir="configs", with_zookeeper=True)
|
|
ch3 = cluster.add_instance('ch3', config_dir="configs", with_zookeeper=True)
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_access_control_on_cluster():
|
|
ch1.query("CREATE USER Alex ON CLUSTER 'cluster'")
|
|
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")
|
|
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")
|
|
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'")
|
|
assert "User `Alex` not found" in ch1.query_and_get_error("SHOW CREATE USER Alex")
|
|
assert "User `Alex` not found" in ch2.query_and_get_error("SHOW CREATE USER Alex")
|
|
assert "User `Alex` not found" in ch3.query_and_get_error("SHOW CREATE USER Alex")
|
|
|