From 61b2a193e79d74783bb1ad4c3a8ccb6ead0f24c0 Mon Sep 17 00:00:00 2001 From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> Date: Wed, 1 Apr 2026 17:57:43 +0900 Subject: fix: lint cleanup after medium priority tasks --- shared/src/shared/broker.py | 4 +--- shared/tests/test_broker.py | 18 ++++-------------- 2 files changed, 5 insertions(+), 17 deletions(-) (limited to 'shared') 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 -- cgit v1.2.3