:wq

erge branch 'asyncapi-post-form' of github.com:vvlrff/faststream into asyncapi-post-form

Author: Lancetnik

faststream/rabbit/testing.py

@@ -6,7 +6,7 @@
 
 import aiormq
 import anyio
-from aio_pika.message import IncomingMessage
+from aio_pika.message import IncomingMessage, encode_expiration
 from pamqp import commands as spec
 from pamqp.header import ContentHeader
 from typing_extensions import override
@@ -178,6 +178,7 @@ def build_message(
                     content_encoding=msg.content_encoding,
                     priority=msg.priority,
                     correlation_id=msg.correlation_id,
+                    expiration=encode_expiration(msg.expiration),
                     message_id=msg.message_id,
                     timestamp=msg.timestamp,
                     message_type=message_type,

pyproject.toml

@@ -131,6 +131,7 @@ test-core = [
     "pytest-xdist>=3.8.0",
     "pytest-cov>=6.2.1",
     "covdefaults>=2.3.0",
+    "freezegun>=1.5.5",
 ]
 
 testing = [

tests/brokers/rabbit/test_test_client.py

@@ -1,7 +1,10 @@
 import asyncio
+import datetime as dt
 from typing import Any
+from unittest.mock import MagicMock
 
 import pytest
+from freezegun import freeze_time
 
 from faststream import BaseMiddleware
 from faststream.exceptions import SubscriberNotFound
@@ -18,6 +21,8 @@
 from .basic import RabbitMemoryTestcaseConfig
 from .test_publish import TestPublishWithExchange as PublishWithExchangeCase
 
+_frozen_time = dt.datetime(2026, 2, 10, 12, 0, 0, tzinfo=dt.timezone.utc)
+
 
 @pytest.mark.rabbit()
 @pytest.mark.asyncio()
@@ -187,6 +192,35 @@ async def test_broker_with_real_patches_publishers_and_subscribers(
     ) -> None:
         await super().test_broker_with_real_patches_publishers_and_subscribers(queue)
 
+    @pytest.mark.asyncio()
+    @pytest.mark.parametrize(
+        ("expiration", "expected"),
+        (
+            pytest.param(None, None, id="none"),
+            pytest.param(1, 1, id="int"),
+            pytest.param(1.5, 1.5, id="float"),
+            pytest.param(dt.timedelta(seconds=1.1), 1.1, id="timedelta"),
+            pytest.param(_frozen_time, 0, id="datetime"),
+        ),
+    )
+    @freeze_time(_frozen_time)
+    async def test_publish_expiration_propagated(
+        self, expiration: Any, expected: Any, queue: str, mock: MagicMock
+    ) -> None:
+        broker = self.get_broker(apply_types=True)
+
+        args, kwargs = self.get_subscriber_params(queue)
+
+        @broker.subscriber(*args, **kwargs)
+        async def m(msg: RabbitMessage) -> None:
+            mock(msg)
+
+        async with self.patch_broker(broker) as br:
+            await br.start()
+            await br.publish("hello", queue, expiration=expiration)
+            msg = mock.call_args[0][0]
+            assert msg.raw_message.expiration == expected
+
 
 @pytest.mark.parametrize(
     ("pattern", "current", "result"),