diff options
| author | 2020-06-15 10:49:18 -0700 | |
|---|---|---|
| committer | 2020-06-15 10:49:18 -0700 | |
| commit | 3aecf14419c87e533d47fe082abeb54ca9edb73c (patch) | |
| tree | 1ae8d85b6561b47c673d4fb4dedba1cbe227e06b /tests | |
| parent | Token remover: move bot check to on_message (diff) | |
Token remover: exit early if message already deleted
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/bot/cogs/test_token_remover.py | 15 | 
1 files changed, 14 insertions, 1 deletions
| diff --git a/tests/bot/cogs/test_token_remover.py b/tests/bot/cogs/test_token_remover.py index 98ea9f823..3349caa73 100644 --- a/tests/bot/cogs/test_token_remover.py +++ b/tests/bot/cogs/test_token_remover.py @@ -3,7 +3,7 @@ from re import Match  from unittest import mock  from unittest.mock import MagicMock -from discord import Colour +from discord import Colour, NotFound  from bot import constants  from bot.cogs import token_remover @@ -282,6 +282,19 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):              channel_id=constants.Channels.mod_alerts          ) +    @mock.patch.object(TokenRemover, "mod_log", new_callable=mock.PropertyMock) +    async def test_take_action_delete_failure(self, mod_log_property): +        """Shouldn't send any messages if the token message can't be deleted.""" +        cog = TokenRemover(self.bot) +        mod_log_property.return_value = mock.create_autospec(ModLog, spec_set=True, instance=True) +        self.msg.delete.side_effect = NotFound(MagicMock(), MagicMock()) + +        token = mock.create_autospec(Token, spec_set=True, instance=True) +        await cog.take_action(self.msg, token) + +        self.msg.delete.assert_called_once_with() +        self.msg.channel.send.assert_not_awaited() +  class TokenRemoverExtensionTests(unittest.TestCase):      """Tests for the token_remover extension.""" | 
