add HttpRunner class as interface

httprunner/__init__.py

@@ -1 +1,3 @@
-__version__ = '0.9.3c'
+__version__ = '0.9.4'
+
+from httprunner.task import HttpRunner

httprunner/cli.py

@@ -7,35 +7,12 @@ from collections import OrderedDict
 
 from httprunner import __version__ as hrun_version
 from httprunner import logger
-from httprunner.exception import TestcaseNotFound
-from httprunner.task import Result, TaskSuite
+from httprunner.task import HttpRunner
 from httprunner.utils import create_scaffold, print_output, string_type
 from pyunitreport import __version__ as pyu_version
 from pyunitreport import HTMLTestRunner
 
 
-def run_suite_path(path, mapping=None, runner=None):
-    """ run suite with YAML/JSON file path
-    @params:
-        - path: testset path
-        - mapping: passed in variables mapping, it will override variables in config block
-        - runner: HTMLTestRunner() or TextTestRunner()
-    """
-    try:
-        mapping = mapping or {}
-        task_suite = TaskSuite(path, mapping)
-    except TestcaseNotFound:
-        sys.exit(1)
-
-    test_runner = runner or unittest.TextTestRunner()
-    result = test_runner.run(task_suite)
-
-    output = {}
-    for task in task_suite.tasks:
-        output.update(task.output)
-
-    return Result(result, output)
-
 def main_hrun():
     """ API test: parse command line options and run commands.
     """
@@ -76,7 +53,7 @@ def main_hrun():
         "failfast": args.failfast
     }
     test_runner = HTMLTestRunner(**kwargs)
-    result = run_suite_path(args.testset_paths, {}, test_runner)
+    result = HttpRunner(args.testset_paths, test_runner).run()
     print_output(result.output)
 
     return 0 if result.success else 1

httprunner/task.py

@@ -1,3 +1,4 @@
+import sys
 import unittest
 
 from httprunner import exception, logger, runner, testcase, utils
@@ -153,6 +154,35 @@ class Result(object):
         return self.Stat(**stat)
 
 
+class HttpRunner(object):
+
+    def __init__(self, path, runner=None):
+        """ initialize HttpRunner with specified testset file path and test runner
+        @params:
+            - path: YAML/JSON testset file path
+            - runner: HTMLTestRunner() or TextTestRunner()
+        """
+        self.path = path
+        self.runner = runner or unittest.TextTestRunner()
+
+    def run(self, mapping=None):
+        """ start to run suite
+            if mapping specified, it will override variables in config block
+        """
+        try:
+            mapping = mapping or {}
+            task_suite = TaskSuite(self.path, mapping)
+        except exception.TestcaseNotFound:
+            sys.exit(1)
+
+        result = self.runner.run(task_suite)
+        output = {}
+        for task in task_suite.tasks:
+            output.update(task.output)
+
+        return Result(result, output)
+
+
 class LocustTask(object):
 
     def __init__(self, path, locust_client, mapping=None):