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 wait_sec(step, sec):
lib.wait_sec(int(sec))
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
