thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-20 00:30:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Properly close connections and add retries

Browse Source
This commit is contained in:
Antonio Andelic 2022-11-29 09:13:23 +00:00
parent a0bd99c8a2
commit e39f97a255
1 changed files with 30 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
tests/integration/test_keeper_zookeeper_converter/test.py
View File

@ -2,14 +2,9 @@
import pytest
from helpers.cluster import ClickHouseCluster
import helpers.keeper_utils as keeper_utils
from kazoo.client import KazooClient, KazooState
from kazoo.security import ACL, make_digest_acl, make_acl
from kazoo.exceptions import (
AuthFailedError,
InvalidACLError,
NoAuthError,
KazooException,
)
from kazoo.client import KazooClient
from kazoo.retry import KazooRetry
from kazoo.security import make_acl
import os
import time
@ -99,7 +94,9 @@ def get_fake_zk(timeout=60.0):
def get_genuine_zk(timeout=60.0):
_genuine_zk_instance = KazooClient(
hosts=cluster.get_instance_ip("node") + ":2181", timeout=timeout
hosts=cluster.get_instance_ip("node") + ":2181",
timeout=timeout,
connection_retry=KazooRetry(max_tries=20),
)
_genuine_zk_instance.start()
return _genuine_zk_instance
@ -225,6 +222,12 @@ def test_smoke(started_cluster, create_snapshots):
compare_states(genuine_connection, fake_connection)
genuine_connection.stop()
genuine_connection.close()
fake_connection.stop()
fake_connection.close()
def get_bytes(s):
return s.encode()
@ -309,6 +312,12 @@ def test_simple_crud_requests(started_cluster, create_snapshots):
second_children = list(sorted(fake_connection.get_children("/test_sequential")))
assert first_children == second_children, "Childrens are not equal on path " + path
genuine_connection.stop()
genuine_connection.close()
fake_connection.stop()
fake_connection.close()
@pytest.mark.parametrize(("create_snapshots"), [True, False])
def test_multi_and_failed_requests(started_cluster, create_snapshots):
@ -379,6 +388,12 @@ def test_multi_and_failed_requests(started_cluster, create_snapshots):
assert eph1 == eph2
compare_stats(stat1, stat2, "/test_multitransactions", ignore_pzxid=True)
genuine_connection.stop()
genuine_connection.close()
fake_connection.stop()
fake_connection.close()
@pytest.mark.parametrize(("create_snapshots"), [True, False])
def test_acls(started_cluster, create_snapshots):
@ -446,3 +461,9 @@ def test_acls(started_cluster, create_snapshots):
"user2:lo/iTtNMP+gEZlpUNaCqLYO3i5U=",
"user3:wr5Y0kEs9nFX3bKrTMKxrlcFeWo=",
)
genuine_connection.stop()
genuine_connection.close()
fake_connection.stop()
fake_connection.close()