From e66237395ab1470002f5dd61de9eeb19ca4600eb Mon Sep 17 00:00:00 2001
From: Sebastiaan Zeeff <33516116+SebastiaanZ@users.noreply.github.com>
Date: Mon, 14 Oct 2019 22:28:51 +0200
Subject: Make test_token_remover use our discord Mocks

This commit replaces the standard MagicMocks by our specialized mocks
for discord.py objects. It also adds the missing `channel` attribute
to the `tests.helpers.MockMessage` mock and moves the file to the
correct folder.
---
 tests/helpers.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'tests/helpers.py')

diff --git a/tests/helpers.py b/tests/helpers.py
index f8e8357f1..892d42e6c 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -415,6 +415,7 @@ class MockMessage(AttributeMock, unittest.mock.MagicMock):
     def __init__(self, **kwargs) -> None:
         super().__init__(spec=message_instance, **kwargs)
         self.author = MockMember()
+        self.channel = MockTextChannel()
 
         # `discord.Message` coroutines
         self.ack = AsyncMock()
-- 
cgit v1.2.3