49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
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()
|