File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -333,7 +333,7 @@ async def publish(
333
333
** kwargs : Any ,
334
334
) -> Optional [Any ]:
335
335
"""Publish message directly."""
336
- assert producer , NOT_CONNECTED_YET # nosec B101)
336
+ assert producer , NOT_CONNECTED_YET # nosec B101
337
337
338
338
publish : "AsyncFunc" = producer .publish
339
339
for m in self ._middlewares :
Original file line number Diff line number Diff line change 16
16
from unittest .mock import AsyncMock , MagicMock
17
17
18
18
from faststream .broker .core .usecase import BrokerUsecase
19
+ from faststream .broker .message import StreamMessage , decode_message , encode_message
19
20
from faststream .broker .middlewares .logging import CriticalLogMiddleware
20
21
from faststream .broker .wrapper .call import HandlerCallWrapper
21
22
from faststream .testing .app import TestApp
@@ -212,6 +213,9 @@ async def call_handler(
212
213
result = await handler .consume (message )
213
214
214
215
if rpc :
215
- return result
216
+ message_body , content_type = encode_message (result )
217
+ msg_to_publish = StreamMessage (raw_message = None , body = message_body , content_type = content_type )
218
+ consumed_data = decode_message (msg_to_publish )
219
+ return consumed_data
216
220
217
221
return None
You can’t perform that action at this time.
0 commit comments