Update tests to use new async cog setup function

7 files changed, 28 insertions, 29 deletions

diff --git a/tests/bot/exts/backend/sync/test_cog.py b/tests/bot/exts/backend/sync/test_cog.py
index 4ec36e39f..ce620aa8d 100644
--- a/tests/bot/exts/backend/sync/test_cog.py
+++ b/tests/bot/exts/backend/sync/test_cog.py
@@ -16,11 +16,11 @@ class SyncExtensionTests(unittest.IsolatedAsyncioTestCase):
     """Tests for the sync extension."""
 
     @staticmethod
-    def test_extension_setup():
+    async def test_extension_setup():
         """The Sync cog should be added."""
         bot = helpers.MockBot()
-        sync.setup(bot)
-        bot.add_cog.assert_called_once()
+        await sync.setup(bot)
+        bot.add_cog.assert_awaited_once()
 
 
 class SyncCogTestCase(unittest.IsolatedAsyncioTestCase):
diff --git a/tests/bot/exts/backend/test_error_handler.py b/tests/bot/exts/backend/test_error_handler.py
index 04a018289..193f1d822 100644
--- a/tests/bot/exts/backend/test_error_handler.py
+++ b/tests/bot/exts/backend/test_error_handler.py
@@ -544,11 +544,11 @@ class IndividualErrorHandlerTests(unittest.IsolatedAsyncioTestCase):
                 push_scope_mock.set_extra.has_calls(set_extra_calls)
 
 
-class ErrorHandlerSetupTests(unittest.TestCase):
+class ErrorHandlerSetupTests(unittest.IsolatedAsyncioTestCase):
     """Tests for `ErrorHandler` `setup` function."""
 
-    def test_setup(self):
+    async def test_setup(self):
         """Should call `bot.add_cog` with `ErrorHandler`."""
         bot = MockBot()
-        setup(bot)
-        bot.add_cog.assert_called_once()
+        await setup(bot)
+        bot.add_cog.assert_awaited_once()
diff --git a/tests/bot/exts/events/test_code_jams.py b/tests/bot/exts/events/test_code_jams.py
index 0856546af..684f7abcd 100644
--- a/tests/bot/exts/events/test_code_jams.py
+++ b/tests/bot/exts/events/test_code_jams.py
@@ -160,11 +160,11 @@ class JamCodejamCreateTests(unittest.IsolatedAsyncioTestCase):
                 member.add_roles.assert_not_awaited()
 
 
-class CodeJamSetup(unittest.TestCase):
+class CodeJamSetup(unittest.IsolatedAsyncioTestCase):
     """Test for `setup` function of `CodeJam` cog."""
 
-    def test_setup(self):
+    async def test_setup(self):
         """Should call `bot.add_cog`."""
         bot = MockBot()
-        code_jams.setup(bot)
-        bot.add_cog.assert_called_once()
+        await code_jams.setup(bot)
+        bot.add_cog.assert_awaited_once()
diff --git a/tests/bot/exts/filters/test_antimalware.py b/tests/bot/exts/filters/test_antimalware.py
index 06d78de9d..7282334e2 100644
--- a/tests/bot/exts/filters/test_antimalware.py
+++ b/tests/bot/exts/filters/test_antimalware.py
@@ -192,11 +192,11 @@ class AntiMalwareCogTests(unittest.IsolatedAsyncioTestCase):
                 self.assertCountEqual(disallowed_extensions, expected_disallowed_extensions)
 
 
-class AntiMalwareSetupTests(unittest.TestCase):
+class AntiMalwareSetupTests(unittest.IsolatedAsyncioTestCase):
     """Tests setup of the `AntiMalware` cog."""
 
-    def test_setup(self):
+    async def test_setup(self):
         """Setup of the extension should call add_cog."""
         bot = MockBot()
-        antimalware.setup(bot)
-        bot.add_cog.assert_called_once()
+        await antimalware.setup(bot)
+        bot.add_cog.assert_awaited_once()
diff --git a/tests/bot/exts/filters/test_security.py b/tests/bot/exts/filters/test_security.py
index c0c3baa42..007b7b1eb 100644
--- a/tests/bot/exts/filters/test_security.py
+++ b/tests/bot/exts/filters/test_security.py
@@ -1,5 +1,4 @@
 import unittest
-from unittest.mock import MagicMock
 
 from discord.ext.commands import NoPrivateMessage
 
@@ -44,11 +43,11 @@ class SecurityCogTests(unittest.TestCase):
         self.assertTrue(self.cog.check_on_guild(self.ctx))
 
 
-class SecurityCogLoadTests(unittest.TestCase):
+class SecurityCogLoadTests(unittest.IsolatedAsyncioTestCase):
     """Tests loading the `Security` cog."""
 
-    def test_security_cog_load(self):
+    async def test_security_cog_load(self):
         """Setup of the extension should call add_cog."""
-        bot = MagicMock()
-        security.setup(bot)
-        bot.add_cog.assert_called_once()
+        bot = MockBot()
+        await security.setup(bot)
+        bot.add_cog.assert_awaited_once()
diff --git a/tests/bot/exts/filters/test_token_remover.py b/tests/bot/exts/filters/test_token_remover.py
index 4db27269a..c1f3762ac 100644
--- a/tests/bot/exts/filters/test_token_remover.py
+++ b/tests/bot/exts/filters/test_token_remover.py
@@ -395,15 +395,15 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
         self.msg.channel.send.assert_not_awaited()
 
 
-class TokenRemoverExtensionTests(unittest.TestCase):
+class TokenRemoverExtensionTests(unittest.IsolatedAsyncioTestCase):
     """Tests for the token_remover extension."""
 
     @autospec("bot.exts.filters.token_remover", "TokenRemover")
-    def test_extension_setup(self, cog):
+    async def test_extension_setup(self, cog):
         """The TokenRemover cog should be added."""
         bot = MockBot()
-        token_remover.setup(bot)
+        await token_remover.setup(bot)
 
         cog.assert_called_once_with(bot)
-        bot.add_cog.assert_called_once()
+        bot.add_cog.assert_awaited_once()
         self.assertTrue(isinstance(bot.add_cog.call_args.args[0], TokenRemover))
diff --git a/tests/bot/exts/utils/test_snekbox.py b/tests/bot/exts/utils/test_snekbox.py
index f68a20089..3c555c051 100644
--- a/tests/bot/exts/utils/test_snekbox.py
+++ b/tests/bot/exts/utils/test_snekbox.py
@@ -403,11 +403,11 @@ class SnekboxTests(unittest.IsolatedAsyncioTestCase):
                 self.assertEqual(actual, expected)
 
 
-class SnekboxSetupTests(unittest.TestCase):
+class SnekboxSetupTests(unittest.IsolatedAsyncioTestCase):
     """Tests setup of the `Snekbox` cog."""
 
-    def test_setup(self):
+    async def test_setup(self):
         """Setup of the extension should call add_cog."""
         bot = MockBot()
-        snekbox.setup(bot)
-        bot.add_cog.assert_called_once()
+        await snekbox.setup(bot)
+        bot.add_cog.assert_awaited_once()