Cleans Up Tests

1 files changed, 5 insertions, 5 deletions

diff --git a/tests/bot/exts/moderation/infraction/test_infractions.py b/tests/bot/exts/moderation/infraction/test_infractions.py
index 4ba5a4feb..13efee054 100644
--- a/tests/bot/exts/moderation/infraction/test_infractions.py
+++ b/tests/bot/exts/moderation/infraction/test_infractions.py
@@ -3,8 +3,9 @@ import unittest
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 from bot.constants import Event
+from bot.exts.moderation.infraction import _utils
 from bot.exts.moderation.infraction.infractions import Infractions
-from tests.helpers import MockBot, MockContext, MockGuild, MockMember, MockRole, MockUser
+from tests.helpers import MockBot, MockContext, MockGuild, MockMember, MockRole, MockUser, autospec
 
 
 class TruncationTests(unittest.IsolatedAsyncioTestCase):
@@ -164,9 +165,8 @@ class VoiceBanTests(unittest.IsolatedAsyncioTestCase):
         )
         self.cog.apply_infraction.assert_awaited_once_with(self.ctx, {"foo": "bar"}, self.user, "my_return_value")
 
-    @patch("bot.exts.moderation.infraction._utils.get_active_infraction", return_value=None)
-    @patch("bot.exts.moderation.infraction._utils.post_infraction")
-    @patch("bot.exts.moderation.infraction.infractions.Infractions.apply_infraction")
+    @autospec(_utils, "post_infraction", "get_active_infraction", return_value=None)
+    @autospec(Infractions, "apply_infraction")
     async def test_voice_ban_user_left_guild(self, apply_infraction_mock, post_infraction_mock, _):
         """Should voice ban user that left the guild without throwing an error."""
         infraction = {"foo": "bar"}
@@ -175,7 +175,7 @@ class VoiceBanTests(unittest.IsolatedAsyncioTestCase):
         user = MockUser()
         await self.cog.voiceban(self.cog, self.ctx, user, reason=None)
         post_infraction_mock.assert_called_once_with(self.ctx, user, "voice_ban", None, active=True)
-        apply_infraction_mock.assert_called_once_with(self.ctx, infraction, user, None)
+        apply_infraction_mock.assert_called_once_with(self.cog, self.ctx, infraction, user, None)
 
     async def test_voice_unban_user_not_found(self):
         """Should include info to return dict when user was not found from guild."""