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

locust-plugins-4.3.3/.github/workflows/stale.yml

@@ -0,0 +1,26 @@
+name: Mark stale issues
+
+on:
+  schedule:
+    - cron: "30 1 * * *"
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  stale:
+    permissions:
+      issues: write  # for actions/stale to close stale issues
+      pull-requests: write  # for actions/stale to close stale PRs
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/stale@v3
+        with:
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+          stale-issue-message: "This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 10 days."
+          close-issue-message: "This issue was closed because it has been stalled for 10 days with no activity. This does not necessarily mean that the issue is bad, but it most likely means that nobody is willing to take the time to fix it. If you have found Locust useful, then consider contributing a fix yourself!"
+          stale-pr-message: "This PR is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 10 days."
+          close-pr-message: "This PR was closed because it has been stalled for 10 days with no activity."
+          days-before-close: 10

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust-plugins
-Version: 4.3.1
+Version: 4.3.3
 Summary: Useful plugins/extensions for Locust
 Home-page: https://github.com/SvenskaSpel/locust-plugins
 Author: Lars Holmberg
@@ -26,7 +26,7 @@ Requires-Dist: typing-extensions
 Provides-Extra: websocket
 Requires-Dist: websocket-client; extra == "websocket"
 Provides-Extra: playwright
-Requires-Dist: playwright>=1.34.0; extra == "playwright"
+Requires-Dist: playwright>=1.40.0; extra == "playwright"
 Provides-Extra: dashboards
 Requires-Dist: psycogreen; extra == "dashboards"
 Requires-Dist: psycopg2-binary; extra == "dashboards"
@@ -44,7 +44,7 @@ Provides-Extra: boto3
 Requires-Dist: boto3; extra == "boto3"
 Provides-Extra: all
 Requires-Dist: websocket-client; extra == "all"
-Requires-Dist: playwright>=1.34.0; extra == "all"
+Requires-Dist: playwright>=1.40.0; extra == "all"
 Requires-Dist: psycogreen; extra == "all"
 Requires-Dist: psycopg2-binary; extra == "all"
 Requires-Dist: confluent-kafka; extra == "all"

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/locust_plugins/__init__.py

@@ -1,4 +1,4 @@
-from .wait_time import constant_ips, constant_total_ips
+from .wait_time import constant_total_ips
 import locust
 from locust.user.task import DefaultTaskSet, TaskSet
 from locust import events
@@ -258,7 +258,6 @@ def do_checks(environment, **_kw):
 
 
 def missing_extra(package, extra):
-    traceback.print_exc()
     logging.error(
         f"'{package}' is not installed by default, you need to install it using 'pip install locust-plugins[{extra}]'"
     )

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

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

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

@@ -2,13 +2,14 @@ from typing import Dict, Iterator, Optional
 import logging
 from gevent.event import AsyncResult
 import greenlet
+import gevent
 from locust.env import Environment
 from locust.runners import WorkerRunner
 
 _results: Dict[int, AsyncResult] = {}
 
 
-class Distributor:
+class Distributor(Iterator):
     def __init__(self, environment: Environment, iterator: Optional[Iterator], name="distributor"):
         """Register distributor method handlers and tie them to use the iterator that you pass.
 
@@ -23,12 +24,8 @@ class Distributor:
         if self.runner:
             # received on master
             def _distributor_request(environment: Environment, msg, **kwargs):
-                item = next(self.iterator)
-                self.runner.send_message(
-                    f"_{name}_response",
-                    {"item": item, "gid": msg.data["gid"]},
-                    client_id=msg.data["client_id"],
-                )
+                # do this in the background to avoid blocking locust's client_listener loop
+                gevent.spawn(self._master_next_and_send, msg.data["gid"], msg.data["client_id"])
 
             # received on worker
             def _distributor_response(environment: Environment, msg, **kwargs):
@@ -37,6 +34,14 @@ class Distributor:
             self.runner.register_message(f"_{name}_request", _distributor_request)
             self.runner.register_message(f"_{name}_response", _distributor_response)
 
+    def _master_next_and_send(self, gid, client_id):
+        item = next(self.iterator)
+        self.runner.send_message(
+            f"_{self.name}_response",
+            {"item": item, "gid": gid},
+            client_id=client_id,
+        )
+
     def __next__(self):
         """Get the next data dict from iterator
 

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/locust_plugins/listeners/timescale.py

@@ -1,3 +1,4 @@
+# See timescale_listener_ex.py for documentation
 from contextlib import contextmanager
 from locust.exception import CatchResponseError  # need to do this first to make sure monkey patching is done
 import json

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

