Skip to content

Commit 9a8e9fe

Browse files
test: [FC-0047] add check for push notif channel
1 parent c1ea794 commit 9a8e9fe

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

edx_ace/tests/channel/test_channel_helpers.py

+3
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from edx_ace.channel import ChannelMap, ChannelType, get_channel_for_message
99
from edx_ace.channel.braze import BrazeEmailChannel
1010
from edx_ace.channel.file import FileEmailChannel
11+
from edx_ace.channel.push_notification import PushNotificationChannel
1112
from edx_ace.errors import UnsupportedChannelError
1213
from edx_ace.message import Message
1314
from edx_ace.recipient import Recipient
@@ -45,6 +46,7 @@ def test_get_channel_for_message(self):
4546
channel_map = ChannelMap([
4647
['file_email', FileEmailChannel()],
4748
['braze_email', BrazeEmailChannel()],
49+
['push_notifications', PushNotificationChannel()],
4850
])
4951

5052
transactional_msg = Message(options={'transactional': True}, **self.msg_kwargs)
@@ -58,6 +60,7 @@ def test_get_channel_for_message(self):
5860
assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_msg), FileEmailChannel)
5961
assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_campaign_msg), BrazeEmailChannel)
6062
assert isinstance(get_channel_for_message(ChannelType.EMAIL, info_msg), BrazeEmailChannel)
63+
assert isinstance(get_channel_for_message(ChannelType.PUSH, info_msg), PushNotificationChannel)
6164

6265
with self.assertRaises(UnsupportedChannelError):
6366
assert get_channel_for_message(ChannelType.PUSH, transactional_msg)

0 commit comments

Comments
 (0)