1 files changed, 102 insertions, 0 deletions
diff --git a/dovetail/api/app/routes.py b/dovetail/api/app/routes.py
new file mode 100644
index 00000000..352d69f3
--- /dev/null
+++ b/dovetail/api/app/routes.py
@@ -0,0 +1,102 @@
+#!flask/bin/python
+
+import json
+import os
+import subprocess
+import time
+import uuid
+
+from flask import Flask, jsonify, request
+from flask_cors import CORS
+
+from app.server import Server
+
+app = Flask(__name__)
+CORS(app)
+
+
+@app.route('/api/v1/scenario/nfvi/testsuites', methods=['GET'])
+def get_all_testsuites():
+    testsuites = Server.list_testsuites()
+    return jsonify({'testsuites': testsuites}), 200
+
+
+@app.route('/api/v1/scenario/nfvi/testcases', methods=['GET'])
+def get_testcases():
+    testcases = Server.list_testcases()
+    return jsonify({'testcases': testcases}), 200
+
+
+@app.route('/api/v1/scenario/nfvi/execution', methods=['POST'])
+def run_testcases():
+    requestId = request.args.get('requestId')
+    if not requestId:
+        requestId = uuid.uuid1()
+    if os.getenv('DOVETAIL_HOME'):
+        dovetail_home = os.getenv('DOVETAIL_HOME')
+    else:
+        return 'No DOVETAIL_HOME found in env.\n', 500
+
+    server = Server(dovetail_home, requestId, request.json)
+
+    msg, ret = server.set_conf_files()
+    if not ret:
+        return msg, 500
+
+    msg, ret = server.set_vm_images()
+    if not ret:
+        return msg, 500
+
+    input_str = server.parse_request()
+
+    repo_dir = os.path.abspath(os.path.join(os.path.dirname(__file__),
+                               os.pardir, os.pardir))
+    run_script = os.path.join(repo_dir, 'run.py')
+
+    cmd = 'python3 {} {}'.format(run_script, input_str)
+    api_home = os.path.join(dovetail_home, str(requestId))
+    subprocess.Popen(cmd, shell=True, env={'DOVETAIL_HOME': api_home,
+                     'LC_ALL': 'C.UTF-8', 'LANG': 'C.UTF-8'})
+
+    testcases_file = os.path.join(dovetail_home, str(requestId),
+                                  'results', 'testcases.json')
+    for loop in range(60):
+        if not os.path.isfile(testcases_file):
+            time.sleep(1)
+        else:
+            break
+    else:
+        return 'Can not get file testcases.json.\n', 500
+
+    with open(testcases_file, "r") as f:
+        for jsonfile in f:
+            data = json.loads(jsonfile)
+        testcases = data['testcases']
+        testsuite = data['testsuite']
+
+    result = server.get_execution_status(testsuite, testcases, testcases)
+
+    return jsonify({'result': result}), 200
+
+
+@app.route('/api/v1/scenario/nfvi/execution/status/<exec_id>',
+           methods=['POST'])
+def get_testcases_status(exec_id):
+    if 'testcase' not in request.json:
+        return 'Need testcases list as input.\n', 400
+
+    testcases = request.json['testcase']
+    dovetail_home = os.getenv('DOVETAIL_HOME')
+
+    server = Server(dovetail_home, exec_id, request.json)
+    testcases_file = os.path.join(dovetail_home, str(exec_id),
+                                  'results', 'testcases.json')
+    with open(testcases_file, "r") as f:
+        for jsonfile in f:
+            data = json.loads(jsonfile)
+        testsuite = data['testsuite']
+
+    result = server.get_execution_status(testsuite, testcases,
+                                         data['testcases'])
+
+    return jsonify({'result': result}), 200