1 files changed, 65 insertions, 12 deletions

diff --git a/api/views.py b/api/views.py
index 883091222..928d8e9eb 100644
--- a/api/views.py
+++ b/api/views.py
@@ -1,29 +1,82 @@
-import json
+##############################################################################
+# Copyright (c) 2016 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
+##############################################################################
 import logging
+import os
 
 from flask import request
 from flask_restful import Resource
+from flasgger.utils import swag_from
 
 from api.utils import common as common_utils
+from api.swagger import models
 from api.actions import test as test_action
-from api import conf
+from api.actions import samples as samples_action
+from api.actions import result as result_action
+from api.actions import env as env_action
 
 logger = logging.getLogger(__name__)
 
 
-class Test(Resource):
+TestCaseActionModel = models.TestCaseActionModel
+TestCaseActionArgsModel = models.TestCaseActionArgsModel
+TestCaseActionArgsOptsModel = models.TestCaseActionArgsOptsModel
+TestCaseActionArgsOptsTaskArgModel = models.TestCaseActionArgsOptsTaskArgModel
+
+
+class Release(Resource):
+    @swag_from(os.getcwd() + '/swagger/docs/testcases.yaml')
     def post(self):
         action = common_utils.translate_to_str(request.json.get('action', ''))
         args = common_utils.translate_to_str(request.json.get('args', {}))
         logger.debug('Input args is: action: %s, args: %s', action, args)
 
-        if action not in conf.TEST_ACTION:
-            logger.error('Wrong action')
-            result = {
-                'status': 'error',
-                'message': 'wrong action'
-            }
-            return json.dumps(result)
+        try:
+            return getattr(test_action, action)(args)
+        except AttributeError:
+            return common_utils.error_handler('Wrong action')
+
+
+class Samples(Resource):
+    def post(self):
+        action = common_utils.translate_to_str(request.json.get('action', ''))
+        args = common_utils.translate_to_str(request.json.get('args', {}))
+        logger.debug('Input args is: action: %s, args: %s', action, args)
+
+        try:
+            return getattr(samples_action, action)(args)
+        except AttributeError:
+            return common_utils.error_handler('Wrong action')
+
+
+ResultModel = models.ResultModel
+
+
+class Results(Resource):
+    @swag_from(os.getcwd() + '/swagger/docs/results.yaml')
+    def get(self):
+        args = common_utils.translate_to_str(request.args)
+        action = args.get('action', '')
+        logger.debug('Input args is: action: %s, args: %s', action, args)
+
+        try:
+            return getattr(result_action, action)(args)
+        except AttributeError:
+            return common_utils.error_handler('Wrong action')
+
+
+class Env(Resource):
+    def post(self):
+        action = common_utils.translate_to_str(request.json.get('action', ''))
+        args = common_utils.translate_to_str(request.json.get('args', {}))
+        logger.debug('Input args is: action: %s, args: %s', action, args)
 
-        method = getattr(test_action, action)
-        return method(args)
+        try:
+            return getattr(env_action, action)(args)
+        except AttributeError:
+            return common_utils.error_handler('Wrong action')