locust 2.31.7.dev2__tar.gz → 2.31.7.dev8__tar.gz

{locust-2.31.7.dev2 → locust-2.31.7.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.31.7.dev2
+Version: 2.31.7.dev8
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT

{locust-2.31.7.dev2 → locust-2.31.7.dev8}/locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.7.dev2"
+__version__ = "2.31.7.dev8"
 version = __version__
-__version_tuple__ = (2, 31, 7, "dev2")
+__version_tuple__ = (2, 31, 7, "dev8")
 version_tuple = __version_tuple__

locust-2.31.7.dev8/locust/contrib/mongodb.py

@@ -0,0 +1,41 @@
+from locust import User, events
+
+import time
+
+from pymongo import MongoClient
+from pymongo.errors import PyMongoError
+
+
+class MongoDBClient(MongoClient):
+    def __init__(self, conn_string, db_name):
+        super().__init__(conn_string)
+        self.db = self.client[db_name]
+
+    def execute_query(self, collection_name, query):
+        start_time = time.time()
+        try:
+            collection = self.db[collection_name]
+            collection.find(query)
+
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(request_type="MONGODB", name="QUERY", response_time=response_time, response_length=0)
+        except PyMongoError as e:
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="MONGODB",
+                name="QUERY",
+                response_time=response_time,
+                response_length=0,
+                exception=e,
+            )
+
+
+class MongoDBUser(User):
+    abstract = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.client = MongoDBClient(conn_string=self.conn_string, db_name=self.db_name)
+
+    def on_stop(self):
+        self.client.close()

{locust-2.31.7.dev2 → locust-2.31.7.dev8}/locust/main.py

@@ -199,6 +199,8 @@ def main():
             sys.exit(1)
 
     children = []
+    logger = logging.getLogger(__name__)
+    logger.info(f"Starting Locust {version}")
 
     if options.processes:
         if os.name == "nt":
@@ -298,7 +300,6 @@ def main():
 
                 atexit.register(kill_workers, children)
 
-    logger = logging.getLogger(__name__)
     greenlet_exception_handler = greenlet_exception_logger(logger)
 
     if options.stop_timeout:
@@ -686,7 +687,6 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
     gevent.signal_handler(signal.SIGTERM, sig_term_handler)
 
     try:
-        logger.info(f"Starting Locust {version}")
         if options.class_picker:
             logger.info("Locust is running with the UserClass Picker Enabled")
         if options.autostart and not options.headless:

{locust-2.31.7.dev2 → locust-2.31.7.dev8}/locust/web.py

@@ -135,6 +135,11 @@ class WebUI:
         # ensures static js files work on Windows
         mimetypes.add_type("application/javascript", ".js")
 
+        if self.web_login:
+            self._login_manager = LoginManager()
+            self._login_manager.init_app(self.app)
+            self._login_manager.login_view = "login"
+
         if environment.runner:
             self.update_template_args()
         if not delayed_start:
@@ -516,13 +521,14 @@ class WebUI:
     @property
     def login_manager(self):
         if self.web_login:
-            login_manager = LoginManager()
-            login_manager.init_app(self.app)
-            login_manager.login_view = "login"
-            return login_manager
+            return self._login_manager
 
         raise AttributeError("The login_manager is only available with --web-login.\n")
 
+    @login_manager.setter
+    def login_manager(self, value):
+        self._login_manager = value
+
     def start(self):
         self.greenlet = gevent.spawn(self.start_server)
         self.greenlet.link_exception(greenlet_exception_handler)

{locust-2.31.7.dev2 → locust-2.31.7.dev8}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry_dynamic_versioning.backend"
 [tool.poetry]
 name = "locust"
 description = "Developer-friendly load testing framework"
-version = "2.31.7.dev2"
+version = "2.31.7.dev8"
 license = "MIT"
 readme = "README.md"
 authors = ["Jonatan Heyman", "Lars Holmberg"]