fix (#2013): allow to create publisher in already connected broker (#2024)

* fix: allow to create publisher in connected broker

* chore: bump version

Author: Lancetnik

faststream/__about__.py

@@ -1,5 +1,5 @@
 """Simple and fast framework to create message brokers based microservices."""
 
-__version__ = "0.5.33"
+__version__ = "0.5.34"
 
 SERVICE_NAME = f"faststream-{__version__}"

faststream/broker/core/usecase.py

@@ -285,7 +285,7 @@ def _publisher_setup_extra(self) -> "AnyDict":
 
     def publisher(self, *args: Any, **kwargs: Any) -> "PublisherProto[MsgType]":
         pub = super().publisher(*args, **kwargs)
-        if self.running:
+        if self.running or self._connection is not None:
             self.setup_publisher(pub)
         return pub
 

tests/brokers/base/publish.py

@@ -578,6 +578,12 @@ async def handler(m):
         assert event.is_set()
         assert not mock.called
 
+    @pytest.mark.asyncio
+    async def test_publisher_after_connect(self, queue: str):
+        async with self.patch_broker(self.get_broker()) as br:
+            # Should pass without error
+            await br.publisher(queue).publish(None)
+
     @pytest.mark.asyncio
     async def test_publisher_after_start(
         self,