From 56c2f0decfdd2865cca8091ce65e7c0c9ec4868f Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Mon, 26 Jun 2017 11:51:08 +0800
Subject: [PATCH] TestRunner: run testcase writen in separate template

---
 ate/runner.py                        | 29 ++++++++++++-
 test/data/demo_template_separate.yml | 62 ++++++++++++++++++++++++++++
 test/test_runner_v2.py               |  9 ++++
 3 files changed, 99 insertions(+), 1 deletion(-)
 create mode 100644 test/data/demo_template_separate.yml

diff --git a/ate/runner.py b/ate/runner.py
index ce2ea3d8..0508d850 100644
--- a/ate/runner.py
+++ b/ate/runner.py
@@ -1,12 +1,39 @@
 import requests
-from ate import utils, exception
+
+from ate import exception, utils
+from ate.context import Context
+from ate.testcase import TestcaseParser
+
 
 class TestRunner(object):
 
     def __init__(self):
         self.client = requests.Session()
+        self.context = Context()
+        self.testcase_parser = TestcaseParser()
+
+    def prepare(self, testcase):
+        """ prepare work before running test.
+        parse testcase with variables binds if it is a template.
+        """
+        requires = testcase.get('requires', [])
+        self.context.import_requires(requires)
+
+        function_binds = testcase.get('function_binds', {})
+        self.context.bind_functions(function_binds)
+
+        variable_binds = testcase.get('variable_binds', [])
+        self.context.bind_variables(variable_binds)
+
+        parsed_testcase = self.testcase_parser.parse(
+            testcase,
+            variables_binds=self.context.variables
+        )
+        return parsed_testcase
 
     def run_single_testcase(self, testcase):
+        testcase = self.prepare(testcase)
+
         req_kwargs = testcase['request']
 
         try:
diff --git a/test/data/demo_template_separate.yml b/test/data/demo_template_separate.yml
new file mode 100644
index 00000000..c0d8568c
--- /dev/null
+++ b/test/data/demo_template_separate.yml
@@ -0,0 +1,62 @@
+
+-
+  name: create user which does not exist
+  requires:
+    - random
+    - string
+    - hashlib
+  function_binds:
+    gen_random_string: "lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(str_len))"
+    gen_md5: "lambda *str_args: hashlib.md5(''.join(str_args).encode('utf-8')).hexdigest()"
+  variable_binds:
+    - TOKEN: debugtalk
+    - random: {"func": "gen_random_string", "args": [5]}
+    - data: '{"name": "user", "password": "123456"}'
+    - authorization: {"func": "gen_md5", "args": ["$TOKEN", "$data", "$random"]}
+    - expected_status_code: 201
+  request:
+    url: http://127.0.0.1:5000/api/users/1000
+    method: POST
+    headers:
+      Content-Type: application/json
+      authorization: "${authorization}"
+      random: "${random}"
+    data: "${data}"
+  response:
+    status_code: "${expected_status_code}"
+    headers:
+      Content-Type: application/json
+    body:
+      success: true
+      msg: user created successfully.
+
+-
+  name: create user which does not exist
+  requires:
+    - random
+    - string
+    - hashlib
+  function_binds:
+    gen_random_string: "lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(str_len))"
+    gen_md5: "lambda *str_args: hashlib.md5(''.join(str_args).encode('utf-8')).hexdigest()"
+  variable_binds:
+    - TOKEN: debugtalk
+    - random: {"func": "gen_random_string", "args": [5]}
+    - data: '{"name": "user", "password": "123456"}'
+    - authorization: {"func": "gen_md5", "args": ["$TOKEN", "$data", "$random"]}
+    - expected_status_code: 500
+  request:
+    url: http://127.0.0.1:5000/api/users/1000
+    method: POST
+    headers:
+      Content-Type: application/json
+      authorization: "${authorization}"
+      random: "${random}"
+    data: "${data}"
+  response:
+    status_code: "${expected_status_code}"
+    headers:
+      Content-Type: application/json
+    body:
+      success: true
+      msg: user created successfully.
diff --git a/test/test_runner_v2.py b/test/test_runner_v2.py
index ba6b03be..72921b0d 100644
--- a/test/test_runner_v2.py
+++ b/test/test_runner_v2.py
@@ -45,3 +45,12 @@ class TestRunnerV2(ApiServerUnittest):
         result = self.test_runner.run_testcase_suite(testcases)
         self.assertEqual(len(result), 2)
         self.assertEqual(result, [(True, {}), (True, {})])
+
+    def test_run_testcase_template_yaml(self):
+        testcase_file_path = os.path.join(
+            os.getcwd(), 'test/data/demo_template_separate.yml')
+        testcases = utils.load_testcases(testcase_file_path)
+        success, _ = self.test_runner.run_single_testcase(testcases[0])
+        self.assertTrue(success)
+        success, _ = self.test_runner.run_single_testcase(testcases[1])
+        self.assertFalse(success)