Add sub wallet created event (#3946)

* add wallet created event

Signed-off-by: Patrick St-Louis <patrick.st-louis@opsecid.ca>

* linting

Signed-off-by: Patrick St-Louis <patrick.st-louis@opsecid.ca>

---------

Signed-off-by: Patrick St-Louis <patrick.st-louis@opsecid.ca>

Author: PatStLouis

acapy_agent/core/util.py

@@ -7,6 +7,11 @@
 STARTUP_EVENT_PATTERN = re.compile(f"^{STARTUP_EVENT_TOPIC}?$")
 SHUTDOWN_EVENT_TOPIC = CORE_EVENT_PREFIX + "shutdown"
 SHUTDOWN_EVENT_PATTERN = re.compile(f"^{SHUTDOWN_EVENT_TOPIC}?$")
+MULTITENANT_EVENT_PREFIX = CORE_EVENT_PREFIX + "multitenant::"
+MULTITENANT_WALLET_CREATED_TOPIC = MULTITENANT_EVENT_PREFIX + "wallet::created"
+MULTITENANT_WALLET_CREATED_PATTERN = re.compile(
+    f"^{MULTITENANT_WALLET_CREATED_TOPIC}::[a-zA-Z0-9-]+$"
+)
 WARNING_DEGRADED_FEATURES = "version-with-degraded-features"
 WARNING_VERSION_MISMATCH = "fields-ignored-due-to-version-mismatch"
 WARNING_VERSION_NOT_SUPPORTED = "version-not-supported"

acapy_agent/multitenant/admin/routes.py

@@ -13,7 +13,9 @@
 from ...admin.decorators.auth import admin_authentication
 from ...admin.request_context import AdminRequestContext
 from ...core.error import BaseError
+from ...core.event_bus import Event, EventBus
 from ...core.profile import ProfileManagerProvider
+from ...core.util import MULTITENANT_WALLET_CREATED_TOPIC
 from ...messaging.models.base import BaseModelError
 from ...messaging.models.openapi import OpenAPISchema
 from ...messaging.models.paginated_query import (
@@ -486,6 +488,20 @@ async def wallet_create(request: web.BaseRequest):
             context, wallet_record, extra_settings=settings
         )
         await attempt_auto_author_with_endorser_setup(wallet_profile)
+
+        event_bus = context.profile.inject_or(EventBus)
+        if event_bus:
+            await event_bus.notify(
+                context.profile,
+                Event(
+                    f"{MULTITENANT_WALLET_CREATED_TOPIC}::{wallet_record.wallet_id}",
+                    {
+                        "wallet_id": wallet_record.wallet_id,
+                        "wallet_name": wallet_record.wallet_name,
+                        "settings": wallet_record.settings,
+                    },
+                ),
+            )
     except BaseError as err:
         raise web.HTTPBadRequest(reason=err.roll_up) from err
 

acapy_agent/multitenant/admin/tests/test_routes.py

@@ -3,6 +3,8 @@
 from marshmallow.exceptions import ValidationError
 
 from ....admin.request_context import AdminRequestContext
+from ....core.event_bus import EventBus, MockEventBus
+from ....core.util import MULTITENANT_WALLET_CREATED_TOPIC
 from ....messaging.models.base import BaseModelError
 from ....storage.error import StorageError, StorageNotFoundError
 from ....tests import mock
@@ -279,6 +281,59 @@ async def test_wallet_create(self):
             assert mock_multitenant_mgr.get_wallet_profile.called
             assert test_module.attempt_auto_author_with_endorser_setup.called
 
+    async def test_wallet_create_emits_wallet_created_event(self):
+        body = {
+            "wallet_name": "event-test",
+            "wallet_type": "askar",
+            "wallet_key": "test",
+            "key_management_mode": "managed",
+            "wallet_webhook_urls": [],
+            "wallet_dispatch_type": "default",
+        }
+        self.request.json = mock.CoroutineMock(return_value=body)
+
+        test_module.attempt_auto_author_with_endorser_setup = mock.CoroutineMock()
+
+        mock_event_bus = MockEventBus()
+        self.profile.context.injector.bind_instance(EventBus, mock_event_bus)
+
+        wallet_mock = mock.MagicMock(
+            serialize=mock.MagicMock(
+                return_value={
+                    "wallet_id": "event-wallet-id",
+                    "settings": {},
+                    "key_management_mode": body["key_management_mode"],
+                }
+            )
+        )
+        wallet_mock.wallet_id = "event-wallet-id"
+        wallet_mock.wallet_name = body["wallet_name"]
+        wallet_mock.settings = {"wallet.name": body["wallet_name"]}
+
+        mock_multitenant_mgr = mock.AsyncMock(BaseMultitenantManager, autospec=True)
+        mock_multitenant_mgr.create_wallet = mock.CoroutineMock(return_value=wallet_mock)
+        mock_multitenant_mgr.create_auth_token = mock.CoroutineMock(
+            return_value="event-token"
+        )
+        mock_multitenant_mgr.get_wallet_profile = mock.CoroutineMock(
+            return_value=mock.MagicMock()
+        )
+        self.profile.context.injector.bind_instance(
+            BaseMultitenantManager, mock_multitenant_mgr
+        )
+
+        await test_module.wallet_create(self.request)
+
+        await mock_event_bus.task_queue.wait_for_completion()
+
+        assert mock_event_bus.events
+        _, event = mock_event_bus.events[-1]
+        assert event.topic == (
+            f"{MULTITENANT_WALLET_CREATED_TOPIC}::{wallet_mock.wallet_id}"
+        )
+        assert event.payload["wallet_id"] == wallet_mock.wallet_id
+        assert event.payload["wallet_name"] == wallet_mock.wallet_name
+
     async def test_wallet_create_x(self):
         body = {}
         self.request.json = mock.CoroutineMock(return_value=body)