feat: scaffold czzy spider parsing

2026-04-18 11:16:07 +08:00
parent 079a058215
commit 4b40851bab
2 changed files with 144 additions and 0 deletions
@@ -0,0 +1,48 @@
+import unittest
+from importlib.machinery import SourceFileLoader
+from pathlib import Path
+
+
+ROOT = Path(__file__).resolve().parents[1]
+MODULE = SourceFileLoader("czzy_spider", str(ROOT / "厂长资源.py")).load_module()
+Spider = MODULE.Spider
+
+
+class TestCZZYSpider(unittest.TestCase):
+    def setUp(self):
+        self.spider = Spider()
+        self.spider.init()
+
+    def test_home_content_exposes_expected_categories(self):
+        content = self.spider.homeContent(False)
+        class_ids = [item["type_id"] for item in content["class"]]
+        self.assertEqual(class_ids[:3], ["movie", "tv", "anime"])
+        self.assertIn("cn_drama", class_ids)
+
+    def test_parse_media_cards_extracts_basic_fields(self):
+        html = """
+        <ul class="mi_ne_kd">
+          <li>
+            <a href="/movie/abc.html" title="链接标题">
+              <img data-original="https://img.example/cover.jpg" alt="测试影片" />
+            </a>
+            <span class="jidi">更新至10集</span>
+          </li>
+        </ul>
+        """
+        cards = self.spider._parse_media_cards(html, "https://www.cz01.org")
+        self.assertEqual(
+            cards,
+            [
+                {
+                    "vod_id": "/movie/abc.html",
+                    "vod_name": "测试影片",
+                    "vod_pic": "https://img.example/cover.jpg",
+                    "vod_remarks": "更新至10集",
+                }
+            ],
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()