ClickHouse/tests/integration/test_grant_and_revoke/test.py

130 lines
4.7 KiB
Python
Raw Normal View History

2020-02-04 01:15:14 +00:00
import pytest
from helpers.cluster import ClickHouseCluster
import re
cluster = ClickHouseCluster(__file__)
instance = cluster.add_instance('instance')
2020-02-04 01:15:14 +00:00
@pytest.fixture(scope="module", autouse=True)
def started_cluster():
try:
cluster.start()
2020-02-20 04:17:44 +00:00
instance.query("CREATE TABLE test_table(x UInt32, y UInt32) ENGINE = MergeTree ORDER BY tuple()")
instance.query("INSERT INTO test_table VALUES (1,5), (2,10)")
2020-02-04 01:15:14 +00:00
yield cluster
finally:
cluster.shutdown()
2020-02-20 04:17:44 +00:00
@pytest.fixture(autouse=True)
def reset_users_and_roles():
try:
yield
finally:
instance.query("DROP USER IF EXISTS A, B")
instance.query("DROP ROLE IF EXISTS R1, R2")
2020-02-04 01:15:14 +00:00
def test_login():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
instance.query("CREATE USER B")
2020-02-04 01:15:14 +00:00
assert instance.query("SELECT 1", user='A') == "1\n"
assert instance.query("SELECT 1", user='B') == "1\n"
def test_grant_and_revoke():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
2020-02-04 01:15:14 +00:00
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT SELECT ON test_table TO A')
2020-02-20 04:17:44 +00:00
assert instance.query("SELECT * FROM test_table", user='A') == "1\t5\n2\t10\n"
2020-02-04 01:15:14 +00:00
instance.query('REVOKE SELECT ON test_table FROM A')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
def test_grant_option():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
instance.query("CREATE USER B")
2020-02-20 04:17:44 +00:00
2020-02-04 01:15:14 +00:00
instance.query('GRANT SELECT ON test_table TO A')
2020-02-20 04:17:44 +00:00
assert instance.query("SELECT * FROM test_table", user='A') == "1\t5\n2\t10\n"
2020-02-04 01:15:14 +00:00
assert "Not enough privileges" in instance.query_and_get_error("GRANT SELECT ON test_table TO B", user='A')
instance.query('GRANT SELECT ON test_table TO A WITH GRANT OPTION')
instance.query("GRANT SELECT ON test_table TO B", user='A')
2020-02-20 04:17:44 +00:00
assert instance.query("SELECT * FROM test_table", user='B') == "1\t5\n2\t10\n"
2020-02-04 01:15:14 +00:00
instance.query('REVOKE SELECT ON test_table FROM A, B')
2020-02-20 04:17:44 +00:00
def test_create_role():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
2020-02-20 04:17:44 +00:00
instance.query('CREATE ROLE R1')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT SELECT ON test_table TO R1')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT R1 TO A')
assert instance.query("SELECT * FROM test_table", user='A') == "1\t5\n2\t10\n"
instance.query('REVOKE R1 FROM A')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
def test_grant_role_to_role():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
2020-02-20 04:17:44 +00:00
instance.query('CREATE ROLE R1')
instance.query('CREATE ROLE R2')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT R1 TO A')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT R2 TO R1')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT SELECT ON test_table TO R2')
assert instance.query("SELECT * FROM test_table", user='A') == "1\t5\n2\t10\n"
def test_combine_privileges():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A ")
2020-02-20 04:17:44 +00:00
instance.query('CREATE ROLE R1')
instance.query('CREATE ROLE R2')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
instance.query('GRANT R1 TO A')
instance.query('GRANT SELECT(x) ON test_table TO R1')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='A')
assert instance.query("SELECT x FROM test_table", user='A') == "1\n2\n"
instance.query('GRANT SELECT(y) ON test_table TO R2')
instance.query('GRANT R2 TO A')
assert instance.query("SELECT * FROM test_table", user='A') == "1\t5\n2\t10\n"
def test_admin_option():
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER A")
instance.query("CREATE USER B")
2020-02-20 04:17:44 +00:00
instance.query('CREATE ROLE R1')
instance.query('GRANT SELECT ON test_table TO R1')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='B')
instance.query('GRANT R1 TO A')
assert "Not enough privileges" in instance.query_and_get_error("GRANT R1 TO B", user='A')
assert "Not enough privileges" in instance.query_and_get_error("SELECT * FROM test_table", user='B')
instance.query('GRANT R1 TO A WITH ADMIN OPTION')
instance.query("GRANT R1 TO B", user='A')
assert instance.query("SELECT * FROM test_table", user='B') == "1\t5\n2\t10\n"