chore(notifications): complete initial use case e2e

GaloyMoney · Jan 11, 2024 · 0e11504 · 0e11504
1 parent a62bd40
commit 0e11504
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/core/notifications/src/account_notification_settings/entity.rs b/core/notifications/src/account_notification_settings/entity.rs
@@ -49,6 +49,18 @@ impl EsEntity for AccountNotificationSettings {
 }
 
 impl AccountNotificationSettings {
+    pub fn new(galoy_account_id: GaloyAccountId) -> Self {
+        let id = AccountNotificationSettingsId::new();
+        Self::try_from(EntityEvents::init(
+            id,
+            [AccountNotificationSettingsEvent::Initialized {
+                id,
+                galoy_account_id,
+            }],
+        ))
+        .expect("Could not create default")
+    }
+
     pub fn disable_channel(&mut self, channel: NotificationChannel) {
         if !self.is_channel_enabled(channel) {
             return;

diff --git a/core/notifications/src/account_notification_settings/repo.rs b/core/notifications/src/account_notification_settings/repo.rs
@@ -17,7 +17,7 @@ impl AccountNotificationSettingsRepo {
 
     pub async fn find_for_account_id(
         &self,
-        account_id: GaloyAccountId,
+        account_id: &GaloyAccountId,
     ) -> Result<Option<AccountNotificationSettings>, AccountNotificationSettingsError> {
         let rows = sqlx::query_as!(
             GenericEvent,

diff --git a/core/notifications/src/app/mod.rs b/core/notifications/src/app/mod.rs
@@ -31,11 +31,10 @@ impl NotificationsApp {
         channel: NotificationChannel,
     ) -> Result<AccountNotificationSettings, ApplicationError> {
         let mut account_settings =
-            if let Some(settings) = self.settings.find_for_account_id(account_id).await? {
+            if let Some(settings) = self.settings.find_for_account_id(&account_id).await? {
                 settings
             } else {
-                unimplemented!()
-                // AccountNotificationSettings::default()
+                AccountNotificationSettings::new(account_id)
             };
         account_settings.disable_channel(channel);
         self.settings.persist(&mut account_settings).await?;