utils: testcases loader

ate/exception.py

@@ -0,0 +1,16 @@
+#coding: utf-8
+from termcolor import colored
+
+class MyBaseError(BaseException):
+    def __init__(self, msg):
+        self.msg = msg
+        self.color_msg = colored(msg, 'red', attrs=['bold'])
+
+    def __repr__(self):
+        return self.msg
+
+    def __str__(self):
+        return self.color_msg
+
+class ParamsError(MyBaseError):
+    pass

ate/utils.py

@@ -0,0 +1,22 @@
+import json
+import yaml
+import os.path
+from ate.exception import ParamsError
+
+def load_yaml_file(yaml_file):
+    with open(yaml_file, 'r+') as stream:
+        return yaml.load(stream)
+
+def load_json_file(json_file):
+    with open(json_file) as data_file:
+        return json.load(data_file)
+
+def load_testcases(testcase_file_path):
+    file_suffix = os.path.splitext(testcase_file_path)[1]
+    if file_suffix == '.json':
+        return load_json_file(testcase_file_path)
+    elif file_suffix in ['.yaml', '.yml']:
+        return load_yaml_file(testcase_file_path)
+    else:
+        # '' or other suffix
+        raise ParamsError("Bad testcase file name!")