diff options
author | 2022-09-21 21:31:21 +0100 | |
---|---|---|
committer | 2022-09-21 21:31:21 +0100 | |
commit | 75b2b564adbc1abd75877aae588ac37411ebfb64 (patch) | |
tree | 5e96e9ec78bc7f2c20f4a0a6c33ea25bb1c105c6 /tests/helpers.py | |
parent | Display mention & str of the mentionable object in `!remind list`. (diff) | |
parent | Merge #2261 - add support to fetch rules via keywords (diff) |
Merge branch 'main' into bot-2231-enhancements
Diffstat (limited to 'tests/helpers.py')
-rw-r--r-- | tests/helpers.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/tests/helpers.py b/tests/helpers.py index 17214553c..a4b919dcb 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -317,7 +317,7 @@ class MockBot(CustomMockMixin, unittest.mock.MagicMock): guild_id=1, intents=discord.Intents.all(), ) - additional_spec_asyncs = ("wait_for", "redis_ready") + additional_spec_asyncs = ("wait_for",) def __init__(self, **kwargs) -> None: super().__init__(**kwargs) @@ -492,6 +492,28 @@ class MockAttachment(CustomMockMixin, unittest.mock.MagicMock): spec_set = attachment_instance +message_reference_instance = discord.MessageReference( + message_id=unittest.mock.MagicMock(id=1), + channel_id=unittest.mock.MagicMock(id=2), + guild_id=unittest.mock.MagicMock(id=3) +) + + +class MockMessageReference(CustomMockMixin, unittest.mock.MagicMock): + """ + A MagicMock subclass to mock MessageReference objects. + + Instances of this class will follow the specification of `discord.MessageReference` instances. + For more information, see the `MockGuild` docstring. + """ + spec_set = message_reference_instance + + def __init__(self, *, reference_author_is_bot: bool = False, **kwargs): + super().__init__(**kwargs) + referenced_msg_author = MockMember(name="bob", bot=reference_author_is_bot) + self.resolved = MockMessage(author=referenced_msg_author) + + class MockMessage(CustomMockMixin, unittest.mock.MagicMock): """ A MagicMock subclass to mock Message objects. |