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

ate/cli.py

@@ -0,0 +1,37 @@
+import os
+import argparse
+import logging
+
+import HtmlTestRunner
+
+from ate.task import create_task
+
+def main():
+    """ parse command line options and run commands.
+    """
+    parser = argparse.ArgumentParser(
+        description='Api Test Engine.')
+    parser.add_argument(
+        '--testcase-path', default='testcases',
+        help="testcase file path")
+    parser.add_argument(
+        '--log-level', default='INFO',
+        help="Specify logging level, default is INFO.")
+    parser.add_argument(
+        '--report-name',
+        help="Specify report name, default is generated time.")
+
+    args = parser.parse_args()
+
+    log_level = getattr(logging, args.log_level.upper())
+    logging.basicConfig(level=log_level)
+
+    testcase_path = args.testcase_path.rstrip('/')
+    task_suite = create_task(testcase_path)
+
+    output_folder_name = os.path.basename(os.path.splitext(testcase_path)[0])
+    kwargs = {
+        "output": output_folder_name,
+        "report_name": args.report_name
+    }
+    HtmlTestRunner.HTMLTestRunner(**kwargs).run(task_suite)

ate/task.py

@@ -1,10 +1,5 @@
-import os
-import argparse
-import logging
 import unittest
 
-import HtmlTestRunner
-
 from ate import runner, utils
 
 
@@ -56,33 +51,3 @@ def create_task(testcase_path):
         task_suite.addTest(suite)
 
     return task_suite
-
-def main():
-    """ parse command line options and run commands.
-    """
-    parser = argparse.ArgumentParser(
-        description='Api Test Engine.')
-    parser.add_argument(
-        '--testcase-path', default='testcases',
-        help="testcase file path")
-    parser.add_argument(
-        '--log-level', default='INFO',
-        help="Specify logging level, default is INFO.")
-    parser.add_argument(
-        '--report-name',
-        help="Specify report name, default is generated time.")
-
-    args = parser.parse_args()
-
-    log_level = getattr(logging, args.log_level.upper())
-    logging.basicConfig(level=log_level)
-
-    testcase_path = args.testcase_path.rstrip('/')
-    task_suite = create_task(testcase_path)
-
-    output_folder_name = os.path.basename(os.path.splitext(testcase_path)[0])
-    kwargs = {
-        "output": output_folder_name,
-        "report_name": args.report_name
-    }
-    HtmlTestRunner.HTMLTestRunner(**kwargs).run(task_suite)