Skip to content

Commit 4a57e0c

Browse files
authored
Remove stompman.Client(on_unhandled_message_frame=) (#49)
1 parent 4418211 commit 4a57e0c

File tree

3 files changed

+1
-7
lines changed

3 files changed

+1
-7
lines changed

README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ async with stompman.Client(
2727

2828
# Handlers:
2929
on_error_frame=lambda error_frame: print(error_frame.body),
30-
on_unhandled_message_frame=lambda message_frame: print(message_frame.body),
3130
on_heartbeat=lambda: print("Server sent a heartbeat"),
3231

3332
# Optional parameters with sensible defaults:

stompman/client.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,6 @@ class Client:
212212

213213
servers: list[ConnectionParameters] = field(kw_only=False)
214214
on_error_frame: Callable[[ErrorFrame], None] | None = None
215-
on_unhandled_message_frame: Callable[[MessageFrame], None] | None = None
216215
on_heartbeat: Callable[[], None] | None = None
217216

218217
heartbeat: Heartbeat = field(default=Heartbeat(1000, 1000))
@@ -296,8 +295,6 @@ async def _listen_to_frames(self) -> None:
296295
case MessageFrame():
297296
if subscription := self._active_subscriptions.get(frame.headers["subscription"]):
298297
task_group.create_task(subscription._run_handler(frame=frame)) # noqa: SLF001
299-
elif self.on_unhandled_message_frame:
300-
self.on_unhandled_message_frame(frame)
301298
case ErrorFrame():
302299
if self.on_error_frame:
303300
self.on_error_frame(frame)

tests/test_client.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ async def test_client_listen_routing_ok(monkeypatch: pytest.MonkeyPatch) -> None
322322
build_dataclass(ReceiptFrame),
323323
(first_message_frame := build_dataclass(MessageFrame, headers={"subscription": first_sub_id})),
324324
(error_frame := build_dataclass(ErrorFrame)),
325-
(second_message_frame := build_dataclass(MessageFrame)),
325+
(_second_message_frame := build_dataclass(MessageFrame)),
326326
(third_message_frame := build_dataclass(MessageFrame, headers={"subscription": second_sub_id})),
327327
HeartbeatFrame(),
328328
]
@@ -335,7 +335,6 @@ async def test_client_listen_routing_ok(monkeypatch: pytest.MonkeyPatch) -> None
335335
connection_class=connection_class,
336336
on_error_frame=(on_error_frame := mock.Mock()),
337337
on_heartbeat=(on_heartbeat := mock.Mock()),
338-
on_unhandled_message_frame=(on_unhandled_message_frame := mock.Mock()),
339338
) as client:
340339
first_subscription = await client.subscribe(
341340
FAKER.pystr(), handler=first_message_handler, on_suppressed_exception=first_error_handler
@@ -356,7 +355,6 @@ async def test_client_listen_routing_ok(monkeypatch: pytest.MonkeyPatch) -> None
356355

357356
on_error_frame.assert_called_once_with(error_frame)
358357
on_heartbeat.assert_called_once_with()
359-
on_unhandled_message_frame.assert_called_once_with(second_message_frame)
360358

361359

362360
@pytest.mark.parametrize("side_effect", [None, SomeError])

0 commit comments

Comments
 (0)