def setUpClass(cls):
"""Set up the bot and User cog"""
cls.bot = commands.Bot(command_prefix='!')
cls.userCog = usercog.UserCog(cls.bot)
cls.bot.add_cog(cls.userCog)
cls.bot.add_cog(animecog.AnimeCog(cls.bot))
cls.bot.add_cog(mangacog.MangaCog(cls.bot))
def __init__(self, *args, **kwargs):
"""
Constructor: initialize the bot and add all the cogs.
"""
super().__init__(*args, **kwargs)
self.add_cog(usercog.UserCog(self))
self.add_cog(animecog.AnimeCog(self))
self.add_cog(mangacog.MangaCog(self))
self.add_cog(databasecog.DatabaseCog(self))
def setUpClass(cls):
"""Set up the bot and Manga cog, as well as some premade manga elements"""
cls.bot = commands.Bot(command_prefix='!')
cls.mangaCog = mangacog.MangaCog(cls.bot)
cls.bot.add_cog(cls.mangaCog)
with open('test/test_data/mangarock_1.html') as file:
tree = html.fromstring(file.read())
cls.mangas1 = cls.mangaCog.source._findMangaElements(tree)
with open('test/test_data/mangarock_2.html') as file:
tree = html.fromstring(file.read())
cls.mangas2 = cls.mangaCog.source._findMangaElements(tree)