summaryrefslogtreecommitdiffstats
path: root/test-scheduler/server/src/rest/router.py
diff options
context:
space:
mode:
Diffstat (limited to 'test-scheduler/server/src/rest/router.py')
-rw-r--r--test-scheduler/server/src/rest/router.py485
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)