broker: automaticly set exchange on send

Author: scailer

microagent/broker.py

@@ -160,6 +160,8 @@ class BoundQueue(QueueProtocol):
     queue: Queue
 
     async def send(self, message: dict, **options: Any) -> None:
+        if 'exchange' not in options and self.queue.exchange:
+            options['exchange'] = self.queue.exchange
         await self.broker.send(self.queue.name, self.queue.serialize(message), **options)
 
     async def length(self) -> int:

tests/test_broker.py

@@ -144,7 +144,7 @@ async def test_Broker_send_ok(broker, test_queue):
 
 async def test_Broker_send_ok_to_exchange(broker, else_queue):
     await broker.ex.send({'uid': 1})
-    broker.send.assert_called_once_with('else_queue', '{"uid": 1}')
+    broker.send.assert_called_once_with('else_queue', '{"uid": 1}', exchange='ex')
 
 
 async def test_Broker_prepared_data_ok(broker, test_queue):