Add tests for reset_slowmode

1 files changed, 22 insertions, 3 deletions

diff --git a/tests/bot/cogs/test_slowmode.py b/tests/bot/cogs/test_slowmode.py
index fb9f3c9ad..a2e5ad346 100644
--- a/tests/bot/cogs/test_slowmode.py
+++ b/tests/bot/cogs/test_slowmode.py
@@ -2,6 +2,7 @@ import unittest
 from unittest import mock
 
 from bot.cogs.slowmode import Slowmode
+from bot.constants import Emojis
 from tests.helpers import MockBot, MockContext, MockTextChannel
 
 
@@ -14,7 +15,7 @@ class SlowmodeTests(unittest.IsolatedAsyncioTestCase):
         self.ctx = MockContext(channel=self.text_channel)
 
     async def test_get_slowmode_no_channel(self) -> None:
-        """Get slowmode without a given channel"""
+        """Get slowmode without a given channel."""
         self.text_channel.mention = '#python-general'
         self.text_channel.slowmode_delay = 5
 
@@ -22,12 +23,30 @@ class SlowmodeTests(unittest.IsolatedAsyncioTestCase):
         self.ctx.send.assert_called_once_with("The slowmode delay for #python-general is 5 seconds.")
 
     async def test_get_slowmode_with_channel(self) -> None:
-        """Get slowmode without a given channel"""
+        """Get slowmode with a given channel."""
         self.text_channel.mention = '#python-language'
         self.text_channel.slowmode_delay = 2
 
         await self.cog.get_slowmode(self.cog, self.ctx, self.text_channel)
-        self.ctx.send.assert_called_once_with("The slowmode delay for #python-language is 2 seconds.")
+        self.ctx.send.assert_called_once_with('The slowmode delay for #python-language is 2 seconds.')
+
+    async def test_reset_slowmode_no_channel(self) -> None:
+        """Reset slowmode without a given channel."""
+        self.text_channel.mention = '#careers'
+
+        await self.cog.reset_slowmode(self.cog, self.ctx, None)
+        self.ctx.send.assert_called_once_with(
+            f'{Emojis.check_mark} The slowmode delay for #careers has been reset to 0 seconds.'
+        )
+
+    async def test_reset_slowmode_with_channel(self) -> None:
+        """Reset slowmode with a given channel."""
+        self.text_channel.mention = '#meta'
+
+        await self.cog.reset_slowmode(self.cog, self.ctx, self.text_channel)
+        self.ctx.send.assert_called_once_with(
+            f'{Emojis.check_mark} The slowmode delay for #meta has been reset to 0 seconds.'
+        )
 
     @mock.patch("bot.cogs.slowmode.with_role_check")
     @mock.patch("bot.cogs.slowmode.MODERATION_ROLES", new=(1, 2, 3))