Add customer io notification (#6306)

airbytehq · May 2, 2023 · d7f36bd · d7f36bd
1 parent 357f82d
commit d7f36bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/airbyte-commons-temporal/src/main/java/io/airbyte/commons/temporal/NotificationClient.java b/airbyte-commons-temporal/src/main/java/io/airbyte/commons/temporal/NotificationClient.java
@@ -72,7 +72,7 @@ private void callNotificationWorkflow(final UUID connectionId,
       throw new RuntimeException(e);
     }
     try {
-      notificationWorkflow.sendNotification(connectionId, "", message, List.of(NotificationType.webhook));
+      notificationWorkflow.sendNotification(connectionId, "", message, List.of(NotificationType.webhook, NotificationType.customerio));
     } catch (final RuntimeException e) {
       log.error("There was an error while sending a Schema Change Notification", e);
       throw e;

diff --git a/...te-commons-temporal/src/test/java/io/airbyte/commons/temporal/NotificationClientTest.java b/...te-commons-temporal/src/test/java/io/airbyte/commons/temporal/NotificationClientTest.java
@@ -19,9 +19,11 @@
 import io.airbyte.featureflag.FeatureFlagClient;
 import io.airbyte.featureflag.TestClient;
 import io.airbyte.featureflag.UseNotificationWorkflow;
+import io.airbyte.notification.NotificationType;
 import io.airbyte.validation.json.JsonValidationException;
 import io.temporal.client.WorkflowClient;
 import java.io.IOException;
+import java.util.List;
 import java.util.UUID;
 import org.junit.jupiter.api.Test;
 
@@ -49,7 +51,7 @@ void testCallNewNotifyWorkflow() {
 
     notificationClient.sendSchemaChangeNotification(connectionId, "", false);
 
-    verify(notificationWorkflow).sendNotification(eq(connectionId), any(), any(), any());
+    verify(notificationWorkflow).sendNotification(eq(connectionId), any(), any(), eq(List.of(NotificationType.webhook, NotificationType.customerio)));
   }
 
   @Test