Add None-value tests for broadcast and anycast channels

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

tests/test_anycast.py

@@ -3,6 +3,8 @@
 
 """Tests for the Channel implementation."""
 
+from __future__ import annotations
+
 import asyncio
 
 import pytest
@@ -147,6 +149,23 @@ async def test_anycast_full() -> None:
         assert False
 
 
+async def test_anycast_none_values() -> None:
+    """Ensure None values can be sent and received."""
+    acast: Anycast[int | None] = Anycast()
+
+    sender = acast.new_sender()
+    receiver = acast.new_receiver()
+
+    await sender.send(5)
+    assert await receiver.receive() == 5
+
+    await sender.send(None)
+    assert await receiver.receive() is None
+
+    await sender.send(10)
+    assert await receiver.receive() == 10
+
+
 async def test_anycast_async_iterator() -> None:
     """Check that the anycast receiver works as an async iterator."""
     acast: Anycast[str] = Anycast()

tests/test_broadcast.py

@@ -3,6 +3,8 @@
 
 """Tests for the Broadcast implementation."""
 
+from __future__ import annotations
+
 import asyncio
 from typing import Tuple
 
@@ -69,6 +71,23 @@ async def update_tracker_on_receive(receiver_id: int, recv: Receiver[int]) -> No
     assert actual_sum == expected_sum
 
 
+async def test_broadcast_none_values() -> None:
+    """Ensure None values can be sent and received."""
+    bcast: Broadcast[int | None] = Broadcast("any_channel")
+
+    sender = bcast.new_sender()
+    receiver = bcast.new_receiver()
+
+    await sender.send(5)
+    assert await receiver.receive() == 5
+
+    await sender.send(None)
+    assert await receiver.receive() is None
+
+    await sender.send(10)
+    assert await receiver.receive() == 10
+
+
 async def test_broadcast_after_close() -> None:
     """Ensure closed channels can't get new messages."""
     bcast: Broadcast[int] = Broadcast("meter_5")