diff options
Diffstat (limited to 'test-scheduler/server/src/rest/router.py')
| -rw-r--r-- | test-scheduler/server/src/rest/router.py | 485 |
1 files changed, 485 insertions, 0 deletions
diff --git a/test-scheduler/server/src/rest/router.py b/test-scheduler/server/src/rest/router.py new file mode 100644 index 00000000..1ceeedd0 --- /dev/null +++ b/test-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) |