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_worker(self):
ip = Configuration.getworker()['zookeeper']['ip']
port = Configuration.getworker()['zookeeper']['port']
zb = ZooBorg(ip, port)
worker_id = 'test_worket_id'
try:
zb.register(ZooBorg.WORKER, worker_id, self.stubhandler)
# client id must be in client list
cl = zb.getList(ZooBorg.WORKER)
#print "workerlist:",cl
assert worker_id in cl
#just created + test infra worker
assert len(cl) >= 2
except:
traceback.print_exc()
assert 0
finally:
zb.unregister(ZooBorg.WORKER, worker_id)
def test_broker(self):
ip = Configuration.getbroker()['zookeeper']['ip']
port = Configuration.getbroker()['zookeeper']['port']
zb = ZooBorg(ip, port)
try:
broker_id = 'broker_test_id'
zb.register(ZooBorg.BROKER, broker_id, self.stubhandler)
# conf must be updated
# broker id must be in broker list
cl = zb.getList(ZooBorg.BROKER)
#print "brokerlist:",cl
assert broker_id in cl
#the one created here + infra broker(s)
assert len(cl) > 1
except:
traceback.print_exc()
assert 0
finally:
zb.unregister(ZooBorg.BROKER, broker_id)
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)
