def test_delete_load_job(self):
"""
删除任务
"""
self.test_stop_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
code, res = Load.delete_load_job(graph_id=graph_id, job_id=job_id)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "删除任务状态码不正确,删除任务失败")
def test_get_load_task(self):
"""
查询导入任务
"""
self.test_stop_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
param = "task_id=%d" % task_id
code, res = Load.query_load_task(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "查询导入状态码不正确,查询导入任务失败")
def test_create_load_job(self):
"""
创建导入任务
"""
graph_id = ID.get_graph_id()
body = {"job_name": "load_01", "job_remarks": "第一次导入"}
code, res = Load.create_load_job(body=body, graph_id=graph_id)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "创建导入任务状态码不正确")
self.assertEqual(res['data']['job_name'], body['job_name'], "导入任务名称有误")
self.assertEqual(res['data']['job_remarks'], body['job_remarks'], "任务备注有误")
def test_retry_load(self):
"""
点击重试导入
"""
self.test_stop_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
param = "task_id=%d" % task_id
code, res = Load.retry_load(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "重试导入状态码不正确,重试导入失败")
self.assertEqual(res['data']['status'], 'RUNNING', "重试导入错误")
def test_stop_load(self):
"""
点击停止导入
"""
self.test_resume_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
param = "task_id=%d" % task_id
code, res = Load.stop_load(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "停止导入状态码不正确,停止导入失败")
self.assertEqual(res['data']['status'], 'STOPPED', "停止导入错误")
def test_pause_load(self):
"""
点击暂停导入
"""
self.test_start_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
param = "task_id=%d" % task_id
code, res = Load.pause_load(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "暂停导入状态码不正确,暂停导入失败")
self.assertEqual(res['data']['status'], 'PAUSED', "暂停导入错误")
def test_start_load(self):
"""
点击开始导入
"""
self.test_add_load_setting()
graph_id, job_id, file_id = ID.get_file_id()
param = "file_mapping_ids=%d" % file_id
code, res = Load.start_load(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "开始导入状态码不正确,开始导入失败")
self.assertEqual(res['data'][0]['status'], 'RUNNING', "开始导入错误")
def test_update_load_job(self):
"""
修改导入任务名称和备注
"""
self.test_get_load_job()
graph_id, job_id = ID.get_job_id()
body = {"job_name": "load_01_update", "job_remarks": "修改任务名称和备注"}
code, res = Load.update_load_job(body=body, graph_id=graph_id, job_id=job_id)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "修改导入任务名称和备注状态码不正确,修改导入任务名称和备注失败")
self.assertEqual(res['data']['job_name'], body['job_name'], "修改导入任务名称有误")
self.assertEqual(res['data']['job_remarks'], body['job_remarks'], "修改导入任务备注有误")
def test_resume_load(self):
"""
点击继续导入
"""
self.test_pause_load()
graph_id, job_id, file_id, task_id = ID.get_task_id()
param = "task_id=%d" % task_id
time.sleep(1)
code, res = Load.resume_load(graph_id=graph_id, job_id=job_id, param=param)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "继续导入状态码不正确,继续导入失败")
self.assertEqual(res['data']['status'], 'RUNNING', "继续导入错误")
def test_add_load_setting(self):
"""
设置导入参数
"""
self.test_add_edge_mapping()
graph_id, job_id = ID.get_job_id()
body = {
"check_vertex": False,
"insert_timeout": 60,
"max_parse_errors": 500,
"max_insert_errors": 500,
"retry_times": 3,
"retry_interval": 10
}
# 点击下一步
Step.mapping_complete_next_step(graph_id=graph_id, job_id=job_id)
code, res = Load.add_load_setting(body=body, graph_id=graph_id, job_id=job_id)
self.assertEqual(code, 200, "响应状态码不正确")
self.assertEqual(res['status'], 200, "设置导入参数状态码不正确")