import unittest import os import urllib3.exceptions from nhentai import constant from nhentai.cmdline import load_config from nhentai.parser import search_parser, doujinshi_parser, favorites_parser class TestParser(unittest.TestCase): def setUp(self) -> None: urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) load_config() constant.CONFIG['cookie'] = os.getenv('NHENTAI_COOKIE') constant.CONFIG['useragent'] = os.getenv('NHENTAI_UA') def test_search(self): result = search_parser('umaru', 'recent', [1], False) self.assertTrue(len(result) > 0) def test_doujinshi_parser(self): result = doujinshi_parser(123456) self.assertTrue(result['pages'] == 84) def test_favorites_parser(self): result = favorites_parser(page=[1]) self.assertTrue(len(result) > 0)