mirror of
https://github.com/tgbot-collection/YYeTsBot.git
synced 2025-11-25 11:29:38 +08:00
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
# coding: utf-8
|
|
|
|
import unittest
|
|
import os
|
|
import sys
|
|
|
|
import requests_mock
|
|
from unittest import mock
|
|
|
|
sys.path.append("../yyetsbot")
|
|
|
|
from fansub import BaseFansub, YYeTs
|
|
|
|
|
|
class TestBaseFunsub(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.ins = BaseFansub()
|
|
cls.cookie_jar = dict(name="hello")
|
|
cls.ins.cookie_file = "test_cookies.dump" # generate on tests/test_cookies.dump
|
|
|
|
@classmethod
|
|
def tearDownClass(cls) -> None:
|
|
cls().ins.redis.flushall()
|
|
os.unlink(cls().ins.cookie_file)
|
|
|
|
def test_save_cookies(self):
|
|
self.ins.__save_cookies__(self.cookie_jar)
|
|
exists = os.path.exists(self.ins.cookie_file)
|
|
self.assertTrue(exists)
|
|
|
|
def test_load_cookies(self):
|
|
self.test_save_cookies()
|
|
cookie = self.ins.__load_cookies__()
|
|
self.assertEqual(cookie, self.cookie_jar)
|
|
|
|
def test_get_from_cache(self):
|
|
value = self.ins.__get_from_cache__("http://test.url", "__hash__")
|
|
self.assertEqual(value, self.ins.__hash__())
|
|
|
|
def test_save_to_cache(self):
|
|
# never expire
|
|
url = "http://test2.url"
|
|
self.ins.__save_to_cache__(url, self.cookie_jar)
|
|
cache_copy = self.ins.__get_from_cache__(url, "never mind method")
|
|
self.assertEqual(cache_copy, self.cookie_jar)
|
|
|
|
|
|
class YYeTsTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.ins = YYeTs()
|
|
cls.cookie_jar = dict(name="hello yyets")
|
|
cls.ins.cookie_file = "test_cookies.dump" # generate on tests/test_cookies.dump
|
|
cls.ins.url = "http://www.rrys2020.com/resource/1988"
|
|
|
|
@classmethod
|
|
def tearDownClass(cls) -> None:
|
|
cls().ins.redis.flushall()
|
|
# os.unlink(cls().ins.cookie_file)
|
|
|
|
def test_get_id(self):
|
|
self.assertEqual(self.ins.id, "1988")
|
|
|
|
@requests_mock.mock()
|
|
def test_get_search_html(self, m):
|
|
with open("yyets_search.html") as f:
|
|
html = f.read()
|
|
m.get('http://www.rrys2020.com/search?keyword=abc&type=resource', text=html)
|
|
response = self.ins.__get_search_html__("abc")
|
|
self.assertEqual(html, response)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|