diff options
| author |  Chris Lovering <[email protected]> | 2022-04-02 22:44:53 +0100 |
|---|---|---|
| committer | Chris Lovering <[email protected]> | 2022-04-18 17:45:31 +0100 |
| commit | 56e38eeb38de10611611f0f81f5cbc429d7f2bc8 (patch) | |
| tree | 997a37cb873ee14c998cf4ceb54e9fe075febae7 /tests/helpers.py | |
| parent | Adding missing kwargs required by BotBase in test helper (diff) | |
Update test helpers with breaking d.py changes
region was removed from the guild object, so this has been replaced with features
add_cog is now async, so it is now an async_mock during tests
Two new required voice_channel attrs were added
channel.type is required to be set to ChannelType due to a new isinstance check in d.py
Diffstat (limited to 'tests/helpers.py')
| -rw-r--r-- | tests/helpers.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/helpers.py b/tests/helpers.py index e6e95c20c..2f0c9b4ad 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -325,6 +325,7 @@ class MockBot(CustomMockMixin, unittest.mock.MagicMock): self.api_client = MockAPIClient(loop=self.loop) self.http_session = unittest.mock.create_autospec(spec=ClientSession, spec_set=True) self.stats = unittest.mock.create_autospec(spec=AsyncStatsClient, spec_set=True) + self.add_cog = unittest.mock.AsyncMock() # Create a TextChannel instance to get a realistic MagicMock of `discord.TextChannel` @@ -337,6 +338,8 @@ channel_data = { 'position': 1, 'nsfw': False, 'last_message_id': 1, + 'bitrate': 1337, + 'user_limit': 25, } state = unittest.mock.MagicMock() guild = unittest.mock.MagicMock() @@ -441,6 +444,7 @@ message_data = { } state = unittest.mock.MagicMock() channel = unittest.mock.MagicMock() +channel.type = discord.ChannelType.text message_instance = discord.Message(state=state, channel=channel, data=message_data) |