diff options
author | Raul Gutierrez S <rgs@itevenworks.net> | 2015-07-11 15:55:08 -0700 |
---|---|---|
committer | Raul Gutierrez S <rgs@itevenworks.net> | 2015-07-11 15:55:08 -0700 |
commit | 7374e19597a24c3feaa07b77af07607f56106c1f (patch) | |
tree | 41dfce99adebca44924e5ea61e9e0f62bbd3a41c | |
parent | 9d436602b06fe08c83d7f78986a8dee789153445 (diff) | |
download | kazoo-7374e19597a24c3feaa07b77af07607f56106c1f.tar.gz |
Allow setting ZooKeeperCluster's size via ZK_CLUSTER_SIZE
Signed-off-by: Raul Gutierrez S <rgs@itevenworks.net>
-rw-r--r-- | kazoo/testing/harness.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/kazoo/testing/harness.py b/kazoo/testing/harness.py index 26fe2d2..bc83b26 100644 --- a/kazoo/testing/harness.py +++ b/kazoo/testing/harness.py @@ -26,6 +26,7 @@ def get_global_cluster(): ZK_HOME = os.environ.get("ZOOKEEPER_PATH") ZK_CLASSPATH = os.environ.get("ZOOKEEPER_CLASSPATH") ZK_PORT_OFFSET = int(os.environ.get("ZOOKEEPER_PORT_OFFSET", 20000)) + ZK_CLUSTER_SIZE = int(os.environ.get("ZOOKEEPER_CLUSTER_SIZE", 3)) assert ZK_HOME or ZK_CLASSPATH, ( "Either ZOOKEEPER_PATH or ZOOKEEPER_CLASSPATH environment " @@ -36,6 +37,7 @@ def get_global_cluster(): install_path=ZK_HOME, classpath=ZK_CLASSPATH, port_offset=ZK_PORT_OFFSET, + size=ZK_CLUSTER_SIZE ) atexit.register(lambda cluster: cluster.terminate(), CLUSTER) return CLUSTER |