mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-19 15:01:03 +00:00
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
import docker
|
|
import os
|
|
import pwd
|
|
import pytest
|
|
import re
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
|
|
def expect_failure_with_message(expected_message):
|
|
cluster = ClickHouseCluster(__file__)
|
|
node = cluster.add_instance('node', with_zookeeper=False)
|
|
|
|
with pytest.raises(Exception):
|
|
cluster.start()
|
|
|
|
current_user_id = os.getuid()
|
|
other_user_id = pwd.getpwnam('nobody').pw_uid
|
|
|
|
docker_api = docker.from_env().api
|
|
container = node.get_docker_handle()
|
|
container.start()
|
|
container.exec_run('chown {0} /var/lib/clickhouse'.format(other_user_id), privileged=True)
|
|
container.exec_run('clickhouse server --config-file=/etc/clickhouse-server/config.xml --log-file=/var/log/clickhouse-server/clickhouse-server.log --errorlog-file=/var/log/clickhouse-server/clickhouse-server.err.log', privileged=True)
|
|
|
|
cluster.shutdown() # cleanup
|
|
|
|
with open(os.path.join(node.path, 'logs/clickhouse-server.err.log')) as log:
|
|
last_message = log.readlines()[-1].strip()
|
|
if re.search(expected_message, last_message) is None:
|
|
pytest.fail('Expected the server to fail with a message "{}", but the last message is "{}"'.format(expected_message, last_message))
|
|
|
|
|
|
def test_different_user():
|
|
current_user_id = os.getuid()
|
|
|
|
if current_user_id == 0:
|
|
current_user_name = pwd.getpwuid(current_user_id).pw_name
|
|
|
|
expect_failure_with_message(
|
|
'Effective user of the process \(.*\) does not match the owner of the data \(.*\)',
|
|
)
|