191 lines
8.4 KiB
Python
191 lines
8.4 KiB
Python
import unittest
|
|
from importlib.machinery import SourceFileLoader
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
MODULE = SourceFileLoader("youknow_spider", str(ROOT / "youknow.py")).load_module()
|
|
Spider = MODULE.Spider
|
|
|
|
|
|
class TestYouKnowSpider(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, ["index", "drama", "movie", "variety", "anime", "short", "doc"])
|
|
|
|
def test_parse_list_cards_extracts_compact_vod_id(self):
|
|
html = """
|
|
<a class="module-poster-item" href="/v/1234.html" title="示例影片" data-original="/cover.jpg">
|
|
<div class="module-item-note">更新至10集</div>
|
|
</a>
|
|
"""
|
|
cards = self.spider._parse_list_cards(html)
|
|
self.assertEqual(
|
|
cards,
|
|
[
|
|
{
|
|
"vod_id": "1234",
|
|
"vod_name": "示例影片",
|
|
"vod_pic": "https://www.youknow.tv/cover.jpg",
|
|
"vod_remarks": "更新至10集",
|
|
}
|
|
],
|
|
)
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_home_video_content_uses_today_updates_page(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<a class="module-poster-item" href="/v/111.html" title="今日更新" data-original="/recent.jpg">
|
|
<div class="module-item-note">HD</div>
|
|
</a>
|
|
"""
|
|
result = self.spider.homeVideoContent()
|
|
self.assertEqual(result["list"][0]["vod_id"], "111")
|
|
self.assertEqual(result["list"][0]["vod_name"], "今日更新")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_category_content_builds_page_result(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<a class="module-poster-item" href="/v/222.html" title="分类影片" data-original="/cate.jpg">
|
|
<div class="module-item-note">完结</div>
|
|
</a>
|
|
"""
|
|
result = self.spider.categoryContent("movie", "2", False, {})
|
|
self.assertEqual(result["page"], 2)
|
|
self.assertEqual(result["list"][0]["vod_id"], "222")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_search_content_reuses_card_parser(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<a class="module-poster-item" href="/v/333.html" title="搜索影片" data-original="/search.jpg">
|
|
<div class="module-item-note">抢先版</div>
|
|
</a>
|
|
"""
|
|
result = self.spider.searchContent("繁花", False, "1")
|
|
self.assertEqual(result["list"][0]["vod_id"], "333")
|
|
self.assertEqual(result["list"][0]["vod_name"], "搜索影片")
|
|
|
|
def test_parse_detail_page_extracts_fields_and_aligns_episode_sources(self):
|
|
html = """
|
|
<div class="module-info-poster">
|
|
<img data-original="/poster.jpg" />
|
|
</div>
|
|
<div class="module-info-heading">
|
|
<h1>示例剧</h1>
|
|
</div>
|
|
<div class="module-info-content">
|
|
<div><span>类型：</span><a>剧情</a></div>
|
|
<div><span>地区：</span><a>大陆</a></div>
|
|
<div><span>年份：</span><a>2026</a></div>
|
|
<div><span>导演：</span><a>张三</a></div>
|
|
<div><span>主演：</span><a>李四</a><a>王五</a></div>
|
|
<div><span>语言：</span><a>国语</a></div>
|
|
<div><span>备注：</span><span>更新至2集</span></div>
|
|
<div><span>简介：</span><p>一段剧情简介</p></div>
|
|
</div>
|
|
<div data-dropdown-value="线路A"></div>
|
|
<div data-dropdown-value="线路B"></div>
|
|
<div class="module-play-list">
|
|
<a href="/p/888-1-1/">第1集</a>
|
|
<a href="/p/888-1-2/">第2集</a>
|
|
</div>
|
|
<div class="module-play-list">
|
|
<a href="/p/888-2-1/">第1集</a>
|
|
<a href="/p/888-2-2/">第2集</a>
|
|
</div>
|
|
"""
|
|
result = self.spider._parse_detail_page(html, "888")
|
|
vod = result["list"][0]
|
|
self.assertEqual(vod["vod_id"], "888")
|
|
self.assertEqual(vod["path"], "https://www.youknow.tv/v/888.html")
|
|
self.assertEqual(vod["vod_name"], "示例剧")
|
|
self.assertEqual(vod["type_name"], "剧情")
|
|
self.assertEqual(vod["vod_area"], "大陆")
|
|
self.assertEqual(vod["vod_year"], "2026")
|
|
self.assertEqual(vod["vod_lang"], "国语")
|
|
self.assertEqual(vod["vod_remarks"], "更新至2集")
|
|
self.assertEqual(vod["vod_director"], "张三")
|
|
self.assertEqual(vod["vod_actor"], "李四,王五")
|
|
self.assertEqual(vod["vod_content"], "一段剧情简介")
|
|
self.assertEqual(vod["vod_play_from"], "YouKnowTV")
|
|
first_episode = vod["vod_play_url"].split("#")[0]
|
|
self.assertTrue(first_episode.startswith("第1集$"))
|
|
payload = self.spider._decode_episode_payload(first_episode.split("$", 1)[1])
|
|
self.assertEqual(payload["vod_id"], "888")
|
|
self.assertEqual(payload["episode_index"], 1)
|
|
self.assertEqual(len(payload["candidates"]), 2)
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_detail_content_builds_detail_request_url_from_vod_id(self, mock_request_html):
|
|
mock_request_html.return_value = '<h1>详情影片</h1><div class="module-play-list"><a href="/p/123-1-1/">第1集</a></div>'
|
|
result = self.spider.detailContent(["123"])
|
|
self.assertEqual(mock_request_html.call_args.args[0], "https://www.youknow.tv/v/123.html")
|
|
self.assertEqual(result["list"][0]["vod_id"], "123")
|
|
|
|
def test_extract_player_config_reads_player_aaaa(self):
|
|
html = '<script>var player_aaaa={"url":"https%3A%2F%2Fcdn.example%2Fa.m3u8","encrypt":"1"};</script>'
|
|
data = self.spider._parse_player_config(html)
|
|
self.assertEqual(data["encrypt"], "1")
|
|
|
|
def test_decode_player_url_supports_encrypt_1_and_2(self):
|
|
self.assertEqual(
|
|
self.spider._decode_player_url("https%3A%2F%2Fvideo.example%2Fa.m3u8", "1"),
|
|
"https://video.example/a.m3u8",
|
|
)
|
|
encoded = "aHR0cHMlM0ElMkYlMkZ2aWRlby5leGFtcGxlJTJGYi5tM3U4"
|
|
self.assertEqual(
|
|
self.spider._decode_player_url(encoded, "2"),
|
|
"https://video.example/b.m3u8",
|
|
)
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_player_content_collects_candidates_from_page_and_iframe(self, mock_request_html):
|
|
payload = self.spider._encode_episode_payload(
|
|
{
|
|
"vod_id": "888",
|
|
"episode_index": 1,
|
|
"title": "第1集",
|
|
"candidates": [
|
|
{"source": "线路A", "source_id": "1", "episode_url": "https://www.youknow.tv/p/888-1-1/"}
|
|
],
|
|
}
|
|
)
|
|
mock_request_html.side_effect = [
|
|
'<script>var player_aaaa={"url":"https%3A%2F%2Fvideo.example%2Fpage.m3u8","encrypt":"1"};</script><iframe class="embed-responsive-item" src="/embed/player.html?url=https%3A%2F%2Fvideo.example%2Fiframe.m3u8"></iframe>',
|
|
'<html><body>https://video.example/iframe-final.m3u8</body></html>',
|
|
]
|
|
result = self.spider.playerContent("YouKnowTV", payload, {})
|
|
self.assertEqual(result["parse"], 0)
|
|
self.assertEqual(result["url"], "https://video.example/page.m3u8")
|
|
self.assertEqual(result["header"]["Referer"], "https://www.youknow.tv/")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_player_content_tries_next_candidate_when_first_fails(self, mock_request_html):
|
|
payload = self.spider._encode_episode_payload(
|
|
{
|
|
"vod_id": "888",
|
|
"episode_index": 1,
|
|
"title": "第1集",
|
|
"candidates": [
|
|
{"source": "线路A", "source_id": "1", "episode_url": "https://www.youknow.tv/p/888-1-1/"},
|
|
{"source": "线路B", "source_id": "2", "episode_url": "https://www.youknow.tv/p/888-2-1/"},
|
|
],
|
|
}
|
|
)
|
|
mock_request_html.side_effect = [
|
|
"<html><body>empty</body></html>",
|
|
"<html><body>https://video.example/backup.m3u8</body></html>",
|
|
]
|
|
result = self.spider.playerContent("YouKnowTV", payload, {})
|
|
self.assertEqual(result["url"], "https://video.example/backup.m3u8")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|