diff options
| author |  wookie184 <[email protected]> | 2024-05-23 16:59:33 +0100 |
|---|---|---|
| committer |  GitHub <[email protected]> | 2024-05-23 16:59:33 +0100 |
| commit | 09cd19394dbfcb135261f6c49278e0e4fc4fe3a7 (patch) | |
| tree | 98a4a5cf1b02886783179282a041d74f504c7da4 /tests/helpers.py | |
| parent | Adjust truncation for 11 lines (diff) | |
| parent | Support events that span from one year to another (#3065) (diff) | |
Merge branch 'main' into vivek/fix-eval-truncation
Diffstat (limited to 'tests/helpers.py')
| -rw-r--r-- | tests/helpers.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/helpers.py b/tests/helpers.py index 580848c25..c51a82a9d 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -500,10 +500,12 @@ class MockContext(CustomMockMixin, unittest.mock.MagicMock): super().__init__(**kwargs) self.me = kwargs.get("me", MockMember()) self.bot = kwargs.get("bot", MockBot()) - self.guild = kwargs.get("guild", MockGuild()) - self.author = kwargs.get("author", MockMember()) - self.channel = kwargs.get("channel", MockTextChannel()) - self.message = kwargs.get("message", MockMessage()) + + self.message = kwargs.get("message", MockMessage(guild=self.guild)) + self.author = kwargs.get("author", self.message.author) + self.channel = kwargs.get("channel", self.message.channel) + self.guild = kwargs.get("guild", self.channel.guild) + self.invoked_from_error_handler = kwargs.get("invoked_from_error_handler", False) @@ -519,10 +521,12 @@ class MockInteraction(CustomMockMixin, unittest.mock.MagicMock): super().__init__(**kwargs) self.me = kwargs.get("me", MockMember()) self.client = kwargs.get("client", MockBot()) - self.guild = kwargs.get("guild", MockGuild()) - self.user = kwargs.get("user", MockMember()) - self.channel = kwargs.get("channel", MockTextChannel()) - self.message = kwargs.get("message", MockMessage()) + + self.message = kwargs.get("message", MockMessage(guild=self.guild)) + self.user = kwargs.get("user", self.message.author) + self.channel = kwargs.get("channel", self.message.channel) + self.guild = kwargs.get("guild", self.channel.guild) + self.invoked_from_error_handler = kwargs.get("invoked_from_error_handler", False) |