Troubles testing publisher (TestRabbitBroker)

[sverdoux]

Hello,

Having troubles testing a RabbitBroker with a TOPIC RabbitExchange. Here is the sample code:
Python 3.11
faststream[rabbitmq] == 0.5.18

from faststream.rabbit import RabbitBroker, RabbitExchange, TestRabbitBroker, ExchangeType
import asyncio

broker = RabbitBroker('amqp://test:test@localhost:5672')
exchange = RabbitExchange('test', ExchangeType.TOPIC, durable=True)

pub1 = broker.publisher(exchange=exchange, routing_key='some.test.key')
pub2 = broker.publisher(exchange=exchange)


async def pub():
    msg = 'message'

    print('pub1')
    await pub1.publish(msg, routing_key='some.test.key')
    assert pub1.mock.call_args is not None  # OK
    pub1.mock.assert_called_with(msg)  # OK

    print('pub2')
    await pub2.publish(msg, routing_key='some.test.key')
    assert pub2.mock.call_args is not None  # NOK
    pub2.mock.assert_called_with(msg)  # NOK


async def main():
    br = TestRabbitBroker(broker)

    await br.__aenter__()
    await broker.start()

    await pub()

    await br.__aexit__(*[None] * 3)

if __name__ == '__main__':
    asyncio.run(main())

The second publisher does not define a default routing key because in my real scenario, this does not make much sense (all publish calls should provide a proper one). It seems that a mismatch is done to associate the pub2 mock to the mock generated by pub2.publish (hash/key of publishers using the routing key ?). Dunno if it is expected, but if a function under test publish several messages to different topics, even if the publisher is defined with one of the keys, all other mocks will not track calls properly ?

Thanks