def __init__(self, maxconnection=10): # copie de l'état lors de l'initialisation d'une nouvelle instance self.__dict__ = self.__shared_state if not(self.initialized): #N MajorDomoClient #print "INIT ConnectionPool" zb = ZooBorg(Configuration.getclient()['zookeeper']['ip'], Configuration.getclient()['zookeeper']['port']) zooconf = zb.getConf(ZooBorg.CLIENT) connection_str=zooconf['broker']['connectionstr'] uniq = Uniq() for _ in range(1, maxconnection): conn=MajorDomoClient(connection_str, False, self) self.__availableconnection.append(conn) #register connexion zb.register(zb.CLIENT, uniq.getid(uniq.CLIENT), self._zconfchanged) self.initialized=True
def test_client(self): ip = Configuration.getclient()['zookeeper']['ip'] port = Configuration.getclient()['zookeeper']['port'] zb = ZooBorg(ip, port) try: client_id = 'client_test_id' zb.register(ZooBorg.CLIENT, client_id, self.stubhandler) # conf must be updated # client id must be in client list cl = zb.getList(ZooBorg.CLIENT) assert client_id in cl #just the one created here (connection pools are dead here) #print "clientlist:",cl assert len(cl) == 1 except: traceback.print_exc() assert 0 finally: zb.unregister(ZooBorg.CLIENT, client_id)
zb.zk.ensure_path("/distark/broker/conf/conf_reload_trigger") if __name__ == '__main__': ############################################## # ARGUMENTS PARSING ############################################## parser = argparse.ArgumentParser(description='Send requests') parser.add_argument('do', help='test initclientconf initworkerconf initbrokerconf initall', type=str) parser.set_defaults(do='test') args= parser.parse_args() print "Program Launched with args:" + str(args) print "Do:" + str(args.do) zb=None ip = Configuration.getclient()['zookeeper']['ip'] port = Configuration.getclient()['zookeeper']['port'] zb = ZooBorg(ip, port) try: if args.do == 'initclientconf': _initclientconf(zb) elif args.do == 'initworkerconf': _initworkerconf(zb) elif args.do == 'initbrokerconf': _initbrokerconf(zb) elif args.do == 'initall': _initworkerconf(zb) _initclientconf(zb) _initbrokerconf(zb) else: print 'do nothing !!!'
def test_conf_client(self): assert Configuration.getclient() is not None assert Configuration.initialized is True assert Configuration.client_initialized is True assert len(Configuration.getclient()['zookeeper']['ip'])>0