From 658f73170100e826fe47ea9f23178a5d071f3932 Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Tue, 19 May 2020 12:50:43 +0300
Subject: EH Tests: Create test for `ErrorHandler` `setup` function

---
 tests/bot/cogs/test_error_handler.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'tests')

diff --git a/tests/bot/cogs/test_error_handler.py b/tests/bot/cogs/test_error_handler.py
index 5766727e8..fbe0d58d5 100644
--- a/tests/bot/cogs/test_error_handler.py
+++ b/tests/bot/cogs/test_error_handler.py
@@ -4,7 +4,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 from discord.ext.commands import errors
 
 from bot.api import ResponseCodeError
-from bot.cogs.error_handler import ErrorHandler
+from bot.cogs.error_handler import ErrorHandler, setup
 from bot.cogs.moderation.silence import Silence
 from bot.cogs.tags import Tags
 from bot.decorators import InWhitelistCheckFailure
@@ -453,3 +453,13 @@ class OtherErrorHandlerTests(unittest.IsolatedAsyncioTestCase):
         # Await coroutines to avoid warnings
         await result
         await expected
+
+
+class ErrorHandlerSetupTests(unittest.TestCase):
+    """Tests for `ErrorHandler` `setup` function."""
+
+    def test_setup(self):
+        """Should call `bot.add_cog` with `ErrorHandler`."""
+        bot = MockBot()
+        setup(bot)
+        bot.add_cog.assert_called_once()
-- 
cgit v1.2.3