Expose connection state in the new asyncio implementation.

Author: aaugustin

docs/reference/new-asyncio/client.rst

@@ -43,6 +43,8 @@ Using a connection
 
     .. autoproperty:: remote_address
 
+    .. autoproperty:: state
+
     The following attributes are available after the opening handshake,
     once the WebSocket connection is open:
 

docs/reference/new-asyncio/common.rst

@@ -33,6 +33,8 @@ Both sides (new :mod:`asyncio`)
 
     .. autoproperty:: remote_address
 
+    .. autoproperty:: state
+
     The following attributes are available after the opening handshake,
     once the WebSocket connection is open:
 

docs/reference/new-asyncio/server.rst

@@ -62,6 +62,8 @@ Using a connection
 
     .. autoproperty:: remote_address
 
+    .. autoproperty:: state
+
     The following attributes are available after the opening handshake,
     once the WebSocket connection is open:
 

src/websockets/asyncio/connection.py

@@ -137,6 +137,18 @@ def remote_address(self) -> Any:
         """
         return self.transport.get_extra_info("peername")
 
+    @property
+    def state(self) -> State:
+        """
+        State of the WebSocket connection, defined in :rfc:`6455`.
+
+        This attribute is provided for completeness. Typical applications
+        shouldn't check its value. Instead, they should call :meth:`~recv` or
+        :meth:`send` and handle :exc:`~exceptions.ConnectionClosed` exceptions.
+
+        """
+        return self.protocol.state
+
     @property
     def subprotocol(self) -> Subprotocol | None:
         """

src/websockets/protocol.py

@@ -156,7 +156,7 @@ def __init__(
     @property
     def state(self) -> State:
         """
-        WebSocket connection state.
+        State of the WebSocket connection.
 
         Defined in 4.1, 4.2, 7.1.3, and 7.1.4 of :rfc:`6455`.
 

tests/asyncio/test_connection.py

@@ -11,7 +11,7 @@
 from websockets.asyncio.connection import *
 from websockets.exceptions import ConnectionClosedError, ConnectionClosedOK
 from websockets.frames import CloseCode, Frame, Opcode
-from websockets.protocol import CLIENT, SERVER, Protocol
+from websockets.protocol import CLIENT, SERVER, Protocol, State
 
 from ..protocol import RecordingProtocol
 from ..utils import MS
@@ -930,6 +930,10 @@ async def test_remote_address(self, get_extra_info):
         self.assertEqual(self.connection.remote_address, ("peer", 1234))
         get_extra_info.assert_called_with("peername")
 
+    async def test_state(self):
+        """Connection has a state attribute."""
+        self.assertEqual(self.connection.state, State.OPEN)
+
     async def test_request(self):
         """Connection has a request attribute."""
         self.assertIsNone(self.connection.request)