feat: add rrdy category parsing

This commit is contained in:
Harold
2026-04-20 15:42:04 +08:00
parent e91f41b048
commit 1dfd766211
2 changed files with 96 additions and 0 deletions
+46
View File
@@ -1,6 +1,7 @@
import unittest
from importlib.machinery import SourceFileLoader
from pathlib import Path
from unittest.mock import patch
ROOT = Path(__file__).resolve().parents[1]
@@ -47,3 +48,48 @@ class TestRenRenDianYingSpider(unittest.TestCase):
self.assertEqual(self.spider._normalize_title("《繁花》"), "繁花")
self.assertEqual(self.spider._normalize_title("「诛仙」特别篇"), "诛仙")
self.assertEqual(self.spider._normalize_title("普通标题"), "普通标题")
def test_parse_cards_extracts_expected_fields(self):
html = """
<ul id="movielist">
<li>
<div class="pure-img"><img class="pure-img" data-original="/poster.jpg" /></div>
<div class="intro">
<h2><a href="/movie/123.html" title="《示例电影》">《示例电影》</a></h2>
</div>
<div class="dou"><b>8.8</b></div>
</li>
</ul>
"""
self.assertEqual(
self.spider._parse_cards(html),
[
{
"vod_id": "/movie/123.html",
"vod_name": "示例电影",
"vod_pic": "https://www.rrdynb.com/poster.jpg",
"vod_remarks": "8.8",
}
],
)
@patch.object(Spider, "_request_html")
def test_category_content_builds_reference_url_and_page_payload(self, mock_request_html):
mock_request_html.return_value = """
<ul id="movielist">
<li>
<img class="pure-img" data-original="/cate.jpg" />
<div class="intro"><h2><a href="/movie/456.html" title="分类影片">分类影片</a></h2></div>
<div class="dou"><b>更新中</b></div>
</li>
</ul>
"""
result = self.spider.categoryContent("movie/list_2", "3", False, {})
self.assertEqual(
mock_request_html.call_args.args[0],
"https://www.rrdynb.com/movie/list_2_3.html",
)
self.assertEqual(result["page"], 3)
self.assertEqual(result["limit"], 1)
self.assertEqual(result["list"][0]["vod_name"], "分类影片")
self.assertNotIn("pagecount", result)