test: multitenant route manager conn from recip

Signed-off-by: Daniel Bluhm <dbluhm@pm.me>

Author: dbluhm

aries_cloudagent/multitenant/route_manager.py

@@ -15,7 +15,7 @@
 from ..protocols.routing.v1_0.manager import RoutingManager
 from ..protocols.routing.v1_0.models.route_record import RouteRecord
 from ..storage.error import StorageNotFoundError
-from .manager import MultitenantManager
+from .base import BaseMultitenantManager
 
 
 LOGGER = logging.getLogger(__name__)
@@ -117,7 +117,7 @@ async def connection_from_recipient_key(
         for sub wallets, we check the sub wallet's connections before the base
         wallet.
         """
-        manager = MultitenantManager(self.root_profile)
+        manager = self.root_profile.inject(BaseMultitenantManager)
         profile_to_search = (
             await manager.get_profile_for_key(profile.context, recipient_key) or profile
         )

aries_cloudagent/multitenant/tests/test_route_manager.py

@@ -8,9 +8,11 @@
 from ...protocols.coordinate_mediation.v1_0.models.mediation_record import (
     MediationRecord,
 )
+from ...protocols.coordinate_mediation.v1_0.route_manager import RouteManager
 from ...protocols.routing.v1_0.manager import RoutingManager
 from ...protocols.routing.v1_0.models.route_record import RouteRecord
 from ...storage.error import StorageNotFoundError
+from ..base import BaseMultitenantManager
 from ..route_manager import MultitenantRouteManager
 
 TEST_RECORD_VERKEY = "3Dn1SJNPaCXcvvJvSbsFWP2xaCjMom3can8CQNhWrTRx"
@@ -360,3 +362,21 @@ async def test_routing_info_with_base_mediator_and_sub_mediator(
         )
     assert keys == [*base_mediation_record.routing_keys, *mediation_record.routing_keys]
     assert endpoint == mediation_record.endpoint
+
+
+@pytest.mark.asyncio
+async def test_connection_from_recipient_key(
+    sub_profile: Profile, route_manager: MultitenantRouteManager
+):
+    manager = mock.MagicMock()
+    manager.get_profile_for_key = mock.CoroutineMock(return_value=sub_profile)
+    route_manager.root_profile.context.injector.bind_instance(
+        BaseMultitenantManager, manager
+    )
+    with mock.patch.object(
+        RouteManager, "connection_from_recipient_key", mock.CoroutineMock()
+    ) as mock_conn_for_recip:
+        result = await route_manager.connection_from_recipient_key(
+            route_manager.root_profile, TEST_VERKEY
+        )
+        assert result == mock_conn_for_recip.return_value