From 4a0b3ea1ef182ddbbb1f9d731b28768a049a531d Mon Sep 17 00:00:00 2001
From: Jannes Jonkers <jjonkers2@gmail.com>
Date: Thu, 7 May 2020 20:23:00 +0200
Subject: AntiMalware Tests - Added unittest for cog setup

---
 tests/bot/cogs/test_antimalware.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'tests')

diff --git a/tests/bot/cogs/test_antimalware.py b/tests/bot/cogs/test_antimalware.py
index da5cd9d11..67c640d23 100644
--- a/tests/bot/cogs/test_antimalware.py
+++ b/tests/bot/cogs/test_antimalware.py
@@ -90,3 +90,13 @@ class AntiMalwareCogTests(unittest.TestCase):
         self.assertIn(
             f"INFO:bot.cogs.antimalware:Tried to delete message `{self.message.id}`, but message could not be found.",
             logs.output)
+
+
+class AntiMalwareSetupTests(unittest.TestCase):
+    """Tests setup of the `AntiMalware` cog."""
+
+    def test_setup(self):
+        """Setup of the extension should call add_cog."""
+        bot = MockBot()
+        antimalware.setup(bot)
+        bot.add_cog.assert_called_once()
-- 
cgit v1.2.3