async def setMangaSource(self, ctx, source: str):
"""
Set the manga source, i.e. where it will retrieve the manga from.
:param ctx: The context.
:param source: Name of the manga source.
"""
if source.lower() == "mangarock":
await ctx.send(
'Unfortunately MangaRock has removed all their reading features. Please use a different source.'
)
return
elif source.lower() == "mangadex":
self.source = MangaDex()
elif source.lower() == "none":
self.source = None
self.list.clear()
await ctx.send(f'Successfully removed the manga source.')
return
else:
await ctx.send('Unknown or unsupported manga source.')
return
self.list.clear()
self.fillCache()
await ctx.send(f'Successfully set the manga source to {source}.')
def __init__(self, bot):
"""
Constructor: initialize the cog.
:param bot: The Discord bot.
"""
super().__init__(bot)
# Replace this with your own 'Manga API' if you want to use a different manga source
self.source = MangaDex()
def _setMangaSource(self, source):
"""
Set the manga source, i.e. where it will retrieve the manga from.
:param source: Name of the manga source.
:return True if successful, False otherwise.
"""
if source.lower() == "mangadex":
self.source = MangaDex()
elif source.lower() == "none":
self.source = None
else:
return False
return True
def setUp(self):
self.mangadex = MangaDex()
class TestMangaDex(unittest.TestCase):
"""Tests for the MangaDex"""
def setUp(self):
self.mangadex = MangaDex()
def test_findMangaElements(self):
"""Test getting the manga html elements from a given MangaDex html file"""
with open('test/main/test_data/mangadex.html') as file:
tree = html.fromstring(file.read())
mangas = self.mangadex._findMangaElements(tree)
self.assertEqual(len(mangas), 42)
titles = []
for manga in mangas:
title = manga.xpath(
".//a[contains(concat(' ', normalize-space(@class), ' '), ' manga_title ')]"
)[0]
titles.append(title.text_content())
self.assertEqual(titles[:10], [
'Hana wa Junai ni Junjiru', 'trash.', 'Paradise Baby',
'Kumo Desu ga, Nani ka? Daily Life of the Four Spider Sisters',
'Kannou Shousetsuka no Neko', 'Persona 5 Mementos Mission',
'Metropolitan System', 'Ouji Hiroimashita', 'Honoo no Suna',
'Enslave Lover'
])
def test_getRecentManga(self):
"""Test getting the most recent manga from the MangaDex homepage"""
with open('test/main/test_data/mangadex.html') as file:
tree = html.fromstring(file.read())
mangas = self.mangadex._findMangaElements(tree)
self.mangadex._findMangaElements = MagicMock(return_value=mangas)
recentManga = self.mangadex.getRecent()
self.assertEqual(len(recentManga), 16)
self.assertEqual([manga.title for manga in recentManga], [
'Hana wa Junai ni Junjiru', 'trash.', 'Paradise Baby',
'Kumo Desu ga, Nani ka? Daily Life of the Four Spider Sisters',
'Kannou Shousetsuka no Neko', 'Persona 5 Mementos Mission',
'Metropolitan System', 'Ouji Hiroimashita', 'Honoo no Suna',
'Enslave Lover', '2D Partner', 'Nessa no Kusari',
'Bliss~End Of Gods', 'Shikanoko Nokonoko Koshitantan',
'Koi no Myouyaku', 'Sawatte, Tokashite'
])
self.assertEqual([manga.progress for manga in recentManga], [
'Vol. 1 Chapter 9.5', 'Vol. 7 Chapter 51', 'Vol. 1 Chapter 7.5',
'Chapter 46', 'Vol. 1 Chapter 2', 'Vol. 1 Chapter 4.5',
'Chapter 323', 'Vol. 1 Chapter 7.5', 'Vol. 1 Chapter 6.5',
'Vol. 1 Chapter 7.5', 'Chapter 16', 'Vol. 1 Chapter 6.6',
'Chapter 2', 'Chapter 2', 'Vol. 1 Chapter 4', 'Vol. 1 Chapter 6.5'
])
self.assertEqual([manga.link for manga in recentManga], [
'https://mangadex.org/chapter/1002666',
'https://mangadex.org/chapter/1002657',
'https://mangadex.org/chapter/1002656',
'https://mangadex.org/chapter/1002650',
'https://mangadex.org/chapter/1002644',
'https://mangadex.org/chapter/1002636',
'https://mangadex.org/chapter/1002635',
'https://mangadex.org/chapter/1002634',
'https://mangadex.org/chapter/1002624',
'https://mangadex.org/chapter/1002612',
'https://mangadex.org/chapter/1002608',
'https://mangadex.org/chapter/1002601',
'https://mangadex.org/chapter/1002586',
'https://mangadex.org/chapter/1002583',
'https://mangadex.org/chapter/1002576',
'https://mangadex.org/chapter/1002572'
])
def test_mangadex(self):
"""Test MangaDex api"""
mangadex = MangaDex()
mangas = mangadex.getRecent()
self.assertTrue(len(mangas) > 0)