mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 08:40:50 +00:00
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import pytest
|
|
import requests
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.network import PartitionManager
|
|
import helpers.keeper_utils as keeper_utils
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
CONFIG_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "configs")
|
|
|
|
node1 = cluster.add_instance(
|
|
"node1", main_configs=["configs/enable_keeper1.xml"], stay_alive=True
|
|
)
|
|
node2 = cluster.add_instance(
|
|
"node2", main_configs=["configs/enable_keeper2.xml"], stay_alive=True
|
|
)
|
|
node3 = cluster.add_instance(
|
|
"node3", main_configs=["configs/enable_keeper3.xml"], stay_alive=True
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_http_readiness_basic_responses(started_cluster):
|
|
leader = keeper_utils.get_leader(cluster, [node1, node2, node3])
|
|
response = requests.get(
|
|
"http://{host}:{port}/ready".format(host=leader.ip_address, port=9182)
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
readiness_data = response.json()
|
|
assert readiness_data["status"] == "ok"
|
|
assert readiness_data["details"]["role"] == "leader"
|
|
|
|
follower = keeper_utils.get_any_follower(cluster, [node1, node2, node3])
|
|
response = requests.get(
|
|
"http://{host}:{port}/ready".format(host=follower.ip_address, port=9182)
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
readiness_data = response.json()
|
|
assert readiness_data["status"] == "ok"
|
|
assert readiness_data["details"]["role"] == "follower"
|
|
assert readiness_data["details"]["hasLeader"] == True
|
|
|
|
|
|
def test_http_readiness_partitioned_cluster(started_cluster):
|
|
with PartitionManager() as pm:
|
|
leader = keeper_utils.get_leader(cluster, [node1, node2, node3])
|
|
follower = keeper_utils.get_any_follower(cluster, [node1, node2, node3])
|
|
|
|
pm.partition_instances(leader, follower)
|
|
keeper_utils.wait_until_quorum_lost(cluster, follower)
|
|
|
|
response = requests.get(
|
|
"http://{host}:{port}/ready".format(host=follower.ip_address, port=9182)
|
|
)
|
|
print(response.json())
|
|
assert response.status_code == 503
|
|
|
|
readiness_data = response.json()
|
|
assert readiness_data["status"] == "fail"
|
|
assert readiness_data["details"]["role"] == "follower"
|
|
assert readiness_data["details"]["hasLeader"] == False
|