Align test broker behavior with real one: dump (and validate by doing that) SendFrame on fake publish (#184)

Author: vrslev

packages/faststream-stomp/faststream_stomp/testing.py

@@ -10,6 +10,8 @@
 import stompman
 from faststream._internal.testing.broker import TestBroker, change_producer
 from faststream.message import encode_message
+from stompman.frames import SendFrame
+from stompman.serde import dump_frame
 
 from faststream_stomp.broker import StompBroker
 from faststream_stomp.models import StompPublishCommand
@@ -72,6 +74,16 @@ async def publish(self, cmd: StompPublishCommand) -> None:
             all_headers["correlation-id"] = cmd.correlation_id  # type: ignore[typeddict-unknown-key]
         if content_type:
             all_headers["content-type"] = content_type
+        dump_frame(
+            SendFrame.build(
+                body=body,
+                destination=cmd.destination,
+                transaction=None,
+                content_type=content_type,
+                add_content_length=False,
+                headers=cmd.headers,
+            )
+        )
         frame = FakeAckableMessageFrame(headers=all_headers, body=body, _subscription=mock.AsyncMock())
         for handler in self.broker.subscribers:
             if typing.cast("StompSubscriber", handler).config.full_destination == cmd.destination:

packages/faststream-stomp/test_faststream_stomp/test_main.py

@@ -1,3 +1,5 @@
+import typing
+
 import faker
 import faststream_stomp
 import pydantic
@@ -59,6 +61,19 @@ def second_handle(body: str) -> None:
             assert third_publisher.mock
             third_publisher.mock.assert_called_once_with(expected_body)
 
+    async def test_non_string_header_values_raise_type_error(
+        self, faker: faker.Faker, broker: faststream_stomp.StompBroker
+    ) -> None:
+        # gotcha: dict[str, typing.Any] is easily happily into dict[str, str]
+        headers: dict[str, typing.Any] = {"key": 123}
+
+        @broker.subscriber(destination := faker.pystr())
+        def handle(body: str) -> None: ...
+
+        async with faststream_stomp.TestStompBroker(broker) as br:
+            with pytest.raises(TypeError):
+                await br.publish(faker.pystr(), destination, headers=headers)
+
     async def test_publish_pydantic(self, faker: faker.Faker, broker: faststream_stomp.StompBroker) -> None:
         class SomePydanticModel(pydantic.BaseModel):
             foo: str