mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-13 00:59:01 +08:00
116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from app.modules.indexer.parser.hddolby import HDDolbySiteUserInfo
|
|
from app.modules.indexer.parser.nexus_php import NexusPhpSiteUserInfo
|
|
|
|
|
|
USER_DATA_JSON = """
|
|
{
|
|
"status": 0,
|
|
"data": [{
|
|
"id": "41088",
|
|
"username": "g0m3e",
|
|
"added": "2023-12-18 13:58:19",
|
|
"class": "5",
|
|
"uploaded": "1000",
|
|
"downloaded": "500",
|
|
"seedbonus": "100.0",
|
|
"unread_messages": "1"
|
|
}]
|
|
}
|
|
"""
|
|
|
|
|
|
def _build_parser(site_cookie: str = "") -> HDDolbySiteUserInfo:
|
|
return HDDolbySiteUserInfo(
|
|
site_name="高清杜比",
|
|
url="https://www.hddolby.com/",
|
|
site_cookie=site_cookie,
|
|
apikey="test-api-key",
|
|
token=None,
|
|
)
|
|
|
|
|
|
def test_hddolby_parse_user_base_info_reads_unread_count():
|
|
parser = _build_parser()
|
|
parser._parse_user_base_info(USER_DATA_JSON)
|
|
|
|
assert parser.userid == "41088"
|
|
assert parser.username == "g0m3e"
|
|
assert parser.message_unread == 1
|
|
|
|
|
|
def test_hddolby_skips_message_body_without_cookie():
|
|
parser = _build_parser(site_cookie="")
|
|
parser.message_unread = 1
|
|
|
|
parser._pase_unread_msgs()
|
|
|
|
assert parser.message_unread_contents == []
|
|
|
|
|
|
def test_hddolby_reads_message_body_via_nexus_when_cookie_present(monkeypatch):
|
|
parser = _build_parser(site_cookie="c_secure_uid=1")
|
|
parser.message_unread = 1
|
|
|
|
def fake_pase_unread_msgs(self):
|
|
self.message_unread_contents = [
|
|
("测试标题", "2026-07-21 10:00:00", "测试正文"),
|
|
]
|
|
|
|
monkeypatch.setattr(NexusPhpSiteUserInfo, "_pase_unread_msgs", fake_pase_unread_msgs)
|
|
|
|
parser._pase_unread_msgs()
|
|
|
|
assert len(parser.message_unread_contents) == 1
|
|
head, date, content = parser.message_unread_contents[0]
|
|
assert head == "测试标题"
|
|
assert date == "2026-07-21 10:00:00"
|
|
assert content == "测试正文"
|
|
|
|
|
|
def test_parser_skips_detail_and_seeding_requests_without_valid_userid(monkeypatch):
|
|
class DummyParser(HDDolbySiteUserInfo):
|
|
def _parse_site_page(self, html_text: str):
|
|
self._user_detail_page = "userdetails.php?id="
|
|
self._user_basic_page = None
|
|
self._user_traffic_page = None
|
|
self._torrent_seeding_page = "getusertorrentlistajax.php?userid="
|
|
|
|
def _parse_user_base_info(self, html_text: str):
|
|
self.userid = None
|
|
|
|
def _parse_user_traffic_info(self, html_text: str):
|
|
raise AssertionError("should not request traffic page without valid userid")
|
|
|
|
def _parse_user_torrent_seeding_info(self, html_text: str, multi_page: bool = False):
|
|
raise AssertionError("should not request seeding list without valid userid")
|
|
|
|
def _parse_logged_in(self, html_text):
|
|
return True
|
|
|
|
def _parse_message_unread_links(self, html_text: str, msg_links: list):
|
|
return None
|
|
|
|
def _parse_message_content(self, html_text):
|
|
return None, None, None
|
|
|
|
parser = DummyParser(
|
|
site_name="高清杜比",
|
|
url="https://www.hddolby.com/",
|
|
site_cookie="",
|
|
apikey="test-api-key",
|
|
token=None,
|
|
)
|
|
calls = []
|
|
|
|
def fake_get_page_content(**kwargs):
|
|
calls.append(kwargs.get("url", ""))
|
|
return ""
|
|
|
|
monkeypatch.setattr(parser, "_get_page_content", fake_get_page_content)
|
|
|
|
parser.parse()
|
|
|
|
assert not any("userdetails.php?id=" in call for call in calls)
|
|
assert not any("getusertorrentlistajax.php?userid=" in call for call in calls)
|