Register StompBroker for AsyncAPI UI (compat 0.6.7) (#181)

Author: vrslev

packages/faststream-stomp/faststream_stomp/__init__.py

@@ -16,3 +16,21 @@
     "StompSubscriber",
     "TestStompBroker",
 ]
+
+try:  # noqa: RUF067
+    import functools
+    import typing
+
+    import faststream.asgi.factories.asyncapi.try_it_out
+    from faststream._internal.broker import BrokerUsecase
+    from faststream._internal.testing.broker import TestBroker
+
+    original_get_broker_registry = faststream.asgi.factories.asyncapi.try_it_out._get_broker_registry
+
+    @functools.lru_cache(maxsize=1)
+    def get_broker_registry() -> dict[type[BrokerUsecase[typing.Any, typing.Any]], type[TestBroker[typing.Any]]]:
+        return {**original_get_broker_registry(), StompBroker: TestStompBroker}
+
+    faststream.asgi.factories.asyncapi.try_it_out._get_broker_registry = get_broker_registry
+except Exception:  # noqa: BLE001, S110
+    pass