7 files changed, 347 insertions, 0 deletions
diff --git a/functest/api/resources/__init__.py b/functest/api/resources/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/functest/api/resources/__init__.py
diff --git a/functest/api/resources/v1/__init__.py b/functest/api/resources/v1/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/functest/api/resources/v1/__init__.py
diff --git a/functest/api/resources/v1/creds.py b/functest/api/resources/v1/creds.py
new file mode 100644
index 00000000..45e4559f
--- /dev/null
+++ b/functest/api/resources/v1/creds.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+"""
+Resources to handle openstack related requests
+"""
+
+import collections
+import logging
+
+from flask import jsonify
+
+from functest.api.base import ApiResource
+from functest.api.common import api_utils
+from functest.cli.commands.cli_os import OpenStack
+from functest.utils import openstack_utils as os_utils
+from functest.utils.constants import CONST
+
+LOGGER = logging.getLogger(__name__)
+
+
+class V1Creds(ApiResource):
+    """ V1Creds Resource class"""
+
+    def get(self):  # pylint: disable=no-self-use
+        """ Get credentials """
+        os_utils.source_credentials(CONST.__getattribute__('openstack_creds'))
+        credentials_show = OpenStack.show_credentials()
+        return jsonify(credentials_show)
+
+    def post(self):
+        """ Used to handle post request """
+        return self._dispatch_post()
+
+    def update_openrc(self, args):  # pylint: disable=no-self-use
+        """ Used to update the OpenStack RC file """
+        try:
+            openrc_vars = args['openrc']
+        except KeyError:
+            return api_utils.result_handler(
+                status=0, data='openrc must be provided')
+        else:
+            if not isinstance(openrc_vars, collections.Mapping):
+                return api_utils.result_handler(
+                    status=0, data='args should be a dict')
+
+        lines = ['export {}={}\n'.format(k, v) for k, v in openrc_vars.items()]
+
+        rc_file = CONST.__getattribute__('openstack_creds')
+        with open(rc_file, 'w') as creds_file:
+            creds_file.writelines(lines)
+
+        LOGGER.info("Sourcing the OpenStack RC file...")
+        try:
+            os_utils.source_credentials(rc_file)
+        except Exception as err:  # pylint: disable=broad-except
+            LOGGER.exception('Failed to source the OpenStack RC file')
+            return api_utils.result_handler(status=0, data=str(err))
+
+        return api_utils.result_handler(
+            status=0, data='Update openrc successfully')
diff --git a/functest/api/resources/v1/envs.py b/functest/api/resources/v1/envs.py
new file mode 100644
index 00000000..9c455198
--- /dev/null
+++ b/functest/api/resources/v1/envs.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+#
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+"""
+Resources to handle environment related requests
+"""
+
+from flask import jsonify
+
+from functest.api.base import ApiResource
+from functest.cli.commands.cli_env import Env
+from functest.api.common import api_utils
+import functest.utils.functest_utils as ft_utils
+
+
+class V1Envs(ApiResource):
+    """ V1Envs Resource class"""
+
+    def get(self):  # pylint: disable=no-self-use
+        """ Get environment """
+        environment_show = Env().show()
+        return jsonify(environment_show)
+
+    def post(self):
+        """ Used to handle post request """
+        return self._dispatch_post()
+
+    def prepare(self, args):  # pylint: disable=no-self-use, unused-argument
+        """ Prepare environment """
+        try:
+            ft_utils.execute_command("prepare_env start")
+        except Exception as err:  # pylint: disable=broad-except
+            return api_utils.result_handler(status=1, data=str(err))
+        return api_utils.result_handler(
+            status=0, data="Prepare env successfully")
diff --git a/functest/api/resources/v1/tasks.py b/functest/api/resources/v1/tasks.py
new file mode 100644
index 00000000..7086e707
--- /dev/null
+++ b/functest/api/resources/v1/tasks.py
@@ -0,0 +1,58 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+"""
+Resources to retrieve the task results
+"""
+
+
+import json
+import logging
+import uuid
+
+from flask import jsonify
+
+from functest.api.base import ApiResource
+from functest.api.common import api_utils
+from functest.api.database.v1.handlers import TasksHandler
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+class V1Tasks(ApiResource):
+    """ V1Tasks Resource class"""
+
+    def get(self, task_id):  # pylint: disable=no-self-use
+        """ GET the result of the task id """
+        try:
+            uuid.UUID(task_id)
+        except ValueError:
+            return api_utils.result_handler(status=1, data='Invalid task id')
+
+        task_handler = TasksHandler()
+        try:
+            task = task_handler.get_task_by_taskid(task_id)
+        except ValueError:
+            return api_utils.result_handler(status=1, data='No such task id')
+
+        status = task.status
+        LOGGER.debug('Task status is: %s', status)
+
+        if status not in ['IN PROGRESS', 'FAIL', 'FINISHED']:
+            return api_utils.result_handler(status=1,
+                                            data='internal server error')
+        if status == 'IN PROGRESS':
+            result = {'status': status, 'result': ''}
+        elif status == 'FAIL':
+            result = {'status': status, 'error': task.error}
+        else:
+            result = {'status': status, 'result': json.loads(task.result)}
+
+        return jsonify(result)
diff --git a/functest/api/resources/v1/testcases.py b/functest/api/resources/v1/testcases.py
new file mode 100644
index 00000000..f146c24c
--- /dev/null
+++ b/functest/api/resources/v1/testcases.py
@@ -0,0 +1,115 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+"""
+Resources to handle testcase related requests
+"""
+
+import os
+import logging
+import uuid
+
+from flask import abort, jsonify
+
+from functest.api.base import ApiResource
+from functest.api.common import api_utils, thread
+from functest.cli.commands.cli_testcase import Testcase
+from functest.api.database.v1.handlers import TasksHandler
+from functest.utils.constants import CONST
+import functest.utils.functest_utils as ft_utils
+
+LOGGER = logging.getLogger(__name__)
+
+
+class V1Testcases(ApiResource):
+    """ V1Testcases Resource class"""
+
+    def get(self):  # pylint: disable=no-self-use
+        """ GET all testcases """
+        testcases_list = Testcase().list()
+        result = {'testcases': testcases_list.split('\n')[:-1]}
+        return jsonify(result)
+
+
+class V1Testcase(ApiResource):
+    """ V1Testcase Resource class"""
+
+    def get(self, testcase_name):  # pylint: disable=no-self-use
+        """ GET the info of one testcase"""
+        testcase = Testcase().show(testcase_name)
+        if not testcase:
+            abort(404, "The test case '%s' does not exist or is not supported"
+                  % testcase_name)
+        testcase_info = api_utils.change_obj_to_dict(testcase)
+        dependency_dict = api_utils.change_obj_to_dict(
+            testcase_info.get('dependency'))
+        testcase_info.pop('name')
+        testcase_info.pop('dependency')
+        result = {'testcase': testcase_name}
+        result.update(testcase_info)
+        result.update({'dependency': dependency_dict})
+        return jsonify(result)
+
+    def post(self):
+        """ Used to handle post request """
+        return self._dispatch_post()
+
+    def run_test_case(self, args):
+        """ Run a testcase """
+        try:
+            case_name = args['testcase']
+        except KeyError:
+            return api_utils.result_handler(
+                status=1, data='testcase name must be provided')
+
+        task_id = str(uuid.uuid4())
+
+        task_args = {'testcase': case_name, 'task_id': task_id}
+
+        task_args.update(args.get('opts', {}))
+
+        task_thread = thread.TaskThread(self._run, task_args, TasksHandler())
+        task_thread.start()
+
+        results = {'testcase': case_name, 'task_id': task_id}
+        return jsonify(results)
+
+    def _run(self, args):  # pylint: disable=no-self-use
+        """ The built_in function to run a test case """
+
+        case_name = args.get('testcase')
+
+        if not os.path.isfile(CONST.__getattribute__('env_active')):
+            raise Exception("Functest environment is not ready.")
+        else:
+            try:
+                cmd = "run_tests -t {}".format(case_name)
+                runner = ft_utils.execute_command(cmd)
+            except Exception:  # pylint: disable=broad-except
+                result = 'FAIL'
+                LOGGER.exception("Running test case %s failed!", case_name)
+            if runner == os.EX_OK:
+                result = 'PASS'
+            else:
+                result = 'FAIL'
+
+            env_info = {
+                'installer': CONST.__getattribute__('INSTALLER_TYPE'),
+                'scenario': CONST.__getattribute__('DEPLOY_SCENARIO'),
+                'build_tag': CONST.__getattribute__('BUILD_TAG'),
+                'ci_loop': CONST.__getattribute__('CI_LOOP')
+            }
+            result = {
+                'task_id': args.get('task_id'),
+                'case_name': case_name,
+                'env_info': env_info,
+                'result': result
+            }
+
+            return {'result': result}
diff --git a/functest/api/resources/v1/tiers.py b/functest/api/resources/v1/tiers.py
new file mode 100644
index 00000000..71a98bea
--- /dev/null
+++ b/functest/api/resources/v1/tiers.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+"""
+Resources to handle tier related requests
+"""
+
+import re
+
+from flask import abort, jsonify
+
+from functest.api.base import ApiResource
+from functest.cli.commands.cli_tier import Tier
+
+
+class V1Tiers(ApiResource):
+    """ V1Tiers Resource class """
+
+    def get(self):
+        # pylint: disable=no-self-use
+        """ GET all tiers """
+        tiers_list = Tier().list()
+        data = re.split("[\n\t]", tiers_list)
+        data = [i.strip() for i in data if i != '']
+        data_dict = dict()
+        for i in range(len(data) / 2):
+            one_data = {data[i * 2]: data[i * 2 + 1]}
+            if i == 0:
+                data_dict = one_data
+            else:
+                data_dict.update(one_data)
+        result = {'tiers': data_dict}
+        return jsonify(result)
+
+
+class V1Tier(ApiResource):
+    """ V1Tier Resource class """
+
+    def get(self, tier_name):  # pylint: disable=no-self-use
+        """ GET the info of one tier """
+        testcases = Tier().gettests(tier_name)
+        if not testcases:
+            abort(404, "The tier with name '%s' does not exist." % tier_name)
+        tier_info = Tier().show(tier_name)
+        tier_info.__dict__.pop('name')
+        tier_info.__dict__.pop('tests_array')
+        result = {'tier': tier_name, 'testcases': testcases}
+        result.update(tier_info.__dict__)
+        return jsonify(result)
+
+
+class V1TestcasesinTier(ApiResource):
+    """ V1TestcasesinTier Resource class """
+
+    def get(self, tier_name):  # pylint: disable=no-self-use
+        """ GET all testcases within given tier """
+        testcases = Tier().gettests(tier_name)
+        if not testcases:
+            abort(404, "The tier with name '%s' does not exist." % tier_name)
+        result = {'tier': tier_name, 'testcases': testcases}
+        return jsonify(result)