change: remove locust for python httprunner

httprunner/ext/locust/__init__.py

@@ -1,113 +0,0 @@
-import sys
-
-if "locust" in sys.argv[0]:
-    try:
-        # monkey patch all at beginning to avoid RecursionError when running locust.
-        # `from gevent import monkey; monkey.patch_all()` will be triggered when importing locust
-        from locust import main as locust_main
-
-        print("NOTICE: gevent monkey patches have been applied !!!")
-    except ImportError:
-        msg = """
-Locust is not installed, install first and try again.
-install with pip:
-$ pip install locust
-"""
-        print(msg)
-        sys.exit(1)
-
-import importlib.util
-import inspect
-import os
-from typing import List
-
-from loguru import logger
-
-
-""" converted pytest files from YAML/JSON testcases
-"""
-pytest_files: List = []
-
-
-def is_httprunner_testcase(item):
-    """ check if a variable is a HttpRunner testcase class
-    """
-    from httprunner import HttpRunner
-
-    # TODO: skip referenced testcase
-    return bool(
-        inspect.isclass(item)
-        and issubclass(item, HttpRunner)
-        and item.__name__ != "HttpRunner"
-    )
-
-
-def prepare_locust_tests() -> List:
-    """ prepare locust testcases
-
-    Returns:
-        list: testcase class list
-    """
-
-    locust_tests = []
-
-    for pytest_file in pytest_files:
-        spec = importlib.util.spec_from_file_location("module.name", pytest_file)
-        module = importlib.util.module_from_spec(spec)
-        spec.loader.exec_module(module)
-
-        for name, item in vars(module).items():
-
-            if not is_httprunner_testcase(item):
-                continue
-
-            for _ in range(item.config.weight):
-                locust_tests.append(item)
-
-    return locust_tests
-
-
-def main_locusts():
-    """ locusts entrance
-    """
-    from httprunner.utils import ga_client
-    ga_client.track_event("RunLoadTests", "locust")
-
-    # avoid print too much log details in console
-    logger.remove()
-    logger.add(sys.stderr, level="WARNING")
-
-    sys.argv[0] = "locust"
-    if len(sys.argv) == 1:
-        sys.argv.extend(["-h"])
-
-    if sys.argv[1] in ["-h", "--help", "-V", "--version"]:
-        locust_main.main()
-
-    def get_arg_index(*target_args):
-        for arg in target_args:
-            if arg not in sys.argv:
-                continue
-
-            return sys.argv.index(arg) + 1
-
-        return None
-
-    # get testcase file path
-    testcase_index = get_arg_index("-f", "--locustfile")
-    if not testcase_index:
-        print("Testcase file is not specified, exit 1.")
-        sys.exit(1)
-
-    from httprunner.make import main_make
-
-    global pytest_files
-    testcase_file_path = sys.argv[testcase_index]
-    pytest_files = main_make([testcase_file_path])
-    if not pytest_files:
-        print("No valid testcases found, exit 1.")
-        sys.exit(1)
-
-    sys.argv[testcase_index] = os.path.join(os.path.dirname(__file__), "locustfile.py")
-
-    locust_main.main()

httprunner/ext/locust/locustfile.py

@@ -1,30 +0,0 @@
-import random
-
-from locust import task, HttpUser, between
-
-from httprunner.ext.locust import prepare_locust_tests
-
-
-class HttpRunnerUser(HttpUser):
-    host = ""
-    wait_time = between(5, 15)
-
-    def on_start(self):
-        locust_tests = prepare_locust_tests()
-        self.testcase_runners = [
-            testcase().with_session(self.client) for testcase in locust_tests
-        ]
-
-    @task
-    def test_any(self):
-        test_runner = random.choice(self.testcase_runners)
-        try:
-            test_runner.run()
-        except Exception as ex:
-            self.environment.events.request_failure.fire(
-                request_type="Failed",
-                name=test_runner.config.name,
-                response_time=0,
-                response_length=0,
-                exception=ex,
-            )