def main():
if len(sys.argv) != 2:
print "USAGE: python %s <dump-file>" % (sys.argv[0], )
return
dump_file = sys.argv[1]
mock_am_conf()
service_mgr = SM.ServiceManager(g_zk_server, g_zk_root, g_zk_user, g_zk_pass)
bh_mgr = BH.BHManager(g_bhcli_path, g_bhcli_cluster)
task_context = AT.TaskContext(service_mgr, bh_mgr)
tool = BST.SchedulerTaskHelper("task_debug", task_context)
ret, err_msg, all_apps, all_machines, all_resource = tool.step_dump(g_service_id, dump_file)
if ret != 0:
print "FAIL:%s" % (err_msg, )
else:
print "SUCC"
class TestRestore:
def __init__(self):
ret = LOG.init_log(am_global.log_prefix)
if ret != 0:
print "[FATAL] InitLogger failed"
sys.exit(1)
ret = am_conf.LoadConf(
os.path.join(am_global.conf_dir, am_global.conf_file))
if ret != 0:
print "[FATAL] LoadConf failed"
sys.exit(1)
try:
self.__service_mgr = service_manager.ServiceManager(
am_conf.g_conf.zk_server, am_conf.g_conf.zk_root)
except Exception, e:
print "init service_mananger failed"
print e
traceback.print_exc()
sys.exit(1)
print "init service_manager sucessfully"
self.__service_mgr.Init()
#TODO: move to upper level
try:
bh_mgr = bhcli_helper.BHManager("./fake_bhcli_for_restore.py",
am_conf.g_conf.bh_cluster)
#bh_mgr = bhcli_helper.BHManager(am_conf.g_conf.bh_loc, am_conf.g_conf.bh_cluster)
logging.info("init bhcli manager successfully")
self.task_context_ = am_task.TaskContext(self.__service_mgr,
bh_mgr)
logging.info("init task context successfully")
self.queue_lock_ = threading.Lock()
self.task_queue_ = {}
logging.info("init task lock and queue successfully")
except Exception, e:
print "init task context failed"
print e
print traceback.print_exc()
sys.exit(1)
if __name__ == "__main__":
ret = am_log.init_log(os.path.join(am_global.log_dir, "test"))
if ret != 0:
print "[FATAL] InitLogger failed"
sys.exit(1)
conf_files = list()
conf_files.append(os.path.join(am_global.conf_dir, "am.conf"))
conf_files.append(os.path.join(am_global.conf_dir, "ip.conf.sample"))
conf_files.append(os.path.join(am_global.conf_dir, "bhcli.conf.sample"))
ret = am_conf.LoadConf(conf_files)
if ret != 0:
print "[FATAL] LoadConf failed"
sys.exit(1)
bh_mgr = bhcli_helper.BHManager(am_conf.g_conf.bh_loc,
am_conf.g_conf.bh_cluster)
#describe app
app_id = "bs_vip_0"
bh_app = bh_mgr.get_app(app_id)
ret, app_descr = bh_app.describe()
print("describe app finished, app_id=[%s], ret=[%d]") % (app_id, ret)
bh_resource = bh_mgr.get_resource(app_id)
ret, group_descr = bh_resource.describe_container_group(app_id)
print("describe resource finished, app_id=[%s], ret=[%d]") % (app_id, ret)
container_group = json.loads(group_descr)
ret, out = bh_resource.update_container_group(app_id, container_group)
print("update resource finished, app_id=[%s], ret=[%d]") % (app_id, ret)