locust 2.46.8.dev47__tar.gz → 2.46.8.dev52__tar.gz

{locust-2.46.8.dev47 → locust-2.46.8.dev52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: locust
-Version: 2.46.8.dev47
+Version: 2.46.8.dev52
 Summary: Developer-friendly load testing framework
 Project-URL: Homepage, https://locust.cloud/
 Project-URL: Repository, https://github.com/locustio/locust

{locust-2.46.8.dev47 → locust-2.46.8.dev52}/locust/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.46.8.dev47'
-__version_tuple__ = version_tuple = (2, 46, 8, 'dev47')
+__version__ = version = '2.46.8.dev52'
+__version_tuple__ = version_tuple = (2, 46, 8, 'dev52')
 
 __commit_id__ = commit_id = None

{locust-2.46.8.dev47 → locust-2.46.8.dev52}/locust/contrib/mongodb.py

@@ -9,13 +9,13 @@ 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]
+        self.db = self[db_name]
 
     def execute_query(self, collection_name, query):
         start_time = time.time()
         try:
             collection = self.db[collection_name]
-            collection.find(query)
+            list(collection.find(query))  # find() is lazy, so fetch the documents to actually run the query
 
             response_time = int((time.time() - start_time) * 1000)
             events.request.fire(request_type="MONGODB", name="QUERY", response_time=response_time, response_length=0)

{locust-2.46.8.dev47 → locust-2.46.8.dev52}/locust/user/users.py

@@ -167,6 +167,8 @@ class User(metaclass=UserMeta):
             except Exception as e:
                 # unhandled exceptions inside tasks are logged in TaskSet.run, but since we're not yet there...
                 logger.error("%s\n%s", e, traceback.format_exc())
+                if not isinstance(e, (StopUser, StopTest)):
+                    self.environment.events.user_error.fire(user_instance=self, exception=e, tb=e.__traceback__)
                 raise
 
             self._taskset_instance.run()