test: [FC-0047] add check for push notif channel

openedx · Jul 25, 2024 · c4a3459 · c4a3459
1 parent c1ea794
commit c4a3459
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/edx_ace/tests/channel/test_channel_helpers.py b/edx_ace/tests/channel/test_channel_helpers.py
@@ -8,6 +8,7 @@
 from edx_ace.channel import ChannelMap, ChannelType, get_channel_for_message
 from edx_ace.channel.braze import BrazeEmailChannel
 from edx_ace.channel.file import FileEmailChannel
+from edx_ace.channel.push_notification import PushNotificationChannel
 from edx_ace.errors import UnsupportedChannelError
 from edx_ace.message import Message
 from edx_ace.recipient import Recipient
@@ -58,6 +59,7 @@ def test_get_channel_for_message(self):
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_msg), FileEmailChannel)
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_campaign_msg), BrazeEmailChannel)
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, info_msg), BrazeEmailChannel)
+            assert isinstance(get_channel_for_message(ChannelType.PUSH, info_msg), PushNotificationChannel)
 
             with self.assertRaises(UnsupportedChannelError):
                 assert get_channel_for_message(ChannelType.PUSH, transactional_msg)