chanx 2.2.0__tar.gz → 2.2.2__tar.gz

{chanx-2.2.0 → chanx-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chanx
-Version: 2.2.0
+Version: 2.2.2
 Summary: The missing toolkit for Channels — auth, logging, consumers, and more.
 Project-URL: Documentation, https://chanx.readthedocs.io/
 Project-URL: Homepage, https://github.com/huynguyengl99/chanx

{chanx-2.2.0 → chanx-2.2.2}/chanx/core/testing.py

@@ -13,8 +13,9 @@ fast-channels, etc.) to create concrete testing utilities.
 import asyncio
 from collections.abc import Collection, Generator
 from contextlib import contextmanager
+from dataclasses import dataclass
 from types import TracebackType
-from typing import Any, Self, TypedDict, cast
+from typing import Any, Self, cast
 
 import humps
 from asgiref.timeout import timeout as async_timeout
@@ -29,7 +30,8 @@ from chanx.core.websocket import ChanxWebsocketConsumerMixin
 from chanx.messages.base import BaseMessage
 
 
-class CapturedBroadcastEvent(TypedDict):
+@dataclass
+class CapturedBroadcastEvent:
     """Structure of a captured broadcast event."""
 
     event: BaseMessage
@@ -267,6 +269,7 @@ class WebsocketCommunicatorMixin:
 @contextmanager
 def capture_broadcast_events(
     consumer: type[ChanxWebsocketConsumerMixin],
+    suppress: bool = True,
 ) -> Generator[list[CapturedBroadcastEvent], None, None]:
     """
     Capture broadcast events sent via broadcast_event() for testing purposes.
@@ -274,6 +277,11 @@ def capture_broadcast_events(
     Similar to structlog's capture_logs(), this context manager captures calls to
     broadcast_event() by monkey-patching the broadcast_event method to spy on events.
 
+    Args:
+        consumer: The consumer class to capture broadcast events from.
+        suppress: If True (default), suppress actual broadcast event calls.
+                  If False, capture events and still call the original broadcast_event.
+
     Returns:
         A list that will be populated with captured broadcast events.
     """
@@ -298,7 +306,9 @@ def capture_broadcast_events(
             )
         )
 
-        await original_broadcast_event(event, groups)
+        # Only call original if not suppressing
+        if not suppress:
+            await original_broadcast_event(event, groups)
 
     # Monkey-patch the method
     consumer.broadcast_event = classmethod(capture_wrapper)  # type: ignore[method-assign, assignment]

{chanx-2.2.0 → chanx-2.2.2}/pyproject.toml

@@ -106,7 +106,7 @@ license = {file = "LICENSE"}
 name = "chanx"
 readme = "README.rst"
 requires-python = ">=3.11,<4.0"
-version = "2.2.0"
+version = "2.2.2"
 
 [project.optional-dependencies]
 channels = [