From 0f074d53eb50bddae69f775c1bb0a45b89774e12 Mon Sep 17 00:00:00 2001 From: kosayoda Date: Mon, 21 Sep 2020 22:27:51 +0800 Subject: Add more role information to server embed. --- tests/bot/exts/info/test_information.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) (limited to 'tests') diff --git a/tests/bot/exts/info/test_information.py b/tests/bot/exts/info/test_information.py index 261533847..b5c71f87c 100644 --- a/tests/bot/exts/info/test_information.py +++ b/tests/bot/exts/info/test_information.py @@ -132,6 +132,15 @@ class InformationCogTests(unittest.TestCase): icon_url='a-lemon.jpg', ) + self.ctx.guild.get_role = unittest.mock.Mock() + self.ctx.guild.get_role.side_effect = lambda id: { + constants.Roles.helpers: helpers.MockRole(name="Helpers", id=id, members=[]), + constants.Roles.moderators: helpers.MockRole(name="Moderators", id=id, members=[]), + constants.Roles.admins: helpers.MockRole(name="Admins", id=id, members=[]), + constants.Roles.owners: helpers.MockRole(name="Owners", id=id, members=[]), + constants.Roles.contributors: helpers.MockRole(name="Contributors", id=id, members=[]), + }[id] + coroutine = self.cog.server_info.callback(self.cog, self.ctx) self.assertIsNone(asyncio.run(coroutine)) @@ -157,9 +166,12 @@ class InformationCogTests(unittest.TestCase): self.assertEqual( member_field.value, textwrap.dedent(f""" - Staff members: 0 - Roles: {len(self.ctx.guild.roles)} - """), + Helpers: 0 + Moderators: 0 + Admins: 0 + Contributors: 0 + Roles: {len(self.ctx.guild.roles) - 1} + """).strip(), ) # Channels -- cgit v1.2.3