mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 11:27:43 +08:00
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
from collections.abc import Iterator
|
||
|
||
import pytest
|
||
|
||
from app.modules.indexer.parser.discuz import DiscuzUserInfo
|
||
|
||
|
||
@pytest.fixture
|
||
def parser() -> Iterator[DiscuzUserInfo]:
|
||
instance = DiscuzUserInfo(
|
||
site_name="Discuz",
|
||
url="https://example.com/",
|
||
site_cookie="",
|
||
apikey=None,
|
||
token=None,
|
||
)
|
||
yield instance
|
||
instance.close()
|
||
|
||
|
||
def test_discuz_traffic_defaults_use_integer_bytes(parser: DiscuzUserInfo) -> None:
|
||
"""上传/下载量使用整数字节，分享率保持浮点数。"""
|
||
for value in (parser.upload, parser.download):
|
||
assert value == 0
|
||
assert type(value) is int
|
||
assert parser.ratio == 0.0
|
||
assert isinstance(parser.ratio, float)
|
||
|
||
|
||
@pytest.mark.parametrize("include_menu", [True, False])
|
||
def test_discuz_skyeysnow_ratio(parser: DiscuzUserInfo, include_menu: bool) -> None:
|
||
"""天雪菜单和资料页的无括号分享率均应解析，且不影响积分和流量。"""
|
||
menu = """
|
||
<ul id="ratio_menu">
|
||
<li><a>上传量 : 845.25 GB</a></li>
|
||
<li><a>分享率 : 24.12</a></li>
|
||
<li><a>保种率 : 31.91</a></li>
|
||
</ul>
|
||
""" if include_menu else ""
|
||
html_text = f"""
|
||
<html><body>
|
||
{menu}
|
||
<ul class="cl bbda pbm mbm">
|
||
<li>
|
||
<em class="xg2">统计信息</em>
|
||
<a>好友数 0</a><span class="pipe">|</span><a>回帖数 1033</a>
|
||
<br /><em class="xg2">PT信息</em>
|
||
保种率 32.09 <span class="pipe">|</span> 分享率 24.12
|
||
<span class="pipe">|</span> 预期保种率 31.91
|
||
</li>
|
||
</ul>
|
||
<ul class="pf_l">
|
||
<li><em>积分</em>58065</li>
|
||
<li><em>上传积分/上传量</em>343.62 G / 845.25 GB</li>
|
||
<li><em>下载积分/下载量</em>35.04 G / 35.04 GB</li>
|
||
</ul>
|
||
</body></html>
|
||
"""
|
||
|
||
parser._parse_user_detail_info(html_text)
|
||
|
||
assert parser.ratio == 24.12
|
||
assert parser.to_dict()["ratio"] == 24.12
|
||
assert parser.bonus == 58065
|
||
assert parser.upload == round(845.25 * 1024 ** 3)
|
||
assert parser.download == round(35.04 * 1024 ** 3)
|
||
|
||
|
||
@pytest.mark.parametrize(("ratio_html", "expected"), [
|
||
("<li>分享率 (24.12)</li>", 24.12),
|
||
("<li><em>分享率</em>24.12</li>", 24.12),
|
||
("<li>\n<em>分享率</em>：<span>24.12</span></li>", 24.12),
|
||
("<li><em>分享率</em>（1,234.56）</li>", 1234.56),
|
||
("<li><a>分享率 : 0.00</a></li>", 0),
|
||
("<li>分享率 (24.12)</li><li>分享率 (31.91)</li>", 24.12),
|
||
("<li>保种率 31.91</li>", 0),
|
||
("<li>分享率 -- <span>保种率 (31.91)</span></li>", 0),
|
||
])
|
||
def test_discuz_ratio_formats_keep_bonus(parser: DiscuzUserInfo, ratio_html: str, expected: float) -> None:
|
||
"""分享率应读取标签后的数值，不误取其他指标或覆盖已有积分。"""
|
||
parser.bonus = 123
|
||
|
||
parser._parse_user_detail_info(f"<html><body><ul>{ratio_html}</ul></body></html>")
|
||
|
||
assert parser.ratio == expected
|
||
assert parser.bonus == 123
|