Fix refactoring error from moving WebSocket mask (#9558)

aio-libs · Oct 28, 2024 · 0a70662 · 0a70662
1 parent 50656ca
commit 0a70662
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/CHANGES/9558.feature.rst b/CHANGES/9558.feature.rst
@@ -0,0 +1 @@
+9554.feature.rst
diff --git a/aiohttp/_websocket/helpers.py b/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

diff --git a/tests/test_websocket_parser.py b/tests/test_websocket_parser.py
@@ -494,6 +494,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: