ClickHouse/tests/integration/test_authentication/test.py

37 lines
1.2 KiB
Python
Raw Normal View History

2020-02-29 12:57:52 +00:00
import pytest
from helpers.cluster import ClickHouseCluster
cluster = ClickHouseCluster(__file__)
instance = cluster.add_instance('instance')
2020-02-29 12:57:52 +00:00
@pytest.fixture(scope="module", autouse=True)
def setup_nodes():
try:
cluster.start()
2020-03-18 14:12:09 +00:00
instance.query("CREATE USER sasha")
instance.query("CREATE USER masha IDENTIFIED BY 'qwerty'")
2020-02-29 12:57:52 +00:00
yield cluster
finally:
cluster.shutdown()
def test_authentication_pass():
assert instance.query("SELECT currentUser()", user='sasha') == 'sasha\n'
assert instance.query("SELECT currentUser()", user='masha', password='qwerty') == 'masha\n'
# 'no_password' authentication type allows to login with any password.
assert instance.query("SELECT currentUser()", user='sasha', password='something') == 'sasha\n'
assert instance.query("SELECT currentUser()", user='sasha', password='something2') == 'sasha\n'
2020-02-29 12:57:52 +00:00
def test_authentication_fail():
# User doesn't exist.
assert "vasya: Authentication failed" in instance.query_and_get_error("SELECT currentUser()", user = 'vasya')
# Wrong password.
assert "masha: Authentication failed" in instance.query_and_get_error("SELECT currentUser()", user = 'masha', password = '123')