diff options
| author |  Sebastiaan Zeeff <[email protected]> | 2019-10-14 22:28:51 +0200 |
|---|---|---|
| committer | Sebastiaan Zeeff <[email protected]> | 2019-10-14 22:31:45 +0200 |
| commit | e66237395ab1470002f5dd61de9eeb19ca4600eb (patch) | |
| tree | cb7a72198b25318d5ecea19b996ce1834b8a8675 /tests/helpers.py | |
| parent | Use `MockBot`. (diff) | |
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.
Diffstat (limited to 'tests/helpers.py')
| -rw-r--r-- | tests/helpers.py | 1 |
1 files changed, 1 insertions, 0 deletions
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() |