diff options
| author | 2022-09-18 17:57:20 +0100 | |
|---|---|---|
| committer | 2022-09-18 17:57:20 +0100 | |
| commit | 9a4dbd8768a02f3309c92b87ade747e095fcaca1 (patch) | |
| tree | 2853cdd923f116caf646db9e3b9f738eef6b5a00 /tests/helpers.py | |
| parent | Add note to docstring, fix type-hints, and update log messages (diff) | |
| parent | Merge pull request #2273 from python-discord/fix-ensure-cached-claimant-none-... (diff) | |
Merge branch 'main' into fix-not-awaited-coroutine-warning
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. | 
