def fill_tables(step, count): db_manager = dbmanager.DBManager(world.config['connections']['mysql']) db = db_manager.get_db() world.farm_id = list() try: for i in range(int(count)): while True: farm_id = random.randint(1, 9999) if db.farms.filter(db.farms.id == farm_id).first() is None: break continue db.farms.insert(id=farm_id, env_id=0, changed_by_id=0) try: os.makedirs('%s/%s/%s' % (world.config['rrd']['dir'], helper.x1x2(farm_id), farm_id)) except OSError as e: if e.args[0] != 17: raise world.farm_id.append(farm_id) db.commit() finally: db.session.remove() lib.wait_sec(1) assert True
def fill_tables(step, count, st, tp): db_manager = dbmanager.DBManager(world.config['connections']['mysql']) db = db_manager.get_db() try: for i in range(int(count)): while True: msg_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter( db.messages.messageid == msg_id).first() is None: break continue while True: farm_id = random.randint(1, 9999) if db.farm_settings.filter( db.farm_settings.farmid == farm_id).first() is None: break continue while True: srv_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter( db.servers.server_id == srv_id).first() is None: break continue while True: event_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.events.filter( db.events.event_id == event_id).first() is None: break continue db.messages.insert(messageid=msg_id, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id, type='%s' % tp, message_version=2, message_name=random.choice(['ExecScript', '']), message_format=random.choice(['xml', 'json']), event_id=event_id) world.msgs_id.setdefault(msg_id, {}).setdefault('status', st) db.servers.insert(farm_id=farm_id, server_id=srv_id, env_id=1, status='Running', remote_ip='127.0.0.1') db.events.insert(event_id=event_id, msg_sent=0) world.srvs_id.append(srv_id) db.server_properties.insert(server_id=srv_id, name='scalarizr.key', value='hoho') db.server_properties.insert(server_id=srv_id, name='scalarizr.ctrl_port', value=8013) db.commit() finally: db.session.remove() lib.wait_sec(1) assert True
def wait_sec(step, sec): lib.wait_sec(int(sec)) assert True
def fill_tables_vpc(step, count, st, tp): db_manager = dbmanager.DBManager(world.config['connections']['mysql']) db = db_manager.get_db() try: for i in range(int(count)): while True: msg_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter( db.messages.messageid == msg_id).first() is None: break continue while True: msg_id_router = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter(db.messages.messageid == msg_id_router).first() is None: break continue while True: farm_id = random.randint(1, 20000) if db.farm_settings.filter( db.farm_settings.farmid == farm_id).first() is None: break continue while True: farm_role_id = random.randint(1, 20000) if db.role_behaviors.filter(db.role_behaviors.role_id == farm_role_id).first() is None: break continue while True: farm_role_id_router = random.randint(1, 20000) if db.role_behaviors.filter( db.role_behaviors.role_id == farm_role_id_router).first() is None: break continue while True: srv_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter( db.servers.server_id == srv_id).first() is None: break continue while True: srv_id_router = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter( db.servers.server_id == srv_id_router).first() is None: break continue while True: event_id = ''.join( random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.events.filter( db.events.event_id == event_id).first() is None: break continue db.messages.insert(messageid=msg_id, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id, type='%s' % tp, message_version=2, message_format=random.choice(['xml', 'json']), event_id=event_id) db.messages.insert(messageid=msg_id_router, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id_router, type='%s' % tp, message_version=2, message_format=random.choice(['xml', 'json'])) db.farms.insert(farm_id=farm_id, env_id=1) db.events.insert(event_id=event_id, msg_sent=0) db.farm_roles.insert(farmid=farm_id, role_id=farm_role_id) db.farm_roles.insert(farmid=farm_id, role_id=farm_role_id_router) db.servers.insert(farm_id=farm_id, farm_roleid=farm_role_id, server_id=srv_id, env_id=1, status='Running', local_ip='244.244.244.244') db.servers.insert(farm_id=farm_id, farm_roleid=farm_role_id_router, server_id=srv_id_router, env_id=1, status='Running', local_ip='254.254.254.254') db.role_behaviors.insert(role_id=farm_role_id, behavior='not router') db.role_behaviors.insert(role_id=farm_role_id_router, behavior='router') db.farm_settings.insert(farmid=farm_id, name='ec2.vpc.id', value='1') id_ = db.farm_roles.filter( db.farm_roles.role_id == farm_role_id_router, db.farm_roles.farmid == farm_id).first().id db.farm_role_settings.insert(farm_roleid=id_, name='router.vpc.ip', value='127.0.0.1') world.srvs_id.append(srv_id) world.srvs_id.append(srv_id_router) db.server_properties.insert(server_id=srv_id, name='scalarizr.key', value='hoho') db.server_properties.insert(server_id=srv_id, name='scalarizr.ctrl_port', value='8013') db.server_properties.insert(server_id=srv_id_router, name='scalarizr.key', value='hoho') db.server_properties.insert(server_id=srv_id_router, name='scalarizr.ctrl_port', value='8013') db.commit() finally: db.session.remove() lib.wait_sec(1) assert True
def fill_tables(step, count, st, tp): db_manager = dbmanager.DBManager(world.config['connections']['mysql']) db = db_manager.get_db() try: for i in range(int(count)): while True: msg_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter(db.messages.messageid == msg_id).first() is None: break continue while True: farm_id = random.randint(1, 9999) if db.farm_settings.filter(db.farm_settings.farmid == farm_id).first() is None: break continue while True: srv_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter(db.servers.server_id == srv_id).first() is None: break continue while True: event_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.events.filter(db.events.event_id == event_id).first() is None: break continue db.messages.insert( messageid=msg_id, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id, type='%s' % tp, message_version=2, message_name=random.choice(['ExecScript', '']), message_format=random.choice(['xml', 'json']), event_id=event_id) world.msgs_id.setdefault(msg_id, {}).setdefault('status', st) db.servers.insert( farm_id=farm_id, server_id=srv_id, env_id=1, status='Running', remote_ip='127.0.0.1') db.events.insert(event_id=event_id, msg_sent=0) world.srvs_id.append(srv_id) db.server_properties.insert( server_id=srv_id, name='scalarizr.key', value='hoho') db.server_properties.insert( server_id=srv_id, name='scalarizr.ctrl_port', value=8013) db.commit() finally: db.session.remove() lib.wait_sec(1) assert True
def fill_tables_vpc(step, count, st, tp): db_manager = dbmanager.DBManager(world.config['connections']['mysql']) db = db_manager.get_db() try: for i in range(int(count)): while True: msg_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter(db.messages.messageid == msg_id).first() is None: break continue while True: msg_id_router = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(75)) if db.messages.filter(db.messages.messageid == msg_id_router).first() is None: break continue while True: farm_id = random.randint(1, 20000) if db.farm_settings.filter(db.farm_settings.farmid == farm_id).first() is None: break continue while True: farm_role_id = random.randint(1, 20000) if db.role_behaviors.filter( db.role_behaviors.role_id == farm_role_id).first() is None: break continue while True: farm_role_id_router = random.randint(1, 20000) if db.role_behaviors.filter( db.role_behaviors.role_id == farm_role_id_router).first() is None: break continue while True: srv_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter(db.servers.server_id == srv_id).first() is None: break continue while True: srv_id_router = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.servers.filter(db.servers.server_id == srv_id_router).first() is None: break continue while True: event_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(36)) if db.events.filter(db.events.event_id == event_id).first() is None: break continue db.messages.insert( messageid=msg_id, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id, type='%s' % tp, message_version=2, message_format=random.choice(['xml', 'json']), event_id=event_id) db.messages.insert( messageid=msg_id_router, status=int(st), handle_attempts=0, dtlasthandleattempt=func.now(), message='some text here', server_id=srv_id_router, type='%s' % tp, message_version=2, message_format=random.choice(['xml', 'json'])) db.farms.insert(farm_id=farm_id, env_id=1) db.events.insert(event_id=event_id, msg_sent=0) db.farm_roles.insert(farmid=farm_id, role_id=farm_role_id) db.farm_roles.insert(farmid=farm_id, role_id=farm_role_id_router) db.servers.insert( farm_id=farm_id, farm_roleid=farm_role_id, server_id=srv_id, env_id=1, status='Running', local_ip='244.244.244.244') db.servers.insert( farm_id=farm_id, farm_roleid=farm_role_id_router, server_id=srv_id_router, env_id=1, status='Running', local_ip='254.254.254.254') db.role_behaviors.insert(role_id=farm_role_id, behavior='not router') db.role_behaviors.insert(role_id=farm_role_id_router, behavior='router') db.farm_settings.insert(farmid=farm_id, name='ec2.vpc.id', value='1') id_ = db.farm_roles.filter( db.farm_roles.role_id == farm_role_id_router, db.farm_roles.farmid == farm_id).first().id db.farm_role_settings.insert( farm_roleid=id_, name='router.vpc.ip', value='127.0.0.1') world.srvs_id.append(srv_id) world.srvs_id.append(srv_id_router) db.server_properties.insert( server_id=srv_id, name='scalarizr.key', value='hoho') db.server_properties.insert( server_id=srv_id, name='scalarizr.ctrl_port', value='8013') db.server_properties.insert( server_id=srv_id_router, name='scalarizr.key', value='hoho') db.server_properties.insert( server_id=srv_id_router, name='scalarizr.ctrl_port', value='8013') db.commit() finally: db.session.remove() lib.wait_sec(1) assert True