aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/bot/cogs/test_antimalware.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/bot/cogs/test_antimalware.py b/tests/bot/cogs/test_antimalware.py
index b4e31b5ce..407fa05c1 100644
--- a/tests/bot/cogs/test_antimalware.py
+++ b/tests/bot/cogs/test_antimalware.py
@@ -23,6 +23,15 @@ class AntiMalwareCogTests(unittest.TestCase):
self.message = MockMessage()
self.message.delete = AsyncMock()
+ def test_message_with_allowed_attachment(self):
+ """Messages with allowed extensions should not be deleted"""
+ attachment = MockAttachment(filename=f"python.{AntiMalwareConfig.whitelist[0]}")
+ self.message.attachments = [attachment]
+
+ coroutine = self.cog.on_message(self.message)
+ asyncio.run(coroutine)
+ self.message.delete.assert_not_called()
+
def test_message_without_attachment(self):
"""Messages without attachments should result in no action."""
coroutine = self.cog.on_message(self.message)