Python setup_cog Exemples

Exemple #1

Fichier : test_token_remover.py Projet : vemel/bot

def test_setup(caplog):
    bot = MagicMock()
    setup_cog(bot)
    [record] = caplog.records

    bot.add_cog.assert_called_once()
    assert record.message == "Cog loaded: TokenRemover"

Exemple #2

Fichier : test_token_remover.py Projet : sapanz/bot

    def test_setup(self):
        """Setup of the cog should log a message at `INFO` level."""
        bot = MockBot()
        with self.assertLogs(logger='bot.cogs.token_remover',
                             level=logging.INFO) as cm:
            setup_cog(bot)

        [line] = cm.output
        bot.add_cog.assert_called_once()
        self.assertIn("Cog loaded: TokenRemover", line)

Exemple #3

Fichier : test_token_remover.py Projet : Aurelius41/song

 def test_setup(self):
     """Setup of the extension should call add_cog."""
     bot = MockBot()
     setup_cog(bot)
     bot.add_cog.assert_called_once()