locust 2.41.7.dev9__py3-none-any.whl → 2.41.7.dev12__py3-none-any.whl

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.dev12'
+__version_tuple__ = version_tuple = (2, 41, 7, 'dev12')
 
 __commit_id__ = commit_id = None

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.dist-info → locust-2.41.7.dev12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.41.7.dev9
+Version: 2.41.7.dev12
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust
@@ -44,6 +44,8 @@ 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.dist-info → locust-2.41.7.dev12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 locust/__init__.py,sha256=HadpgGidiyCDPSKwkxrk1Qw6eB7dTmftNJVftuJzAiw,1876
 locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-locust/_version.py,sha256=Yuozf9jPPJg_81Vo2Ajl_VdRzYaQQGCEGm5m6MKl5o0,719
+locust/_version.py,sha256=d_fUCMcfYMMze2J8xvAB-buiRdWco8KVqXBpGG-SjSE,721
 locust/argument_parser.py,sha256=E0HM9jOlKCzPM9S_E_UKSwgiF_CL0aIli5eCcOG8Xac,33547
 locust/clients.py,sha256=OfMTOT3LwijczdL3u6bpDwDNM2TDcPCIwXBelizPOyw,20776
 locust/debug.py,sha256=7CCm8bIg44uGH2wqBlo1rXBzV2VzwPicLxLewz8r5CQ,5099
@@ -18,6 +18,7 @@ locust/shape.py,sha256=t-lwBS8LOjWcKXNL7j2U3zroIXJ1b0fazUwpRYQOKXw,1973
 locust/stats.py,sha256=qyoSKT0i7RunLDj5pMGqizK1Sp8bcqUsXwh2m4_DpR8,47203
 locust/web.py,sha256=pLYuocmx9gifJ4vsVDgGDYIPkQhQxI7kKjxoXcUajqM,31865
 locust/contrib/__init__.py,sha256=LtZN7MczpIAbZkN7PT2h8W2wgb9nBl6cDXbFCVsV4fo,290
+locust/contrib/dns.py,sha256=dL36S4dwqQkrQOcy2_2OuGOPPzAljS9-nIswTZN7Adw,1665
 locust/contrib/fasthttp.py,sha256=vuHo6EFzSHJgr5SoYf_11Mu-ApnC0mz3hOU6TzQnioI,31145
 locust/contrib/milvus.py,sha256=ebOIVo6-rO3hnAMPKzMyof5MQd7kxOZFMkBNZYOluYk,13015
 locust/contrib/mongodb.py,sha256=1seUYgJOaNKwybYOP9PUEVhgl8hGy-G33f8lFj3R8W8,1246
@@ -57,8 +58,8 @@ locust/webui/dist/assets/terminal.gif,sha256=iw80LO2u0dnf4wpGfFJZauBeKTcSpw9iUfI
 locust/webui/dist/assets/testruns-dark.png,sha256=G4p2VZSBuuqF4neqUaPSshIp5OKQJ_Bvb69Luj6XuVs,125231
 locust/webui/dist/assets/testruns-light.png,sha256=JinGDiiBPOkhpfF-XCbmQqhRInqItrjrBTLKt5MlqVI,130301
 pytest_locust/plugin.py,sha256=3jr_5H_1SHCJTdQdgnTDJ-09r5jXzNKxJPkocfE9HF4,1669
-locust-2.41.7.dev9.dist-info/METADATA,sha256=yPK1LLio_glXnXBdhbLMalFcqU_4Jwq_bnIC-yKxWiA,9775
-locust-2.41.7.dev9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-locust-2.41.7.dev9.dist-info/entry_points.txt,sha256=0uIHcQ71R1qaWhM_sd8uBUCCJgp7gJfGHUVMnJeZfcY,86
-locust-2.41.7.dev9.dist-info/licenses/LICENSE,sha256=5hnz-Vpj0Z3kSCQl0LzV2hT1TLc4LHcbpBp3Cy-EuyM,1110
-locust-2.41.7.dev9.dist-info/RECORD,,
+locust-2.41.7.dev12.dist-info/METADATA,sha256=ByL_5_uUxbzcDdtDQq81uz47BXaBf8qRHDTEArzZMhE,9844
+locust-2.41.7.dev12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+locust-2.41.7.dev12.dist-info/entry_points.txt,sha256=0uIHcQ71R1qaWhM_sd8uBUCCJgp7gJfGHUVMnJeZfcY,86
+locust-2.41.7.dev12.dist-info/licenses/LICENSE,sha256=5hnz-Vpj0Z3kSCQl0LzV2hT1TLc4LHcbpBp3Cy-EuyM,1110
+locust-2.41.7.dev12.dist-info/RECORD,,