def test_clear_notification_when_mention_removed(self) -> None:
mentioned_user = self.example_user('iago')
self.assertEqual(get_apns_badge_count(mentioned_user), 0)
with mock.patch(
'zerver.lib.push_notifications.push_notifications_enabled',
return_value=True):
message_id = self._login_and_send_original_stream_message(
content="@**Iago**", )
self.assertEqual(get_apns_badge_count(mentioned_user), 1)
self._get_queued_data_for_message_update(message_id=message_id,
content="Removed mention")
self.assertEqual(get_apns_badge_count(mentioned_user), 0)
def test_clear_notification_when_mention_removed(
self, mock_push_notifications: mock.MagicMock) -> None:
mentioned_user = self.example_user("iago")
self.assertEqual(get_apns_badge_count(mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(mentioned_user), 0)
message_id = self._login_and_send_original_stream_message(
content="@**Iago**", )
self.assertEqual(get_apns_badge_count(mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(mentioned_user), 1)
self._get_queued_data_for_message_update(message_id=message_id,
content="Removed mention")
self.assertEqual(get_apns_badge_count(mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(mentioned_user), 0)
def test_clear_notification_when_group_mention_removed(self) -> None:
group_mentioned_user = self.example_user('cordelia')
self.assertEqual(get_apns_badge_count(group_mentioned_user), 0)
with mock.patch(
'zerver.lib.push_notifications.push_notifications_enabled',
return_value=True):
message_id = self._login_and_send_original_stream_message(
content="Hello @*hamletcharacters*", )
self.assertEqual(get_apns_badge_count(group_mentioned_user), 1)
self._get_queued_data_for_message_update(
message_id=message_id,
content="Removed group mention",
expect_short_circuit=True,
)
self.assertEqual(get_apns_badge_count(group_mentioned_user), 0)
def test_clear_notification_when_group_mention_removed(
self, mock_push_notifications: mock.MagicMock) -> None:
group_mentioned_user = self.example_user("cordelia")
self.assertEqual(get_apns_badge_count(group_mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(group_mentioned_user), 0)
message_id = self._login_and_send_original_stream_message(
content="Hello @*hamletcharacters*", )
self.assertEqual(get_apns_badge_count(group_mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(group_mentioned_user), 1)
self._get_queued_data_for_message_update(
message_id=message_id,
content="Removed group mention",
expect_short_circuit=True,
)
self.assertEqual(get_apns_badge_count(group_mentioned_user), 0)
self.assertEqual(get_apns_badge_count_future(group_mentioned_user), 0)
