From 8edc689554bd76045a1db0f93a778cbb05ebc70e Mon Sep 17 00:00:00 2001 From: chenjiankun Date: Thu, 20 Jul 2017 01:56:24 +0000 Subject: Add API(v2) to add suite to task JIRA: YARDSTICK-740 API: /api/v2/yardstick/tasks/ METHOD: PUT PARAMS: { 'action': 'add_suite', 'args': { 'suite_name': 'opnfv_smoke', 'suite_content': suite_content } } Change-Id: I1f81c45f143b2fea636cb87f60f2f394bf1380b1 Signed-off-by: chenjiankun --- api/resources/v2/tasks.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'api/resources/v2') diff --git a/api/resources/v2/tasks.py b/api/resources/v2/tasks.py index 26da6715e..668f7648e 100644 --- a/api/resources/v2/tasks.py +++ b/api/resources/v2/tasks.py @@ -164,3 +164,26 @@ class V2Task(ApiResource): task_handler.update_attr(task_id, task_update_data) return result_handler(consts.API_SUCCESS, {'uuid': task_id}) + + def add_suite(self, args): + task_id = args['task_id'] + try: + name = args['suite_name'] + except KeyError: + return result_handler(consts.API_ERROR, 'suite_name must be provided') + + try: + content = args['suite_content'] + except KeyError: + return result_handler(consts.API_ERROR, 'suite_content must be provided') + + LOG.info('update suite info in task') + task_handler = V2TaskHandler() + task_update_data = { + 'case_name': name, + 'content': content, + 'suite': True + } + task_handler.update_attr(task_id, task_update_data) + + return result_handler(consts.API_SUCCESS, {'uuid': task_id}) -- cgit 1.2.3-korg