diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/bot/exts/info/test_information.py | 20 | ||||
| -rw-r--r-- | tests/bot/exts/moderation/infraction/test_utils.py | 12 | ||||
| -rw-r--r-- | tests/bot/utils/test_services.py | 4 | 
3 files changed, 18 insertions, 18 deletions
| diff --git a/tests/bot/exts/info/test_information.py b/tests/bot/exts/info/test_information.py index 80731c9f0..a996ce477 100644 --- a/tests/bot/exts/info/test_information.py +++ b/tests/bot/exts/info/test_information.py @@ -283,6 +283,7 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          user = helpers.MockMember()          user.nick = None          user.__str__ = unittest.mock.Mock(return_value="Mr. Hemlock") +        user.colour = 0          embed = await self.cog.create_user_embed(ctx, user) @@ -298,6 +299,7 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          user = helpers.MockMember()          user.nick = "Cat lover"          user.__str__ = unittest.mock.Mock(return_value="Mr. Hemlock") +        user.colour = 0          embed = await self.cog.create_user_embed(ctx, user) @@ -311,10 +313,9 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          """Created `!user` embeds should not contain mention of the @everyone-role."""          ctx = helpers.MockContext(channel=helpers.MockTextChannel(id=1))          admins_role = helpers.MockRole(name='Admins') -        admins_role.colour = 100          # A `MockMember` has the @Everyone role by default; we add the Admins to that. -        user = helpers.MockMember(roles=[admins_role], top_role=admins_role) +        user = helpers.MockMember(roles=[admins_role], colour=100)          embed = await self.cog.create_user_embed(ctx, user) @@ -332,12 +333,11 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          ctx = helpers.MockContext(channel=helpers.MockTextChannel(id=50))          moderators_role = helpers.MockRole(name='Moderators') -        moderators_role.colour = 100          infraction_counts.return_value = ("Infractions", "expanded infractions info")          nomination_counts.return_value = ("Nominations", "nomination info") -        user = helpers.MockMember(id=314, roles=[moderators_role], top_role=moderators_role) +        user = helpers.MockMember(id=314, roles=[moderators_role], colour=100)          embed = await self.cog.create_user_embed(ctx, user)          infraction_counts.assert_called_once_with(user) @@ -367,11 +367,10 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          ctx = helpers.MockContext(channel=helpers.MockTextChannel(id=100))          moderators_role = helpers.MockRole(name='Moderators') -        moderators_role.colour = 100          infraction_counts.return_value = ("Infractions", "basic infractions info") -        user = helpers.MockMember(id=314, roles=[moderators_role], top_role=moderators_role) +        user = helpers.MockMember(id=314, roles=[moderators_role], colour=100)          embed = await self.cog.create_user_embed(ctx, user)          infraction_counts.assert_called_once_with(user) @@ -407,12 +406,11 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          ctx = helpers.MockContext()          moderators_role = helpers.MockRole(name='Moderators') -        moderators_role.colour = 100 -        user = helpers.MockMember(id=314, roles=[moderators_role], top_role=moderators_role) +        user = helpers.MockMember(id=314, roles=[moderators_role], colour=100)          embed = await self.cog.create_user_embed(ctx, user) -        self.assertEqual(embed.colour, discord.Colour(moderators_role.colour)) +        self.assertEqual(embed.colour, discord.Colour(100))      @unittest.mock.patch(          f"{COG_PATH}.basic_user_infraction_counts", @@ -422,7 +420,7 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          """The embed should be created with a blurple colour if the user has no assigned roles."""          ctx = helpers.MockContext() -        user = helpers.MockMember(id=217) +        user = helpers.MockMember(id=217, colour=discord.Colour.default())          embed = await self.cog.create_user_embed(ctx, user)          self.assertEqual(embed.colour, discord.Colour.blurple()) @@ -435,7 +433,7 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):          """The embed thumbnail should be set to the user's avatar in `png` format."""          ctx = helpers.MockContext() -        user = helpers.MockMember(id=217) +        user = helpers.MockMember(id=217, colour=0)          user.avatar_url_as.return_value = "avatar url"          embed = await self.cog.create_user_embed(ctx, user) diff --git a/tests/bot/exts/moderation/infraction/test_utils.py b/tests/bot/exts/moderation/infraction/test_utils.py index 5b62463e0..ee9ff650c 100644 --- a/tests/bot/exts/moderation/infraction/test_utils.py +++ b/tests/bot/exts/moderation/infraction/test_utils.py @@ -146,7 +146,7 @@ class ModerationUtilsTests(unittest.IsolatedAsyncioTestCase):                      name=utils.INFRACTION_AUTHOR_NAME,                      url=utils.RULES_URL,                      icon_url=Icons.token_removed -                ).set_footer(text=utils.INFRACTION_APPEAL_FOOTER), +                ).set_footer(text=utils.INFRACTION_APPEAL_MODMAIL_FOOTER),                  "send_result": True              },              { @@ -164,9 +164,11 @@ class ModerationUtilsTests(unittest.IsolatedAsyncioTestCase):                      name=utils.INFRACTION_AUTHOR_NAME,                      url=utils.RULES_URL,                      icon_url=Icons.token_removed -                ), +                ).set_footer(text=utils.INFRACTION_APPEAL_MODMAIL_FOOTER),                  "send_result": False              }, +            # Note that this test case asserts that the DM that *would* get sent to the user is formatted +            # correctly, even though that message is deliberately never sent.              {                  "args": (self.user, "note", None, None, Icons.defcon_denied),                  "expected_output": Embed( @@ -182,7 +184,7 @@ class ModerationUtilsTests(unittest.IsolatedAsyncioTestCase):                      name=utils.INFRACTION_AUTHOR_NAME,                      url=utils.RULES_URL,                      icon_url=Icons.defcon_denied -                ), +                ).set_footer(text=utils.INFRACTION_APPEAL_MODMAIL_FOOTER),                  "send_result": False              },              { @@ -200,7 +202,7 @@ class ModerationUtilsTests(unittest.IsolatedAsyncioTestCase):                      name=utils.INFRACTION_AUTHOR_NAME,                      url=utils.RULES_URL,                      icon_url=Icons.defcon_denied -                ).set_footer(text=utils.INFRACTION_APPEAL_FOOTER), +                ).set_footer(text=utils.INFRACTION_APPEAL_MODMAIL_FOOTER),                  "send_result": False              },              { @@ -218,7 +220,7 @@ class ModerationUtilsTests(unittest.IsolatedAsyncioTestCase):                      name=utils.INFRACTION_AUTHOR_NAME,                      url=utils.RULES_URL,                      icon_url=Icons.defcon_denied -                ).set_footer(text=utils.INFRACTION_APPEAL_FOOTER), +                ).set_footer(text=utils.INFRACTION_APPEAL_MODMAIL_FOOTER),                  "send_result": True              }          ] diff --git a/tests/bot/utils/test_services.py b/tests/bot/utils/test_services.py index 1b48f6560..3b71022db 100644 --- a/tests/bot/utils/test_services.py +++ b/tests/bot/utils/test_services.py @@ -30,9 +30,9 @@ class PasteTests(unittest.IsolatedAsyncioTestCase):          """Url with specified extension is returned on successful requests."""          key = "paste_key"          test_cases = ( -            (f"https://paste_service.com/{key}.txt", "txt"), +            (f"https://paste_service.com/{key}.txt?noredirect", "txt"),              (f"https://paste_service.com/{key}.py", "py"), -            (f"https://paste_service.com/{key}", ""), +            (f"https://paste_service.com/{key}?noredirect", ""),          )          response = MagicMock(              json=AsyncMock(return_value={"key": key}) | 
