aboutsummaryrefslogtreecommitdiffstats
path: root/tests/helpers.py
diff options
context:
space:
mode:
authorGravatar ks129 <[email protected]>2020-05-18 14:29:07 +0300
committerGravatar ks129 <[email protected]>2020-05-18 14:29:07 +0300
commit28e962bcaa47efca836ad20df5721266336d028e (patch)
treed51a42f637dbf197830db664f8020ba78f0f6ecc /tests/helpers.py
parentError Handler: Changed `CommandNotFound` error check for testing (diff)
Test Helpers: Added new attribute to `MockContext`
Added `invoked_from_error_handler` attribute that is `False` default.
Diffstat (limited to 'tests/helpers.py')
-rw-r--r--tests/helpers.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/helpers.py b/tests/helpers.py
index 2b79a6c2a..0e1581e23 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -368,6 +368,7 @@ message_instance = discord.Message(state=state, channel=channel, data=message_da
# Create a Context instance to get a realistic MagicMock of `discord.ext.commands.Context`
context_instance = Context(message=unittest.mock.MagicMock(), prefix=unittest.mock.MagicMock())
+context_instance.invoked_from_error_handler = None
class MockContext(CustomMockMixin, unittest.mock.MagicMock):
@@ -385,6 +386,7 @@ class MockContext(CustomMockMixin, unittest.mock.MagicMock):
self.guild = kwargs.get('guild', MockGuild())
self.author = kwargs.get('author', MockMember())
self.channel = kwargs.get('channel', MockTextChannel())
+ self.invoked_from_error_handler = kwargs.get('invoked_from_error_handler', False)
attachment_instance = discord.Attachment(data=unittest.mock.MagicMock(id=1), state=unittest.mock.MagicMock())