diff options
| author | 2020-05-11 10:37:59 -0700 | |
|---|---|---|
| committer | 2020-05-11 12:03:12 -0700 | |
| commit | e4790b330da1605573b5d23615bfe62b481e1e04 (patch) | |
| tree | a385f6e2a334ce3122838105bd88e4ea9fea1f31 | |
| parent | Simplify token remover's message mock (diff) | |
Test token remover's message deletion
| -rw-r--r-- | tests/bot/cogs/test_token_remover.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/bot/cogs/test_token_remover.py b/tests/bot/cogs/test_token_remover.py index 5ca863926..d65ce2ce5 100644 --- a/tests/bot/cogs/test_token_remover.py +++ b/tests/bot/cogs/test_token_remover.py @@ -209,6 +209,15 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase): if user_return: valid_time.assert_called_once_with("y") + async def test_delete_message(self): + """The message should be deleted, and a message should be sent to the same channel.""" + await TokenRemover.delete_message(self.msg) + + self.msg.delete.assert_called_once_with() + self.msg.channel.send.assert_called_once_with( + DELETION_MESSAGE_TEMPLATE.format(mention=self.msg.author.mention) + ) + def test_censors_valid_tokens(self): """Valid tokens are censored.""" cases = ( |