mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 16:50:48 +00:00
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from helpers.cluster import ClickHouseCluster
|
||
|
|
|
||
|
|
cluster = ClickHouseCluster(__file__)
|
||
|
|
node = cluster.add_instance("node")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="module")
|
||
|
|
def started_cluster():
|
||
|
|
try:
|
||
|
|
cluster.start()
|
||
|
|
yield cluster
|
||
|
|
|
||
|
|
finally:
|
||
|
|
cluster.shutdown()
|
||
|
|
|
||
|
|
|
||
|
|
def test_basic(started_cluster):
|
||
|
|
# 1. Without VALID UNTIL
|
||
|
|
node.query("CREATE USER user_basic")
|
||
|
|
|
||
|
|
assert node.query("SHOW CREATE USER user_basic") == "CREATE USER user_basic\n"
|
||
|
|
assert node.query("SELECT 1", user="user_basic") == "1\n"
|
||
|
|
|
||
|
|
# 2. With valid VALID UNTIL
|
||
|
|
node.query("ALTER USER user_basic VALID UNTIL '06/11/2040 08:03:20 Z+3'")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
node.query("SHOW CREATE USER user_basic")
|
||
|
|
== "CREATE USER user_basic VALID UNTIL \\'2040-11-06 05:03:20\\'\n"
|
||
|
|
)
|
||
|
|
assert node.query("SELECT 1", user="user_basic") == "1\n"
|
||
|
|
|
||
|
|
# 3. With invalid VALID UNTIL
|
||
|
|
node.query("ALTER USER user_basic VALID UNTIL '06/11/2010 08:03:20 Z+3'")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
node.query("SHOW CREATE USER user_basic")
|
||
|
|
== "CREATE USER user_basic VALID UNTIL \\'2010-11-06 05:03:20\\'\n"
|
||
|
|
)
|
||
|
|
|
||
|
|
error = "Authentication failed"
|
||
|
|
assert error in node.query_and_get_error("SELECT 1", user="user_basic")
|
||
|
|
|
||
|
|
# 4. Reset VALID UNTIL
|
||
|
|
node.query("ALTER USER user_basic VALID UNTIL 'infinity'")
|
||
|
|
|
||
|
|
assert node.query("SHOW CREATE USER user_basic") == "CREATE USER user_basic\n"
|
||
|
|
assert node.query("SELECT 1", user="user_basic") == "1\n"
|
||
|
|
|
||
|
|
|
||
|
|
def test_details(started_cluster):
|
||
|
|
# 1. Does not do anything
|
||
|
|
node.query("CREATE USER user_details_infinity VALID UNTIL 'infinity'")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
node.query("SHOW CREATE USER user_details_infinity")
|
||
|
|
== "CREATE USER user_details_infinity\n"
|
||
|
|
)
|
||
|
|
|
||
|
|
# 2. Time only is not supported
|
||
|
|
node.query("CREATE USER user_details_time_only VALID UNTIL '22:03:40'")
|
||
|
|
|
||
|
|
assert (
|
||
|
|
node.query("SHOW CREATE USER user_details_time_only")
|
||
|
|
== "CREATE USER user_details_time_only VALID UNTIL \\'2000-01-01 22:03:40\\'\n"
|
||
|
|
)
|