locust-plugins 4.3.0__tar.gz → 4.3.1__tar.gz

{locust-plugins-4.3.0/locust_plugins.egg-info → locust-plugins-4.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust-plugins
-Version: 4.3.0
+Version: 4.3.1
 Summary: Useful plugins/extensions for Locust
 Home-page: https://github.com/SvenskaSpel/locust-plugins
 Author: Lars Holmberg
@@ -21,7 +21,7 @@ Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: System Administrators
 Requires-Python: >=3.8, <4
 License-File: LICENSE
-Requires-Dist: locust>=2.16.1
+Requires-Dist: locust>=2.20.0
 Requires-Dist: typing-extensions
 Provides-Extra: websocket
 Requires-Dist: websocket-client; extra == "websocket"

{locust-plugins-4.3.0 → locust-plugins-4.3.1}/locust_plugins/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '4.3.0'
-__version_tuple__ = version_tuple = (4, 3, 0)
+__version__ = version = '4.3.1'
+__version_tuple__ = version_tuple = (4, 3, 1)

{locust-plugins-4.3.0 → locust-plugins-4.3.1}/locust_plugins/distributor.py

@@ -23,16 +23,16 @@ class Distributor:
         if self.runner:
             # received on master
             def _distributor_request(environment: Environment, msg, **kwargs):
-                data = next(self.iterator)
+                item = next(self.iterator)
                 self.runner.send_message(
                     f"_{name}_response",
-                    {"payload": data, "user_id": msg.data["user_id"]},
+                    {"item": item, "gid": msg.data["gid"]},
                     client_id=msg.data["client_id"],
                 )
 
             # received on worker
             def _distributor_response(environment: Environment, msg, **kwargs):
-                _results[msg.data["user_id"]].set(msg.data)
+                _results[msg.data["gid"]].set(msg.data)
 
             self.runner.register_message(f"_{name}_request", _distributor_request)
             self.runner.register_message(f"_{name}_response", _distributor_response)
@@ -53,7 +53,7 @@ class Distributor:
             logging.warning("This user was already waiting for data. Strange.")
 
         _results[gid] = AsyncResult()
-        self.runner.send_message(f"_{self.name}_request", {"user_id": gid, "client_id": self.runner.client_id})
-        data = _results[gid].get()["payload"]  # this waits for the reply
+        self.runner.send_message(f"_{self.name}_request", {"gid": gid, "client_id": self.runner.client_id})
+        item = _results[gid].get()["item"]  # this waits for the reply
         del _results[gid]
-        return data
+        return item

{locust-plugins-4.3.0 → locust-plugins-4.3.1}/locust_plugins/mongoreader.py

@@ -38,7 +38,9 @@ class MongoLRUReader(Iterator[Dict]):
             coll or MongoClient(env["LOCUST_MONGO"])[env["LOCUST_MONGO_DATABASE"]][env["LOCUST_MONGO_COLLECTION"]]
         )
         self.cursor: Cursor = self.coll.find(filter, sort=[(self.timestamp_field, 1)])
-        self.cursor._refresh()  # trigger fetch immediately instead of waiting for the first next()
+        records_in_buffer = self.cursor._refresh()  # trigger fetch immediately instead of waiting for the first next()
+        if not records_in_buffer:
+            logging.warning(f"No records returned from query {filter}")
 
     def __next__(self) -> dict:
         try:

{locust-plugins-4.3.0 → locust-plugins-4.3.1/locust_plugins.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust-plugins
-Version: 4.3.0
+Version: 4.3.1
 Summary: Useful plugins/extensions for Locust
 Home-page: https://github.com/SvenskaSpel/locust-plugins
 Author: Lars Holmberg
@@ -21,7 +21,7 @@ Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: System Administrators
 Requires-Python: >=3.8, <4
 License-File: LICENSE
-Requires-Dist: locust>=2.16.1
+Requires-Dist: locust>=2.20.0
 Requires-Dist: typing-extensions
 Provides-Extra: websocket
 Requires-Dist: websocket-client; extra == "websocket"

{locust-plugins-4.3.0 → locust-plugins-4.3.1}/locust_plugins.egg-info/requires.txt

@@ -1,4 +1,4 @@
-locust>=2.16.1
+locust>=2.20.0
 typing-extensions
 
 [all]

{locust-plugins-4.3.0 → locust-plugins-4.3.1}/setup.py

@@ -46,7 +46,7 @@ setup(
     package_data={"locust_plugins": ["py.typed"]},
     zip_safe=False,
     install_requires=[
-        "locust>=2.16.1",
+        "locust>=2.20.0",
         "typing-extensions",
     ],
     extras_require={