@@ -34,9 +34,12 @@ class MongoLRUReader(Iterator[Dict]):
 
         """
         self.timestamp_field = timestamp_field
-        self.coll: Collection = (
-            coll or MongoClient(env["LOCUST_MONGO"])[env["LOCUST_MONGO_DATABASE"]][env["LOCUST_MONGO_COLLECTION"]]
-        )
+        if not coll:
+            self.coll: Collection = MongoClient(env["LOCUST_MONGO"])[env["LOCUST_MONGO_DATABASE"]][
+                env["LOCUST_MONGO_COLLECTION"]
+            ]
+        else:
+            self.coll = coll
         self.cursor: Cursor = self.coll.find(filter, sort=[(self.timestamp_field, 1)])
         records_in_buffer = self.cursor._refresh()  # trigger fetch immediately instead of waiting for the first next()
         if not records_in_buffer:
@@ -46,7 +49,7 @@ class MongoLRUReader(Iterator[Dict]):
         try:
             with dblock:
                 doc: dict = next(self.cursor)
-            self.coll.find_one_and_update(
+            self.coll.update_one(
                 {"_id": doc["_id"]},
                 {"$set": {self.timestamp_field: datetime.now(tz=timezone.utc)}},
             )

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/locust_plugins/users/webdriver.py

@@ -27,7 +27,8 @@ class WebdriverClient(webdriver.Remote):
     def __init__(self, user: User):
         self.user = user
         options = Options()
-        options.headless = self.user.headless
+        if self.user.headless:
+            options.add_argument("--headless")
         for arg in [
             "--disable-translate",
             "--disable-extensions",

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/locust_plugins/wait_time.py

@@ -1,4 +1,3 @@
-import sys
 import logging
 import time
 from collections import deque, namedtuple
@@ -57,11 +56,5 @@ def constant_total_ips(ips: float):
     return func
 
 
-def constant_ips(_ips):
-    """this function has been removed, now that it is available in locust.wait_time.constant_throughput"""
-    logging.error("constant_ips has been removed, now that it is available in locust.wait_time.constant_throughput")
-    sys.exit(1)
-
-
 def constant_total_pacing(seconds: float):
     return constant_total_ips(1.0 / seconds)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust-plugins
-Version: 4.3.1
+Version: 4.3.3
 Summary: Useful plugins/extensions for Locust
 Home-page: https://github.com/SvenskaSpel/locust-plugins
 Author: Lars Holmberg
@@ -26,7 +26,7 @@ Requires-Dist: typing-extensions
 Provides-Extra: websocket
 Requires-Dist: websocket-client; extra == "websocket"
 Provides-Extra: playwright
-Requires-Dist: playwright>=1.34.0; extra == "playwright"
+Requires-Dist: playwright>=1.40.0; extra == "playwright"
 Provides-Extra: dashboards
 Requires-Dist: psycogreen; extra == "dashboards"
 Requires-Dist: psycopg2-binary; extra == "dashboards"
@@ -44,7 +44,7 @@ Provides-Extra: boto3
 Requires-Dist: boto3; extra == "boto3"
 Provides-Extra: all
 Requires-Dist: websocket-client; extra == "all"
-Requires-Dist: playwright>=1.34.0; extra == "all"
+Requires-Dist: playwright>=1.40.0; extra == "all"
 Requires-Dist: psycogreen; extra == "all"
 Requires-Dist: psycopg2-binary; extra == "all"
 Requires-Dist: confluent-kafka; extra == "all"

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/locust_plugins.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ pyproject.toml
 setup.py
 tox.ini
 tusk.yaml
+.github/workflows/stale.yml
 .github/workflows/tests.yml
 .vscode/launch.json
 .vscode/settings.json

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

@@ -3,7 +3,7 @@ typing-extensions
 
 [all]
 websocket-client
-playwright>=1.34.0
+playwright>=1.40.0
 psycogreen
 psycopg2-binary
 confluent-kafka
@@ -33,7 +33,7 @@ pymongo
 paho-mqtt>=1.5.0
 
 [playwright]
-playwright>=1.34.0
+playwright>=1.40.0
 
 [resource]
 lxml

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

@@ -6,7 +6,7 @@ from setuptools import find_packages, setup
 
 extras = {
     "websocket": ["websocket-client"],
-    "playwright": ["playwright>=1.34.0"],
+    "playwright": ["playwright>=1.40.0"],
     "dashboards": ["psycogreen", "psycopg2-binary"],
     "kafka": ["confluent-kafka"],
     "mongo": ["pymongo"],

{locust-plugins-4.3.1 → locust-plugins-4.3.3}/test/test_cloudwatch_plugin.py

@@ -10,12 +10,12 @@ from unittest import TestCase
 
 setup_logging("INFO", None)
 
+
 class CloudwatchMock:
     def __init__(self):
         self.call_count = 0
         self.metrics_dict = {}
 
-
     def put_metric_data(self, Namespace, MetricData):
         self.call_count += 1
         self.metrics_dict[Namespace] = MetricData
@@ -23,6 +23,7 @@ class CloudwatchMock:
 
 cw = CloudwatchMock()
 
+
 def on_locust_init(environment, **_kwargs):
     CloudwatchAdapter(environment, "MyExampleService", "perf", cw)