def setUp(self): """ make sure that the prefix dir is empty """ self.tests_path = os.getenv("ZKSHELL_PREFIX_DIR", "/tests") self.zk_hosts = ",".join(server.address for server in get_global_cluster()) self.username = os.getenv("ZKSHELL_USER", "user") self.password = os.getenv("ZKSHELL_PASSWD", "user") self.digested_password = os.getenv("ZKSHELL_DIGESTED_PASSWD", "F46PeTVYeItL6aAyygIVQ9OaaeY=") self.super_password = os.getenv("ZKSHELL_SUPER_PASSWD", "test") self.scheme = os.getenv("ZKSHELL_AUTH_SCHEME", "digest") self.client = KazooClient(self.zk_hosts, 5) self.client.start() self.client.add_auth(self.scheme, self.auth_id) if self.client.exists(self.tests_path): self.client.delete(self.tests_path, recursive=True) self.client.create(self.tests_path, str.encode("")) self.output = XStringIO() self.shell = Shell([self.zk_hosts], 5, self.output, setup_readline=False, asynchronous=False) # Create an empty test dir (needed for some tests) self.temp_dir = tempfile.mkdtemp()
def test_connect_async(self): """ test async """ # SIGUSR2 is emitted when connecting asyncronously, so handle it def handler(*args, **kwargs): pass signal.signal(signal.SIGUSR2, handler) shell = Shell([], 1, self.output, setup_readline=False, async=True) shell.onecmd("connect %s" % (self.zk_hosts)) self.assertTrue(wait_connected(shell))
def setUp(self): """ make sure that the prefix dir is empty """ self.zk_hosts = ",".join(server.address for server in get_global_cluster()) self.output = StringIO() self.shell = Shell([], 1, self.output, setup_readline=False, async=False)