diff options
| author | 2020-08-17 20:18:14 -0700 | |
|---|---|---|
| committer | 2020-08-17 20:18:26 -0700 | |
| commit | ed30502710e805de5e3793b762a3848a0295582d (patch) | |
| tree | a3e846ca275fe4c70f41972a27a26bf4d70532f4 | |
| parent | Silence tests: assert against message constants (diff) | |
Silence tests: add a subtest for the manual unsilence message
| -rw-r--r-- | tests/bot/cogs/moderation/test_silence.py | 11 | 
1 files changed, 7 insertions, 4 deletions
diff --git a/tests/bot/cogs/moderation/test_silence.py b/tests/bot/cogs/moderation/test_silence.py index 168794b6f..254480a6d 100644 --- a/tests/bot/cogs/moderation/test_silence.py +++ b/tests/bot/cogs/moderation/test_silence.py @@ -280,14 +280,17 @@ class UnsilenceTests(unittest.IsolatedAsyncioTestCase):      async def test_sent_correct_message(self):          """Appropriate failure/success message was sent by the command.""" +        unsilenced_overwrite = PermissionOverwrite(send_messages=True, add_reactions=True)          test_cases = ( -            (True, silence.MSG_UNSILENCE_SUCCESS), -            (False, silence.MSG_UNSILENCE_FAIL) +            (True, silence.MSG_UNSILENCE_SUCCESS, unsilenced_overwrite), +            (False, silence.MSG_UNSILENCE_FAIL, unsilenced_overwrite), +            (False, silence.MSG_UNSILENCE_MANUAL, self.overwrite),          ) -        for was_unsilenced, message in test_cases: +        for was_unsilenced, message, overwrite in test_cases:              ctx = MockContext() -            with self.subTest(was_unsilenced=was_unsilenced, message=message): +            with self.subTest(was_unsilenced=was_unsilenced, message=message, overwrite=overwrite):                  with mock.patch.object(self.cog, "_unsilence", return_value=was_unsilenced): +                    ctx.channel.overwrites_for.return_value = overwrite                      await self.cog.unsilence.callback(self.cog, ctx)                      ctx.channel.send.assert_called_once_with(message)  |