From 3bb103e53c2ca0c187b789fda84a7eb17a89c3f3 Mon Sep 17 00:00:00 2001
From: LeoQi <QibinZheng2014@tongji.edu.cn>
Date: Fri, 20 Jul 2018 02:47:26 +0800
Subject: provide REST api for frontend of testing-scheduler

JIRA: BOTTLENECK-235

the REST api code in backend part.

Change-Id: I1f53ed5f0f87e6908ff4fd27f752ec2c185dc9be
Signed-off-by: Zheng Qibin <QibinZheng2014@tongji.edu.cn>
---
 testing-scheduler/server/src/rest/__init__.py      |  10 +
 testing-scheduler/server/src/rest/router.py        | 485 +++++++++++++++++++++
 .../server/src/rest/test_service_demo.py           |  77 ++++
 3 files changed, 572 insertions(+)
 create mode 100644 testing-scheduler/server/src/rest/__init__.py
 create mode 100644 testing-scheduler/server/src/rest/router.py
 create mode 100644 testing-scheduler/server/src/rest/test_service_demo.py

(limited to 'testing-scheduler/server')

diff --git a/testing-scheduler/server/src/rest/__init__.py b/testing-scheduler/server/src/rest/__init__.py
new file mode 100644
index 00000000..b5139031
--- /dev/null
+++ b/testing-scheduler/server/src/rest/__init__.py
@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+##############################################################################
+# Copyright (c) 2018 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
+##############################################################################
diff --git a/testing-scheduler/server/src/rest/router.py b/testing-scheduler/server/src/rest/router.py
new file mode 100644
index 00000000..b9cbd928
--- /dev/null
+++ b/testing-scheduler/server/src/rest/router.py
@@ -0,0 +1,485 @@
+##############################################################################
+# Copyright (c) 2018 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
+##############################################################################
+
+from flask import Flask
+from flask import jsonify
+from flask import request
+from flask_cors import CORS
+import os
+import json
+import time
+import pyaml
+import yaml
+import traceback
+
+import src.test_parser as test_parser
+
+
+BASE_DIR = os.path.abspath(os.path.dirname(__file__))
+TESTSUITE_DIR = os.path.join(BASE_DIR, "..", "..", "test", "test_case")
+SERVICE_DIR = os.path.join(BASE_DIR, "..", "env", "service")
+CONTEXT_FILE_DIR = os.path.join(BASE_DIR, "..", "env", "context",
+                                "context.yaml")
+app = Flask(__name__)
+CORS(app)
+
+
+###############
+# 1. EXECUTE API
+###########################################################################
+@app.route("/")
+def hello():
+    return "Hello, World! This is a greet from parser." + SERVICE_DIR
+
+
+@app.route("/execute/testcase", methods=['POST'])
+def runTestcase():
+    suiteName = request.values.get('suiteName')
+    caseName = request.values.get('caseName')
+    try:
+        casePath = os.path.join(TESTSUITE_DIR, suiteName, caseName)
+        if os.path.exists(casePath):
+            workflowId = test_parser.parse(casePath)
+            if workflowId is None or workflowId == '':
+                return jsonify({"code": 500, "error": "Server Error."})
+            return jsonify({"code": 200, "result": {"workflowId": workflowId}})
+        else:
+            return jsonify({"code": 300, "error": "no such test case:  %s" %
+                           (os.path.join(suiteName, caseName))})
+    except BaseException, e:
+        return returnServerError(e)
+
+
+@app.route("/story-content")
+def getStoryContent():
+    try:
+        story_name = request.args['story']
+        service_name = request.args['service']
+        storyFileDir = os.path.join("/tmp", "generate_workflow.json")
+        with open(storyFileDir, "r") as f:
+            storyContent = f.read()
+    except BaseException, e:
+        return returnServerError(e)
+
+    result = {"code": 200, "result":
+              {"service": service_name, "story": story_name,
+               "content": storyContent}}
+    return jsonify(result)
+
+
+###############
+# 2. TESTCASE CRUD
+###########################################################################
+@app.route("/testsuite/list")
+def getAllSuite():
+    res = []
+    id = 1
+    try:
+        for fileName in os.listdir(TESTSUITE_DIR):
+            suiteInfo = {}
+            suiteInfo["id"] = id
+            suiteInfo["testsuite"] = fileName
+            res.append(suiteInfo)
+            id = id + 1
+    except BaseException, e:
+        print e
+        app.logger.error(traceback.format_exc())
+        return jsonify({"code": 500, "error": "Server error"})
+
+    return jsonify({"code": 200, "result": res})
+
+
+@app.route("/testsuite/content")
+def getSuiteContent():
+    res = []
+    id = 1
+    try:
+        suiteName = request.values.get("suiteName")
+        exSuitePath = os.path.join(TESTSUITE_DIR, suiteName)
+        if os.path.exists(exSuitePath):
+            for fileName in os.listdir(exSuitePath):
+                tcInfo = {}
+                tcInfo["id"] = id
+                tcInfo["testcase"] = fileName
+                res.append(tcInfo)
+                id = id + 1
+        else:
+            return jsonify({"code": 300, "error": "no such test suite!"})
+    except BaseException, e:
+        print e
+        app.logger.error(traceback.format_exc())
+        return jsonify({"code": 500, "error": "Server error"})
+
+    return jsonify({"code": 200, "result": res})
+
+
+@app.route("/testcase/content")
+def getTCContent():
+    res = ""
+    editorRes = ""
+    try:
+        suiteName = request.values.get("suiteName")
+        caseName = request.values.get("caseName")
+        casePath = os.path.join(suiteName, caseName)
+        casePath = os.path.join(TESTSUITE_DIR, casePath)
+        if os.path.exists(casePath):
+            with open(casePath, "r") as f:
+                fileContent = f.read()
+            res = fileContent
+            editorRes = test_parser.getWebTestcase(yaml.load(res))
+        else:
+            return jsonify({"code": 300, "error": "no such file!"})
+    except BaseException, e:
+        print e
+        app.logger.error(traceback.format_exc())
+        return jsonify({"code": 500, "error": "Server error"})
+
+    return jsonify({"code": 200, "result":
+                    {"content": res, "editorContent": editorRes}})
+
+
+@app.route("/testsuite/new", methods=['POST'])
+def addNewSuite():
+    try:
+        suiteName = request.values.get("suiteName")
+        for fileName in os.listdir(TESTSUITE_DIR):
+            if fileName == suiteName:
+                return jsonify({"code": 300,
+                                "error": "testsuite already exists!"})
+        testSuitePath = os.path.join(TESTSUITE_DIR, suiteName)
+        os.mkdir(testSuitePath)
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 200, "result": "ok"})
+
+
+@app.route("/testsuite/delete", methods=['POST'])
+def deleteSuite():
+    try:
+        suiteName = request.values.get("suiteName")
+        for fileName in os.listdir(TESTSUITE_DIR):
+            if fileName == suiteName:
+                testSuitePath = os.path.join(TESTSUITE_DIR, fileName)
+                del_file(testSuitePath)
+                os.rmdir(testSuitePath)
+                return jsonify({"code": 200, "result": "ok"})
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 300, "error": "no such testsuite!"})
+
+
+def del_file(path):
+    for i in os.listdir(path):
+        path_file = os.path.join(path, i)
+        if os.path.isfile(path_file):
+            os.remove(path_file)
+        else:
+            del_file(path_file)
+
+
+@app.route("/testcase/new", methods=['POST'])
+def createTestcase():
+    try:
+        suiteName = request.values.get("suiteName")
+        caseName = request.values.get("caseName")
+        exSuitePath = os.path.join(TESTSUITE_DIR, suiteName)
+        if os.path.exists(exSuitePath):
+            for fileName in os.listdir(exSuitePath):
+                if fileName == caseName:
+                    return jsonify({"code": 301,
+                                    "error": "testcase already exists!"})
+            casePath = os.path.join(exSuitePath, caseName)
+            with open(casePath, "w") as f:
+                # the next line is a placeholder.
+                print f
+        else:
+            return jsonify({"code": 300, "error": "no such test suite!"})
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 200, "result": "ok"})
+
+
+@app.route("/testcase/delete", methods=['POST'])
+def deleteTestcase():
+    try:
+        suiteName = request.values.get("suiteName")
+        caseName = request.values.get("caseName")
+        exSuitePath = os.path.join(TESTSUITE_DIR, suiteName)
+        if os.path.exists(exSuitePath):
+            for fileName in os.listdir(exSuitePath):
+                if fileName == caseName:
+                    casePath = os.path.join(exSuitePath, caseName)
+                    os.remove(casePath)
+                    return jsonify({"code": 200, "result": "ok"})
+            return jsonify({"code": 301, "error": "no such test case!"})
+        else:
+            return jsonify({"code": 300, "error": "no such test suite!"})
+    except BaseException, e:
+        return returnServerError(e)
+
+
+@app.route("/testcase/save", methods=["POST"])
+def saveTCContent():
+    try:
+        suiteName = request.values.get("suiteName")
+        caseName = request.values.get("caseName")
+        stepList = json.loads(request.values.get("stepList"))
+        subflowList = json.loads(request.values.get("subflowList"))
+        mainOrdersList = json.loads(request.values.get("mainOrdersList"))
+        jsonObj = {"stepList": stepList, "subflowList": subflowList,
+                   "mainOrdersList": mainOrdersList}
+        parseData = test_parser.parseWebTestcase(jsonObj)
+
+        casePath = os.path.join(suiteName, caseName)
+        casePath = os.path.join(TESTSUITE_DIR, casePath)
+        if os.path.exists(casePath):
+            with open(casePath, "w") as f:
+                pyaml.dump(parseData, f, safe=True)
+        else:
+            return jsonify({"code": 300, "error": "no such file!"})
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 200, "result": "save success"})
+
+
+###############
+# 3.1 API FOR SERVICE
+############################################################
+@app.route("/service/list")
+def getAllServices():
+    res = []
+    try:
+        for fileName in os.listdir(SERVICE_DIR):
+            serviceName = os.path.splitext(fileName)[0]
+            res.append(serviceName)
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 200, "result": res})
+
+
+@app.route("/service/content")
+def getServiceContent():
+    res = {}
+    try:
+        serviceName = request.values.get("serviceName")
+        for fileName in os.listdir(SERVICE_DIR):
+            if serviceName == os.path.splitext(fileName)[0]:
+                res["actions"] = []
+                filePath = os.path.join(SERVICE_DIR, fileName)
+                with open(filePath, "r") as f:
+                    content = yaml.load(f)
+                    apisArr = content[serviceName]['apis']
+                    for i in range(len(apisArr)):
+                        apisArr[i].pop("method")
+                        apisArr[i].pop("baseuri")
+                    res["actions"] = apisArr
+    except BaseException, e:
+        return returnServerError(e)
+
+    if res == {}:
+        return jsonify({"code": 300, "error": "no such service!"})
+
+    return jsonify({"code": 200, "result": res})
+
+
+def paramTransform(paramDict):
+    res = []
+    for (key, value) in paramDict.items():
+        paramJson = {}
+        paramJson["name"] = key
+        paramJson["description"] = value["help"]
+        if "params" in value:
+            paramJson["params"] = paramTransform(value["params"])
+        res.append(paramJson)
+    return res
+
+
+@app.route("/service/action_response")
+def actionResponse():
+    res = {}
+    try:
+        serviceName = request.values.get("serviceName")
+        actionName = request.values.get("actionName")
+        for fileName in os.listdir(SERVICE_DIR):
+            if serviceName == os.path.splitext(fileName)[0]:
+                res["responseParams"] = []
+                filePath = os.path.join(SERVICE_DIR, fileName)
+                with open(filePath, "r") as f:
+                    content = yaml.load(f)
+                    apisArr = content[serviceName]['apis']
+                for i in range(len(apisArr)):
+                    if actionName == apisArr[i]['name'] and (
+                       "response" in apisArr[i]):
+                        res["responseParams"] = apisArr[i]["response"]
+    except BaseException, e:
+        return returnServerError(e)
+    if res == {}:
+        return jsonify({"code": 300, "error": "no such service!"})
+    return jsonify({"code": 200, "result": res})
+
+
+###############
+# 3.2 API FOR ENVIRONMENT SERVICE AND CONTEXT
+###########################################################################
+@app.route('/env/getAllServices')
+def getAllService():
+    res = []
+    id = 1
+    try:
+        for fileName in os.listdir(SERVICE_DIR):
+            item = {}
+            item['id'] = id
+            item['name'] = os.path.splitext(fileName)[0]
+            filePath = os.path.join(SERVICE_DIR, fileName)
+            filemt = time.localtime(os.stat(filePath).st_mtime)
+            item['time'] = time.strftime("%Y-%m-%d", filemt)
+            res.append(item)
+            id = id + 1
+    except BaseException, e:
+        return returnServerError(e)
+    return jsonify({"code": 200, "result": res})
+
+
+@app.route('/env/getService')
+def getService():
+    try:
+        serviceName = request.values.get('serviceName')
+        serviceFile = serviceName + '.yaml'
+        servicePath = os.path.join(SERVICE_DIR, serviceFile)
+        if os.path.exists(servicePath):
+            with open(servicePath, "r") as f:
+                serviceDict = yaml.load(f)
+                serviceDict = serviceDict[serviceName]
+            return jsonify({"code": 200, "result": serviceDict})
+        else:
+            return jsonify({"code": 300, "error": "no such service!"})
+    except BaseException, e:
+        return returnServerError(e)
+
+
+@app.route('/env/createService', methods=['POST'])
+def createService():
+    try:
+        name = str(request.values.get('name'))
+        ip = str(request.values.get('ip'))
+        port = int(request.values.get('port'))
+        apis = json.loads(request.values.get('apis'))
+        service = {
+            name: {
+                'ip': ip,
+                'port': port,
+                'apis': apis
+            }
+        }
+        serviceJson = json.dumps(service, indent=True)
+        print serviceJson
+        app.logger.debug(service)
+
+        serviceFile = name + '.yaml'
+        servicePath = os.path.join(SERVICE_DIR, serviceFile)
+        with open(servicePath, 'w') as f:
+            pyaml.dump(service, f, safe=True)
+    except BaseException, e:
+        return returnServerError(e)
+    return jsonify({"code": 200, "result": "create success!"})
+
+
+@app.route('/env/editService', methods=['POST'])
+def editService():
+    try:
+        oldName = str(request.values.get('oldName'))
+        name = str(request.values.get('newName'))
+        ip = str(request.values.get('ip'))
+        port = int(request.values.get('port'))
+        apis = json.loads(request.values.get('apis'))
+        app.logger.debug(apis)
+        service = {
+            name: {
+                'ip': ip,
+                'port': port,
+                'apis': apis
+            }
+        }
+        serviceJson = json.dumps(service, indent=True)
+        print serviceJson
+        app.logger.debug(service)
+
+        for fileName in os.listdir(SERVICE_DIR):
+            serviceName = os.path.splitext(fileName)[0]
+            if serviceName == oldName:
+                filePath = os.path.join(SERVICE_DIR, fileName)
+                os.remove(filePath)
+
+        serviceFile = name + '.yaml'
+        servicePath = os.path.join(SERVICE_DIR, serviceFile)
+        with open(servicePath, 'w') as f:
+            pyaml.dump(service, f, safe=True)
+    except BaseException, e:
+        return returnServerError(e)
+    return jsonify({"code": 200, "result": "edit success!"})
+
+
+@app.route('/env/deleteService', methods=['POST'])
+def deleteService():
+    try:
+        name = str(request.values.get('serviceName'))
+
+        for fileName in os.listdir(SERVICE_DIR):
+            serviceName = os.path.splitext(fileName)[0]
+            if serviceName == name:
+                filePath = os.path.join(SERVICE_DIR, fileName)
+                os.remove(filePath)
+    except BaseException, e:
+        return returnServerError(e)
+    return jsonify({"code": 200, "result": "delete success!"})
+
+
+@app.route('/env/getContext')
+def getContext():
+    try:
+        with open(CONTEXT_FILE_DIR, "r") as f:
+            fileContent = f.read()
+        res = fileContent
+    except BaseException, e:
+        return returnServerError(e)
+    return jsonify({"code": 200, "result": {"context": res}})
+
+
+@app.route('/env/editContext', methods=['POST'])
+def editContext():
+    try:
+        context = request.values.get("context")
+        test = yaml.load(context)
+        print test
+        with open(CONTEXT_FILE_DIR, "w") as f:
+            f.write(context)
+    except yaml.constructor.ConstructorError, e:
+        app.logger.error(traceback.format_exc())
+        return jsonify({"code": 500, "error":
+                        "context content error: not a .yaml file!"})
+    except BaseException, e:
+        return returnServerError(e)
+
+    return jsonify({"code": 200, "result": "edit context success!"})
+###########################################################################
+
+
+def returnServerError(e, msg="Server Error"):
+    print e
+    app.logger.error(traceback.format_exc())
+    return jsonify({"code": 500, "error": msg})
+
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', port=5310)
diff --git a/testing-scheduler/server/src/rest/test_service_demo.py b/testing-scheduler/server/src/rest/test_service_demo.py
new file mode 100644
index 00000000..e6f4e38e
--- /dev/null
+++ b/testing-scheduler/server/src/rest/test_service_demo.py
@@ -0,0 +1,77 @@
+##############################################################################
+# Copyright (c) 2018 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
+##############################################################################
+
+from flask import Flask
+from flask_cors import CORS
+from flask import request
+from flask import jsonify
+import time
+import json
+from random import randint
+
+app = Flask(__name__)
+CORS(app)
+
+
+@app.route("/greet")
+def greet():
+    return "hello"
+
+
+@app.route("/answer", methods=["POST"])
+def answer():
+    app.logger.debug(request.form)
+    app.logger.debug(request.data)
+    if jsonify(request.form) != {} and 'ping' in request.form:
+        return "answer: ping is: \"" + request.form['ping'] + "\" end."
+    elif request.data != "":
+        requestDict = json.loads(request.data)
+        if 'ping' in requestDict:
+            return "answer: the ping is: \"" + requestDict['ping'] + "\" end."
+    else:
+        return "answer ping is null"
+
+
+@app.route("/answer2", methods=["POST"])
+def answer2():
+    return "ok"
+
+
+@app.route("/five")
+def sleepFiveSeconds():
+    time.sleep(5)
+    return "five: receive the request."
+
+
+@app.route("/ten")
+def sleepTenSeconds():
+    time.sleep(10)
+    return "ten: receive the request."
+
+
+@app.route("/switch")
+def switchValue():
+    value = randint(0, 10)
+    if value > 4:
+        return jsonify({'code': 200, 'result': 'A'})
+    else:
+        return jsonify({'code': 200, 'result': 'B'})
+
+
+@app.route("/switch_2")
+def switchValue_2():
+    value = randint(0, 10)
+    if value > 4:
+        return jsonify({'code': 200, 'result': 'C'})
+    else:
+        return jsonify({'code': 200, 'result': 'D'})
+
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', port=5312, debug=True)
-- 
cgit 1.2.3-korg