diff options
Diffstat (limited to 'shared')
| -rw-r--r-- | shared/src/shared/broker.py | 4 | ||||
| -rw-r--r-- | shared/tests/test_broker.py | 18 |
2 files changed, 5 insertions, 17 deletions
diff --git a/shared/src/shared/broker.py b/shared/src/shared/broker.py index c060c24..fbe4576 100644 --- a/shared/src/shared/broker.py +++ b/shared/src/shared/broker.py @@ -63,9 +63,7 @@ class RedisBroker: count: int = 10, ) -> list[tuple[str, dict[str, Any]]]: """Read pending (unacknowledged) messages for this consumer.""" - results = await self._redis.xreadgroup( - group, consumer, {stream: "0"}, count=count - ) + results = await self._redis.xreadgroup(group, consumer, {stream: "0"}, count=count) messages = [] if results: for _stream, entries in results: diff --git a/shared/tests/test_broker.py b/shared/tests/test_broker.py index c33f6ec..9be84b0 100644 --- a/shared/tests/test_broker.py +++ b/shared/tests/test_broker.py @@ -109,9 +109,7 @@ async def test_broker_read_group(): mock_redis = AsyncMock() mock_redis.xreadgroup = AsyncMock( - return_value=[ - (b"stream", [(b"1-0", {b"payload": b'{"type": "test"}'})]) - ] + return_value=[(b"stream", [(b"1-0", {b"payload": b'{"type": "test"}'})])] ) broker = RedisBroker.__new__(RedisBroker) broker._redis = mock_redis @@ -142,9 +140,7 @@ async def test_broker_read_pending(): mock_redis = AsyncMock() mock_redis.xreadgroup = AsyncMock( - return_value=[ - (b"stream", [(b"1-0", {b"payload": b'{"type": "pending"}'})]) - ] + return_value=[(b"stream", [(b"1-0", {b"payload": b'{"type": "pending"}'})])] ) broker = RedisBroker.__new__(RedisBroker) broker._redis = mock_redis @@ -154,9 +150,7 @@ async def test_broker_read_pending(): assert messages[0][0] == "1-0" assert messages[0][1] == {"type": "pending"} # Verify it uses "0" (not ">") to read pending - mock_redis.xreadgroup.assert_called_once_with( - "group", "consumer", {"stream": "0"}, count=10 - ) + mock_redis.xreadgroup.assert_called_once_with("group", "consumer", {"stream": "0"}, count=10) @pytest.mark.asyncio @@ -165,11 +159,7 @@ async def test_broker_read_pending_skips_empty_fields(): from shared.broker import RedisBroker mock_redis = AsyncMock() - mock_redis.xreadgroup = AsyncMock( - return_value=[ - (b"stream", [(b"1-0", {})]) - ] - ) + mock_redis.xreadgroup = AsyncMock(return_value=[(b"stream", [(b"1-0", {})])]) broker = RedisBroker.__new__(RedisBroker) broker._redis = mock_redis |