def setUp(self): super(ClientSessionTests, self).setUp() self.cluster.start() self.client = None self.client2 = None zookeeper.deterministic_conn_order(True) zookeeper.set_debug_level(0)
def set_deterministic_order(self, boolean): """ The zookeeper client will by default randomize the server hosts it will connect to unless this is set to True. This is a global setting across connections. """ zookeeper.deterministic_conn_order(bool(boolean))
def setup_zookeeper(self): """Create a ZK cluster and chrooted :class:`KazooClient` The cluster will only be created on the first invocation and won't be fully torn down until exit. """ zookeeper.deterministic_conn_order(True) if not self.cluster[0].running: self.cluster.start() namespace = "/kazootests" + uuid.uuid4().hex self.hosts = self.servers + namespace self.client = self._get_client() self.client.connect() self.client.ensure_path("/")
import os import uuid import sys from collections import namedtuple import threading import unittest import zookeeper from kazoo.client import Callback from kazoo.client import KazooClient from kazoo.client import KazooState from kazoo.testing.common import ZookeeperCluster log = logging.getLogger(__name__) zookeeper.deterministic_conn_order(True) CLUSTER = None def get_global_cluster(): global CLUSTER if CLUSTER is None: ZK_HOME = os.environ.get("ZOOKEEPER_PATH") assert ZK_HOME, ( "ZOOKEEPER_PATH environment variable must be defined.\n " "For deb package installations this is /usr/share/java") CLUSTER = ZookeeperCluster(ZK_HOME) atexit.register(lambda cluster: cluster.terminate(), CLUSTER) return CLUSTER