aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar ks129 <[email protected]>2020-06-13 08:26:19 +0300
committerGravatar GitHub <[email protected]>2020-06-13 08:26:19 +0300
commit9dbfe7da4cbc4d1820507e25ce56929b7fb55327 (patch)
treea311ea72ea9987786ef475974405f5852ff72c9a
parentJams: Use `Guild` instead `Context` for helper functions (diff)
Jam Tests: Update `Context` to `Guild` for tests too
-rw-r--r--tests/bot/cogs/test_jams.py38
1 files changed, 19 insertions, 19 deletions
diff --git a/tests/bot/cogs/test_jams.py b/tests/bot/cogs/test_jams.py
index 54fe0b5f2..17b86601f 100644
--- a/tests/bot/cogs/test_jams.py
+++ b/tests/bot/cogs/test_jams.py
@@ -47,23 +47,23 @@ class JamCreateTeamTests(unittest.IsolatedAsyncioTestCase):
async def test_category_dont_exist(self):
"""Should create code jam category."""
self.utils_mock.get.return_value = None
- await self.cog.get_category(self.ctx)
- self.ctx.guild.create_category_channel.assert_awaited_once()
- category_overwrites = self.ctx.guild.create_category_channel.call_args[1]["overwrites"]
+ await self.cog.get_category(self.guild)
+ self.guild.create_category_channel.assert_awaited_once()
+ category_overwrites = self.guild.create_category_channel.call_args[1]["overwrites"]
- self.assertFalse(category_overwrites[self.ctx.guild.default_role].read_messages)
- self.assertTrue(category_overwrites[self.ctx.guild.me].read_messages)
+ self.assertFalse(category_overwrites[self.guild.default_role].read_messages)
+ self.assertTrue(category_overwrites[self.guild.me].read_messages)
async def test_category_channel_exist(self):
"""Should not try to create category channel."""
- await self.cog.get_category(self.ctx)
- self.ctx.guild.create_category_channel.assert_not_awaited()
+ await self.cog.get_category(self.guild)
+ self.guild.create_category_channel.assert_not_awaited()
async def test_channel_overwrites(self):
"""Should have correct permission overwrites for users and roles."""
leader = MockMember()
members = [leader] + [MockMember() for _ in range(4)]
- overwrites = self.cog.get_overwrites(members, self.ctx)
+ overwrites = self.cog.get_overwrites(members, self.guild)
# Leader permission overwrites
self.assertTrue(overwrites[leader].manage_messages)
@@ -77,10 +77,10 @@ class JamCreateTeamTests(unittest.IsolatedAsyncioTestCase):
self.assertTrue(overwrites[member].connect)
# Everyone and verified role overwrite
- self.assertFalse(overwrites[self.ctx.guild.default_role].read_messages)
- self.assertFalse(overwrites[self.ctx.guild.default_role].connect)
- self.assertFalse(overwrites[self.ctx.guild.get_role(Roles.verified)].read_messages)
- self.assertFalse(overwrites[self.ctx.guild.get_role(Roles.verified)].connect)
+ self.assertFalse(overwrites[self.guild.default_role].read_messages)
+ self.assertFalse(overwrites[self.guild.default_role].connect)
+ self.assertFalse(overwrites[self.guild.get_role(Roles.verified)].read_messages)
+ self.assertFalse(overwrites[self.guild.get_role(Roles.verified)].connect)
async def test_team_channels_creation(self):
"""Should create new voice and text channel for team."""
@@ -90,18 +90,18 @@ class JamCreateTeamTests(unittest.IsolatedAsyncioTestCase):
self.cog.get_overwrites = MagicMock()
self.cog.get_category = AsyncMock()
self.ctx.guild.create_text_channel.return_value = MockTextChannel(mention="foobar-channel")
- actual = await self.cog.create_channels(self.ctx, "my-team", members)
+ actual = await self.cog.create_channels(self.guild, "my-team", members)
self.assertEqual("foobar-channel", actual)
- self.cog.get_overwrites.assert_called_once_with(members, self.ctx)
- self.cog.get_category.assert_awaited_once_with(self.ctx)
+ self.cog.get_overwrites.assert_called_once_with(members, self.guild)
+ self.cog.get_category.assert_awaited_once_with(self.guild)
- self.ctx.guild.create_text_channel.assert_awaited_once_with(
+ self.guild.create_text_channel.assert_awaited_once_with(
"my-team",
overwrites=self.cog.get_overwrites.return_value,
category=self.cog.get_category.return_value
)
- self.ctx.guild.create_voice_channel.assert_awaited_once_with(
+ self.guild.create_voice_channel.assert_awaited_once_with(
"My Team",
overwrites=self.cog.get_overwrites.return_value,
category=self.cog.get_category.return_value
@@ -111,11 +111,11 @@ class JamCreateTeamTests(unittest.IsolatedAsyncioTestCase):
"""Should add team leader role to leader and jam role to every team member."""
leader_role = MockRole(name="Team Leader")
jam_role = MockRole(name="Jammer")
- self.ctx.guild.get_role.side_effect = [leader_role, jam_role]
+ self.guild.get_role.side_effect = [leader_role, jam_role]
leader = MockMember()
members = [leader] + [MockMember() for _ in range(4)]
- await self.cog.add_roles(self.ctx, members)
+ await self.cog.add_roles(self.guild, members)
leader.add_roles.assert_any_await(leader_role)
for member in members: