[PR #9558/0a706625 backport][3.11] Fix refactoring error from moving WebSocket mask (#9563)

Co-authored-by: J. Nick Koston <[email protected]>

Author: patchback[bot]

CHANGES/9558.feature.rst

@@ -0,0 +1 @@
+9554.feature.rst

aiohttp/_websocket/helpers.py

@@ -56,7 +56,7 @@ def _websocket_mask_python(mask: bytes, data: bytearray) -> None:
     websocket_mask = _websocket_mask_python
 else:
     try:
-        from ._websocket import _websocket_mask_cython  # type: ignore[import-not-found]
+        from .mask import _websocket_mask_cython  # type: ignore[import-not-found]
 
         websocket_mask = _websocket_mask_cython
     except ImportError:  # pragma: no cover

tests/test_websocket_parser.py

@@ -438,6 +438,9 @@ def test_websocket_mask_cython() -> None:
     message = bytearray(websocket_mask_data)
     _websocket_helpers._websocket_mask_cython(websocket_mask_mask, message)  # type: ignore[attr-defined]
     assert message == websocket_mask_masked
+    assert (
+        _websocket_helpers.websocket_mask is _websocket_helpers._websocket_mask_cython  # type: ignore[attr-defined]
+    )
 
 
 def test_websocket_mask_python_empty() -> None: