diff options
| author |  ks129 <[email protected]> | 2020-05-23 18:37:27 +0300 |
|---|---|---|
| committer | ks129 <[email protected]> | 2020-06-11 08:59:42 +0300 |
| commit | b5b05adc41e55dd58810608f4ac7ade6281cdf84 (patch) | |
| tree | 7456640faf1fe8d18d848fc93b5195859deff984 | |
| parent | Jam Tests: Create test for team voice channel creating (diff) | |
Jam Tests: Create test for team jam roles adding
| -rw-r--r-- | tests/bot/cogs/test_jams.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/bot/cogs/test_jams.py b/tests/bot/cogs/test_jams.py index 2e1419f8e..16caa98c6 100644 --- a/tests/bot/cogs/test_jams.py +++ b/tests/bot/cogs/test_jams.py @@ -106,3 +106,17 @@ class JamCreateTeamTests(unittest.IsolatedAsyncioTestCase): # All other arguments is possible to get somewhere else except this overwrites = self.ctx.guild.create_voice_channel.call_args[1]["overwrites"] self.ctx.guild.create_voice_channel.assert_awaited_once_with("My Team", overwrites=overwrites, category="foo") + + async def test_jam_roles_adding(self): + """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 = [MockRole(), leader_role, jam_role] + + leader = MockMember() + members = [leader] + [MockMember() for _ in range(4)] + await self.cog.createteam(self.cog, self.ctx, "foo", members) + + leader.add_roles.assert_any_await(leader_role) + for member in members: + member.add_roles.assert_any_await(jam_role) |