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, "设置导入参数状态码不正确")