diff options
author | 2020-02-01 15:32:16 -0800 | |
---|---|---|
committer | 2020-02-12 10:07:56 -0800 | |
commit | 535095ff647277922b7d1930da8d038f15af74fd (patch) | |
tree | 3bd8491da3320abc607ddee0d9b4d900b249896c | |
parent | Sync tests: test Sync cog's on_guild_role_delete listener (diff) |
Tests: use objects for colour and permissions of MockRole
Instances of discord.Colour and discord.Permissions will be created
by default or when ints are given as values for those attributes.
-rw-r--r-- | tests/helpers.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/helpers.py b/tests/helpers.py index b18a27ebe..a40673bb9 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -270,9 +270,21 @@ class MockRole(CustomMockMixin, unittest.mock.Mock, ColourMixin, HashableMixin): information, see the `MockGuild` docstring. """ def __init__(self, **kwargs) -> None: - default_kwargs = {'id': next(self.discord_id), 'name': 'role', 'position': 1} + default_kwargs = { + 'id': next(self.discord_id), + 'name': 'role', + 'position': 1, + 'colour': discord.Colour(0xdeadbf), + 'permissions': discord.Permissions(), + } super().__init__(spec_set=role_instance, **collections.ChainMap(kwargs, default_kwargs)) + if isinstance(self.colour, int): + self.colour = discord.Colour(self.colour) + + if isinstance(self.permissions, int): + self.permissions = discord.Permissions(self.permissions) + if 'mention' not in kwargs: self.mention = f'&{self.name}' |