From 59e71b5e0a8c8eb62fe26e905ce46ff85d098b54 Mon Sep 17 00:00:00 2001 From: youhuo Date: Fri, 16 Sep 2022 17:12:53 +0800 Subject: [PATCH] feat: choose tone server machine --- services/plan_service.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/services/plan_service.py b/services/plan_service.py index 144528a..ec564af 100644 --- a/services/plan_service.py +++ b/services/plan_service.py @@ -211,12 +211,18 @@ async def __start_auto_task(auto_tasks): if case.tone_case in case_dict: continue case_dict[case.tone_case] = True - suite_name = case.custom_fields.get('tone_suite_name') case_name = case.custom_fields.get('tone_case_name') if suite_name not in suite_dict: suite_dict[suite_name] = list() - suite_dict[suite_name].append({'test_case': case_name}) + case_obj = {'test_case': case_name} + if task.device_ip: + case_obj['server'] = {'ip': task.device_ip, 'channel_type': 'toneagent'} + if task.cluster: + case_obj['server'] = {'ip': task.cluster} + if task.tags: + case_obj['server'] = {'tags': task.tags} + suite_dict[suite_name].append(case_obj) for key, value in suite_dict.items(): test_config.append({ 'test_suite': key, -- Gitee