Update test framework: fix run_tests.py to support all test files, add auto-import-check for test files

base_framework/platform_tools/IDF/common/apiResponse.py

@@ -0,0 +1,54 @@
+# encoding: UTF-8
+
+from flask import make_response
+import json
+from ..const import RES_CODE
+
+
+class ApiResponse(object):
+    def __init__(self):
+        self.success = False
+        self.code = ''
+        self.message = ''
+        self.data = {}
+
+    @staticmethod
+    def build_success(code=20000, message='', data=None):
+        if data is None:
+            data = {}
+        response = ApiResponse()
+        response.success = True
+        response.code = code
+        response.message = message
+        response.data = data
+        return response.cors_response(make_response(json.dumps(response, default=obj_2_json)))
+
+    @staticmethod
+    def build_failure(code, msg='', data=None):
+        response = ApiResponse()
+        if data is None:
+            data = {}
+        if not msg:
+            response.message = RES_CODE[code]
+        else:
+            response.message = msg
+        response.success = False
+        response.code = code
+        response.data = data
+        return response.cors_response(make_response(json.dumps(response, default=obj_2_json)))
+
+    @staticmethod
+    def cors_response(res):
+        res.headers['Access-Control-Allow-Origin'] = '*'
+        res.headers['Access-Control-Allow-Methods'] = 'POST'
+        res.headers['Access-Control-Allow-Headers'] = 'x-requested-with,content-type'
+        return res
+
+
+def obj_2_json(obj):
+    return {
+        'success': obj.success,
+        'code': obj.code,
+        'message': obj.message,
+        'data': obj.data
+    }