fix: formatting, static types and timedelta in test for mock elapsed

Author: BrunoMmS

httpx/_transports/mock.py

@@ -49,17 +49,11 @@ async def handle_async_request(
         self._apply_elapsed(response)
         return response
 
-    def _apply_elapsed(self, response):
+    def _apply_elapsed(self, response: Response) -> None:
         # If the handler already set `response._elapsed`, it is preserved.
         # If a delay was provided to MockTransport, `.elapsed` is set to that duration.
         # If no delay is provided, `.elapsed` is explicitly set to None.
         if hasattr(response, "_elapsed"):
             return
 
-        if self.delay is not None:
-            if isinstance(self.delay, timedelta):
-                response._elapsed = self.delay
-            else:
-                response._elapsed = timedelta(seconds=self.delay)
-        else:
-            response._elapsed = None
+        response._elapsed = self.delay

tests/client/test_client.py

@@ -468,7 +468,7 @@ def handler(request):
         r.elapsed = timedelta(seconds=1)
         return r
 
-    transport = httpx.MockTransport(handler, delay=0.5)
+    transport = httpx.MockTransport(handler, delay=timedelta(seconds=0.5))
     client = httpx.Client(transport=transport)
 
     response = client.get("https://example.com")
@@ -480,7 +480,7 @@ def test_mocktransport_sets_elapsed_to_delay():
     def handler(request):
         return httpx.Response(200)
 
-    transport = httpx.MockTransport(handler, delay=0.5)
+    transport = httpx.MockTransport(handler, delay=timedelta(seconds=0.5))
     client = httpx.Client(transport=transport)
 
     response = client.get("https://example.com")