chore: polish docs

airtai · Dec 30, 2024 · 6fb8eae · 6fb8eae
1 parent 313d823
commit 6fb8eae
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 12 deletions.
diff --git a/docs/docs/en/rabbit/declare.md b/docs/docs/en/rabbit/declare.md
@@ -30,6 +30,6 @@ These methods require just one argument (`RabbitQueue`/`RabbitExchange`) contain
 
 It is also possible to bind a queue and an exchange using low-level **aio-pika** `RobustQueue.bind` method:
 
-```python linenums="1" hl_lines="15-20 22-27"
+```python linenums="1" hl_lines="23-25 27-29 31-34"
 {! docs_src/rabbit/bind.py !}
 ```
diff --git a/docs/docs_src/rabbit/bind.py b/docs/docs_src/rabbit/bind.py
@@ -1,3 +1,4 @@
+import aio_pika
 from faststream import FastStream
 from faststream.rabbit import (
     ExchangeType,
@@ -10,10 +11,7 @@
 app = FastStream(broker)
 
 
-some_queue = RabbitQueue(
-    name="some-queue",
-    durable=True,
-)
+some_queue = RabbitQueue("some-queue")
 
 some_exchange = RabbitExchange(
     name="some-exchange",
@@ -22,16 +20,15 @@
 
 @app.after_startup
 async def bind_queue_exchange():
-    queue = await broker.declare_queue(
+    queue: aio_pika.RobustQueue = await broker.declare_queue(
         some_queue
-    ) # This gives a aio_pika.RobustQueue object
+    )
 
-    exchange = await broker.declare_exchange(
+    exchange: aio_pika.RobustExchange = await broker.declare_exchange(
         some_exchange
     )
 
     await queue.bind(
-        queue=queue,
         exchange=exchange,
-        routing_key=queue.name # This parameter is optional
+        routing_key=queue.name  # Optional parameter
     )
diff --git a/tests/a_docs/rabbit/test_bind.py b/tests/a_docs/rabbit/test_bind.py
@@ -15,15 +15,16 @@ async def test_bind(monkeypatch, async_mock: AsyncMock):
 
     with monkeypatch.context() as m:
         m.setattr(RobustQueue, "bind", async_mock)
+
         async with TestApp(app):
             assert len(broker.declarer._RabbitDeclarer__queues) == 2  # with `reply-to`
             assert len(broker.declarer._RabbitDeclarer__exchanges) == 1
 
             assert some_queue in broker.declarer._RabbitDeclarer__queues
             assert some_exchange in broker.declarer._RabbitDeclarer__exchanges
 
+            row_exchange = await broker.declarer.declare_exchange(some_exchange)
             async_mock.assert_awaited_once_with(
-                queue=broker.declarer._RabbitDeclarer__queues[some_queue],
-                exchange=broker.declarer._RabbitDeclarer__exchanges[some_exchange],
+                exchange=row_exchange,
                 routing_key=some_queue.name,
             )