コード例 #1
0
 async def test_correct_error_when_bad_channel_is_called(self):
     context = AsyncMock(commands.context.Context)
     context.channel = Mock(discord.TextChannel)
     context.channel.id = 4
     context.channel.mention = "test_mention"
     context.guild = Mock(discord.Guild)
     context.guild.id = 5
     context.send = AsyncMock(return_value=Mock(discord.Message))
     with self.assertRaises(commands.BadArgument):
         await self.logger.log(self, context, "Join", "test")
     self.assertFalse(context.send.called)
コード例 #2
0
 async def test_log_command(self):
     context = AsyncMock(commands.context.Context)
     context.channel = Mock(discord.TextChannel)
     context.channel.id = 4
     context.channel.mention = "test_mention"
     context.guild = Mock(discord.Guild)
     context.guild.id = 5
     context.send = AsyncMock(return_value=Mock(discord.Message))
     await self.logger.log(self, context, "Join", "here")
     self.assertTrue(context.send.called)
     self.assertTrue(self.db.set_log_channel.called)