fix types

tests/unit/vendor/httpx_ws/_api.py

@@ -118,7 +118,12 @@ def __enter__(self) -> "WebSocketSession":
 
         return self
 
-    def __exit__(self, exc_type, exc, tb):
+    def __exit__(
+        self,
+        exc_type: typing.Optional[type[BaseException]],
+        exc: typing.Optional[BaseException],
+        tb: typing.Optional[TracebackType],
+    ) -> None:
         self.close()
         self._background_receive_task.join()
         if self._background_keepalive_ping_task is not None:
@@ -416,7 +421,7 @@ def receive_json(
             data = self.receive_bytes(timeout)
         return json.loads(data)
 
-    def close(self, code: int = 1000, reason: typing.Optional[str] = None):
+    def close(self, code: int = 1000, reason: typing.Optional[str] = None) -> None:
         """
         Close the WebSocket session.
 
@@ -522,7 +527,7 @@ def _background_keepalive_ping(
             pass
 
     def _wait_until_closed(
-        self, callable: typing.Callable[..., TaskResult], *args, **kwargs
+        self, callable: typing.Callable[..., TaskResult], *args: typing.Any, **kwargs: typing.Any
     ) -> TaskResult:
         try:
             executor, should_close_task = self._get_executor_should_close_task()
@@ -921,7 +926,7 @@ async def receive_json(
             data = await self.receive_bytes(timeout)
         return json.loads(data)
 
-    async def close(self, code: int = 1000, reason: typing.Optional[str] = None):
+    async def close(self, code: int = 1000, reason: typing.Optional[str] = None) -> None:
         """
         Close the WebSocket session.
 

tests/unit/vendor/httpx_ws/_ping.py

@@ -20,7 +20,7 @@ def create(self, ping_id: typing.Optional[bytes] = None) -> tuple[bytes, threadi
         self._pings[ping_id] = event
         return ping_id, event
 
-    def ack(self, ping_id: typing.Union[bytes, bytearray]):
+    def ack(self, ping_id: typing.Union[bytes, bytearray]) -> None:
         event = self._pings.pop(bytes(ping_id))
         event.set()
 
@@ -35,6 +35,6 @@ def create(self, ping_id: typing.Optional[bytes] = None) -> tuple[bytes, anyio.E
         self._pings[ping_id] = event
         return ping_id, event
 
-    def ack(self, ping_id: typing.Union[bytes, bytearray]):
+    def ack(self, ping_id: typing.Union[bytes, bytearray]) -> None:
         event = self._pings.pop(bytes(ping_id))
         event.set()

tests/unit/vendor/httpx_ws/transport.py

@@ -38,7 +38,7 @@ def __init__(self, app: ASGIApp, scope: Scope) -> None:
         self._send_queue: asyncio.Queue[Message] = asyncio.Queue()
         self.connection = wsproto.WSConnection(wsproto.ConnectionType.SERVER)
         self.connection.initiate_upgrade_connection(scope["headers"], scope["path"])
-        self.tasks: list[asyncio.Task] = []
+        self.tasks: list[asyncio.Task[None]] = []
 
     async def __aenter__(
         self,
@@ -65,7 +65,7 @@ async def __aexit__(self, *args: typing.Any) -> None:
         await self.aclose()
         await self.exit_stack.aclose()
 
-    async def _cancel_tasks(self):
+    async def _cancel_tasks(self) -> None:
         # Cancel all running tasks
         for task in self.tasks:
             task.cancel()
@@ -166,7 +166,7 @@ def _build_accept_response(self, message: Message) -> bytes:
 
 
 class ASGIWebSocketTransport(ASGITransport):
-    def __init__(self, *args, **kwargs) -> None:
+    def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None:
         super().__init__(*args, **kwargs)
         self.exit_stack: typing.Optional[contextlib.AsyncExitStack] = None