add support for import module custom functions

test/data/custom_functions.py

@@ -0,0 +1,42 @@
+import hashlib
+import json
+import random
+import string
+
+try:
+    string_type = basestring
+    PYTHON_VERSION = 2
+except NameError:
+    string_type = str
+    PYTHON_VERSION = 3
+
+
+def gen_random_string(str_len):
+    return ''.join(
+        random.choice(string.ascii_letters + string.digits) for _ in range(str_len))
+
+def gen_md5(*args):
+    args = [handle_req_data(item) for item in args]
+    return hashlib.md5("".join(args).encode('utf-8')).hexdigest()
+
+def handle_req_data(data):
+
+    if PYTHON_VERSION == 3 and isinstance(data, bytes):
+        # In Python3, convert bytes to str
+        data = data.decode('utf-8')
+
+    if not data:
+        return data
+
+    if isinstance(data, str):
+        # check if data in str can be converted to dict
+        try:
+            data = json.loads(data)
+        except ValueError:
+            pass
+
+    if isinstance(data, dict):
+        # sort data in dict with keys, then convert to str
+        data = json.dumps(data, sort_keys=True)
+
+    return data

test/data/demo_import_functions.yml

@@ -0,0 +1,41 @@
+- config:
+    name: "create user testsets."
+    import_module_functions:
+        - test.data.custom_functions
+    variable_binds:
+        - TOKEN: debugtalk
+        - json: {"name": "user", "password": "123456"}
+        - random: {"func": "gen_random_string", "args": [5]}
+        - authorization: {"func": "gen_md5", "args": ["${TOKEN}", "${json}", "${random}"]}
+
+- test:
+    name: create user which does not exist
+    variable_binds:
+        - json: {"name": "user", "password": "123456"}
+    request:
+        url: http://127.0.0.1:5000/api/users/1000
+        method: POST
+        headers:
+            Content-Type: application/json
+            authorization: "${authorization}"
+            random: "${random}"
+        json: "${json}"
+    validators:
+        - {"check": "status_code", "comparator": "eq", "expected": 201}
+        - {"check": "content.success", "comparator": "eq", "expected": true}
+
+- test:
+    name: create user which does not exist
+    variable_binds:
+        - json: {"name": "user", "password": "123456"}
+    request:
+        url: http://127.0.0.1:5000/api/users/1000
+        method: POST
+        headers:
+            Content-Type: application/json
+            authorization: "${authorization}"
+            random: "${random}"
+        json: "${json}"
+    validators:
+        - {"check": "status_code", "comparator": "eq", "expected": 500}
+        - {"check": "content.success", "comparator": "eq", "expected": false}