test: add new overload

airtai · Jan 5, 2025 · 9d769d9 · 9d769d9
1 parent 5296158
commit 9d769d9
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/faststream/redis/broker/broker.py b/faststream/redis/broker/broker.py
@@ -372,13 +372,27 @@ def _subscriber_setup_extra(self) -> "AnyDict":
     async def publish(  # type: ignore[override]
             self,
             message: "SendableMessage" = None,
+            channel: None = None,
+            *,
+            reply_to: str = "",
+            headers: Optional["AnyDict"] = None,
+            correlation_id: Optional[str] = None,
+            list: Optional[str] = None,
+            stream: Optional[str] = None,
+            maxlen: Optional[int] = None,
+    ) -> None: ...
+
+    @overload
+    async def publish(  # type: ignore[override]
+            self,
+            message: None = None,
             channel: Optional[str] = None,
             *,
             reply_to: str = "",
             headers: Optional["AnyDict"] = None,
             correlation_id: Optional[str] = None,
             list: Optional[str] = None,
-            stream: None = None,
+            stream: Optional[str] = None,
             maxlen: Optional[int] = None,
     ) -> None: ...