def test_it(worker_count, my_vars_file):
runname = str(uuid.uuid1()).replace("-", "")
conf = os.path.join(os.environ['EPUMGMT_HOME'], "etc/epumgmt/main.conf")
epu_opts = EPUMgmtOpts(name=runname, conf_file=conf)
epu_opts.jsonvars = my_vars_file
#epu_action.set_logfile(os.path.join(os.environ['EPUMGMT_HOME'], "tests/tests.logs"))
cyvm_a = []
(c, p, ac) = epumgmt.api.get_common(opts=epu_opts)
persist = Persistence(p, c)
persist.validate()
cm = persist.cdb
s_retry_count = 10
print "RUNNAME IS %s" % (runname)
service_name = os.environ['EPU_SERVICE']
rc = 0
error_msg = "SUCCESS"
try:
print "starting provisioner"
epu_opts.haservice = "provisioner"
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
print "starting %s" % (service_name)
epu_opts.haservice = service_name
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
epu_opts.haservice = None
pre_kill_len = 0
retry_count = s_retry_count
while pre_kill_len != 2 + worker_count:
print "finding workers"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
epu_opts.action = ACTIONS.LOGFETCH
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
cm.commit()
pre_kill_len = len(cyvm_a)
retry_count = retry_count - 1
if retry_count < 0:
raise Exception("No new workers found after %d tries %d" % (s_retry_count, worker_count))
time.sleep(20)
epu_opts.action = ACTIONS.STATUS
epumgmt_run(epu_opts)
except Exception, ex:
traceback.print_tb(sys.exc_info()[2])
print "========= ERROR ===================="
print ex
print "============================="
error_msg = str(ex)
rc = 1
def main(argv=sys.argv[1:]):
service_name = os.environ['EPU_SERVICE']
runname = str(uuid.uuid1()).replace("-", "")
conf = os.path.join(os.environ['EPUMGMT_HOME'], "etc/epumgmt/main.conf")
epu_opts = EPUMgmtOpts(name=runname, conf_file=conf)
my_vars_file = os.environ['EPU_TEST_VARS']
epu_opts.jsonvars = my_vars_file
#epu_action.set_logfile(os.path.join(os.environ['EPUMGMT_HOME'], "tests/tests.logs"))
cyvm_a = []
(c, p, ac) = epumgmt.api.get_common(opts=epu_opts)
persist = Persistence(p, c)
persist.validate()
cm = persist.cdb
try:
epu_opts.haservice = "provisioner"
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
epu_opts.haservice = service_name
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
if len(cyvm_a) != 2:
raise Exception("the wrong number of VMs found by miner")
found = 0
for c in cyvm_a:
if c.service_type == "provisioner":
found = found + 1
elif c.service_type == service_name:
found = found + 1
else:
raise Exception("unknown service type found")
if found != 2:
raise Exception("did not find %s and/or provisioner" % (service_name))
except Exception, ex:
print ex
return 1
def main(argv=sys.argv[1:]):
runname = str(uuid.uuid1()).replace("-", "")
conf = os.path.join(os.environ["EPUMGMT_HOME"], "etc/epumgmt/main.conf")
epu_opts = EPUMgmtOpts(name=runname, conf_file=conf)
my_vars_file = os.environ["EPU_TEST_VARS"]
epu_opts.jsonvars = my_vars_file
# epu_action.set_logfile(os.path.join(os.environ['EPUMGMT_HOME'], "tests/tests.logs"))
cyvm_a = []
(c, p, ac) = epumgmt.api.get_common(opts=epu_opts)
persist = Persistence(p, c)
persist.validate()
cm = persist.cdb
s_retry_count = 5
worker_count = 2 # need to get this from conf file
print "RUNNAME IS %s" % (runname)
service_name = os.environ["EPU_SERVICE"]
rc = 0
error_msg = "SUCCESS"
try:
print "starting provisioner"
epu_opts.haservice = "provisioner"
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
print "starting %s" % (service_name)
epu_opts.haservice = service_name
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
epu_opts.haservice = None
pre_kill_len = 0
retry_count = s_retry_count
while pre_kill_len != 2 + worker_count:
print "finding workers"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
cm.commit()
pre_kill_len = len(cyvm_a)
retry_count = retry_count - 1
if retry_count < 0:
raise Exception("No new workers found after %d tries" % (s_retry_count))
time.sleep(10)
print "make sure hostnames have been achieved"
done = False
while not done:
done = True
epu_opts.action = ACTIONS.LOGFETCH
epumgmt_run(epu_opts)
print "finding workers hostnames"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
for c in cyvm_a:
if c.hostname == None or c.hostname.strip() == "":
done = False
cm.commit()
epu_opts.action = ACTIONS.STATUS
epumgmt_run(epu_opts)
orig_iaas_ids = [c.iaasid for c in cyvm_a]
print "KILLING A WORKER!"
epu_opts.killnum = 1
epu_opts.action = ACTIONS.FETCH_KILL
epumgmt_run(epu_opts)
cyvm_a = None
post_kill_len = 0
retry_count = s_retry_count
while post_kill_len <= pre_kill_len:
print "fetch logs"
epu_opts.action = ACTIONS.LOGFETCH
epumgmt_run(epu_opts)
print "finding workers"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
post_kill_len = len(cyvm_a)
retry_count = retry_count - 1
if retry_count < 0:
raise Exception("No new workers found after %d tries" % (s_retry_count))
cm.commit()
time.sleep(10)
epu_opts.action = ACTIONS.STATUS
epumgmt_run(epu_opts)
new_iaas_ids = [c.iaasid for c in cyvm_a]
orig_iaas_ids.sort()
new_iaas_ids.sort()
if orig_iaas_ids == new_iaas_ids:
raise Exception("The iaas ids did not change: %s -> %s" % (str(orig_iaas_ids), str(new_iaas_ids)))
except Exception, ex:
traceback.print_tb(sys.exc_info()[2])
print "========= ERROR ===================="
print ex
print "============================="
error_msg = str(ex)
rc = 1
def main(argv=sys.argv[1:]):
service_name = os.environ['EPU_SERVICE']
worker_count = 2
runname = str(uuid.uuid1()).replace("-", "")
conf = os.path.join(os.environ['EPUMGMT_HOME'], "etc/epumgmt/main.conf")
epu_opts = EPUMgmtOpts(name=runname, conf_file=conf)
my_vars_file = os.environ['EPU_TEST_VARS']
epu_opts.jsonvars = my_vars_file
#epu_action.set_logfile(os.path.join(os.environ['EPUMGMT_HOME'], "tests/tests.logs"))
cyvm_a = []
(c, p, ac) = epumgmt.api.get_common(opts=epu_opts)
persist = Persistence(p, c)
persist.validate()
cm = persist.cdb
try:
epu_opts.haservice = "provisioner"
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
epu_opts.haservice = service_name
epu_opts.action = ACTIONS.CREATE
epumgmt_run(epu_opts)
epu_opts.haservice = None
s_retry_count = 10
pre_kill_len = 0
retry_count = s_retry_count
while pre_kill_len != 2 + worker_count:
print "finding workers"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
cm.commit()
pre_kill_len = len(cyvm_a)
retry_count = retry_count - 1
if retry_count < 0:
raise Exception("No new workers found after %d tries" % (s_retry_count))
time.sleep(10)
print "make sure hostnames have been achieved"
done = False
while not done:
done = True
epu_opts.action = ACTIONS.LOGFETCH
epumgmt_run(epu_opts)
print "finding workers hostnames"
epu_opts.action = ACTIONS.FIND_WORKERS_ONCE
epumgmt_run(epu_opts)
cyvm_a = cm.get_iaas_by_runname(runname)
for c in cyvm_a:
if c.hostname == None or c.hostname.strip() == "":
done = False
cm.commit()
time.sleep(10)
# now check the logs to make sure the init events are there
types = ["provisioner", "worker", service_name]
for t in types:
vm_a = cm.get_vms_by_type(runname, t)
cm.commit()
for vm in vm_a:
found = False
for e in vm.events:
if e.name == 'init_end':
found = True
if not found:
raise Exception("init_end not found for %s" % (t))
except Exception, ex:
print ex
return 1
