locust 2.41.7.dev9__tar.gz → 2.41.7.dev14__tar.gz

{locust-2.41.7.dev9 → locust-2.41.7.dev14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.41.7.dev9
+Version: 2.41.7.dev14
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust
@@ -39,11 +39,13 @@ Requires-Dist: python-engineio>=4.12.2
 Requires-Dist: python-socketio[client]>=5.13.0
 Requires-Dist: pywin32; sys_platform == 'win32'
 Requires-Dist: pyzmq>=25.0.0
-Requires-Dist: requests>=2.32.2
+Requires-Dist: requests<2.32.5,>=2.32.2
 Requires-Dist: setuptools>=70.0.0
 Requires-Dist: tomli>=1.1.0; python_version < '3.11'
 Requires-Dist: typing-extensions>=4.6.0; python_version < '3.12'
 Requires-Dist: werkzeug>=2.0.0
+Provides-Extra: dns
+Requires-Dist: dnspython>=2.8.0; extra == 'dns'
 Provides-Extra: milvus
 Requires-Dist: pymilvus>=2.5.0; extra == 'milvus'
 Provides-Extra: mqtt

{locust-2.41.7.dev9 → locust-2.41.7.dev14}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.41.7.dev9'
-__version_tuple__ = version_tuple = (2, 41, 7, 'dev9')
+__version__ = version = '2.41.7.dev14'
+__version_tuple__ = version_tuple = (2, 41, 7, 'dev14')
 
 __commit_id__ = commit_id = None

locust-2.41.7.dev14/locust/contrib/dns.py

@@ -0,0 +1,50 @@
+from locust import User
+from locust.exception import LocustError
+
+import time
+from collections.abc import Callable
+
+import dns.query
+from dns.exception import DNSException
+from dns.message import Message
+
+
+class DNSClient:
+    def __init__(self, request_event):
+        self.request_event = request_event
+
+    def __getattr__(self, function_name) -> Callable[..., Message]:
+        func = getattr(dns.query, function_name)
+
+        def wrapper(message: Message, *args, name=None, **kwargs) -> Message:
+            response = None
+            request_meta = {
+                "request_type": "DNS",
+                "name": name or function_name,
+                "start_time": time.time(),
+                "response_length": 0,
+                "context": {},
+                "exception": None,
+            }
+            start_perf_counter = time.perf_counter()
+            try:
+                response = func(message, *args, **kwargs)
+            except DNSException as e:
+                request_meta["exception"] = e
+            else:
+                if not response.answer:
+                    request_meta["exception"] = LocustError("No answer in DNS response")
+            request_meta["response_time"] = (time.perf_counter() - start_perf_counter) * 1000
+            request_meta["response"] = response
+            self.request_event.fire(**request_meta)
+            return response
+
+        return wrapper  # for some reason, pyright still wont infer the return type to be Message
+
+
+class DNSUser(User):
+    abstract = True
+
+    def __init__(self, environment):
+        super().__init__(environment)
+        self.client = DNSClient(environment.events.request)

{locust-2.41.7.dev9 → locust-2.41.7.dev14}/pyproject.toml

@@ -35,7 +35,7 @@ classifiers = [
 dependencies = [
     "flask>=2.0.0",
     "Werkzeug>=2.0.0",
-    "requests>=2.32.2",
+    "requests>=2.32.2,<2.32.5",
     "msgpack>=1.0.0",
     "pyzmq>=25.0.0",
     "geventhttpclient>=2.3.1",
@@ -57,6 +57,7 @@ dependencies = [
 [project.optional-dependencies]
 milvus = ["pymilvus>=2.5.0"]
 mqtt = ["paho-mqtt>=2.1.0"]
+dns = ["dnspython>=2.8.0"]
 
 [project.urls]
 homepage = "https://locust.io/"
@@ -107,6 +108,7 @@ docs = [
 ]
 milvus = ["pymilvus>=2.5.0"]
 mqtt = ["paho-mqtt>=2.1.0"]
+dns = ["dnspython>=2.8.0"]
 
 [project.scripts]
 locust = "locust.main:main"