aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar MarkKoz <[email protected]>2020-05-11 10:37:59 -0700
committerGravatar MarkKoz <[email protected]>2020-05-11 12:03:12 -0700
commite4790b330da1605573b5d23615bfe62b481e1e04 (patch)
treea385f6e2a334ce3122838105bd88e4ea9fea1f31
parentSimplify token remover's message mock (diff)
Test token remover's message deletion
-rw-r--r--tests/bot/cogs/test_token_remover.py9
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 = (