def initConnection(self): host = Configuration.getworker()['mongo']['host'] port = Configuration.getworker()['mongo']['port'] db = Configuration.getworker()['mongo']['db'] self.mp = MongoPool(host, port, db, self.MAX_CONNECTION) #print mp return self.mp.getConnection()
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)
import re from sh import wc from sh import cat from sh import mv from distark.commons.utils.db.mongopool import MongoPool from distark.commons.utils.MyConfiguration import Configuration MODE_DESC='description' MODE_POINT='points' MODE_PROPERTIES='prop' MODE_QTE="qte" MODE_NOTHING='nothing' MAX_CONN = 1 host = Configuration.getworker()['mongo']['host'] port = Configuration.getworker()['mongo']['port'] db = Configuration.getworker()['mongo']['db'] mp = MongoPool(host, port, db, MAX_CONN) dbcon = mp.getConnection() refloat = re.compile('^\d+(\.\d+)*$') def is_mode_qte(line): res=False qte = ['100 g', '1 part', '1 pot', u'1 unité', '1 tranche', '1 portion', '100 ml', '1 canette', 'c.c.'] if line in qte: res = True return res
def test_conf_worker(self): assert Configuration.getworker() is not None assert Configuration.initialized is True assert Configuration.worker_initialized is True assert len(Configuration.getworker()['zookeeper']['ip'])>0