split ate/main.py to ate/cli.py and ate/task.py

ate/task.py

@@ -0,0 +1,53 @@
+import unittest
+
+from ate import runner, utils
+
+
+class ApiTestCase(unittest.TestCase):
+    """ create a testcase.
+    """
+    def __init__(self, test_runner, testcase):
+        super(ApiTestCase, self).__init__()
+        self.test_runner = test_runner
+        self.testcase = testcase
+
+    def runTest(self):
+        """ run testcase and check result.
+        """
+        result = self.test_runner.run_test(self.testcase)
+        self.assertEqual(result, (True, []))
+
+def create_suite(testset):
+    """ create test suite with a testset, it may include one or several testcases.
+        each suite should initialize a seperate Runner() with testset config.
+    """
+    suite = unittest.TestSuite()
+
+    test_runner = runner.Runner()
+    config_dict = testset.get("config", {})
+    test_runner.init_config(config_dict, level="testset")
+    testcases = testset.get("testcases", [])
+
+    for testcase in testcases:
+        if utils.PYTHON_VERSION == 3:
+            ApiTestCase.runTest.__doc__ = testcase['name']
+        else:
+            ApiTestCase.runTest.__func__.__doc__ = testcase['name']
+
+        test = ApiTestCase(test_runner, testcase)
+        suite.addTest(test)
+
+    return suite
+
+def create_task(testcase_path):
+    """ create test task suite with specified testcase path.
+        each task suite may include one or several test suite.
+    """
+    task_suite = unittest.TestSuite()
+    testsets = utils.load_testcases_by_path(testcase_path)
+
+    for testset in testsets:
+        suite = create_suite(testset)
+        task_suite.addTest(suite)
+
+    return task_suite