163 lines
7.5 KiB
Python
163 lines
7.5 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("feikuai_spider", str(ROOT / "飞快TV.py")).load_module()
|
|
Spider = MODULE.Spider
|
|
|
|
|
|
class TestFeikuaiSpider(unittest.TestCase):
|
|
def setUp(self):
|
|
Spider._instance = None
|
|
self.spider = Spider()
|
|
self.spider.init()
|
|
|
|
def test_home_content_returns_expected_classes(self):
|
|
result = self.spider.homeContent(False)
|
|
self.assertEqual(
|
|
result["class"],
|
|
[
|
|
{"type_id": "1", "type_name": "电影"},
|
|
{"type_id": "2", "type_name": "剧集"},
|
|
{"type_id": "3", "type_name": "综艺"},
|
|
{"type_id": "4", "type_name": "动漫"},
|
|
],
|
|
)
|
|
|
|
def test_home_video_content_returns_empty_list(self):
|
|
self.assertEqual(self.spider.homeVideoContent(), {"list": []})
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_category_content_parses_short_vod_id(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<a class="module-poster-item" href="/voddetail/12345.html" title="分类影片">
|
|
<img class="lazy" data-original="/cover.jpg" />
|
|
<div class="module-item-note">更新至10集</div>
|
|
</a>
|
|
"""
|
|
result = self.spider.categoryContent("2", "3", False, {})
|
|
self.assertEqual(
|
|
mock_request_html.call_args.args[0],
|
|
"https://feikuai.tv/vodshow/2--------3---.html",
|
|
)
|
|
self.assertEqual(
|
|
result["list"],
|
|
[
|
|
{
|
|
"vod_id": "/voddetail/12345.html",
|
|
"vod_name": "分类影片",
|
|
"vod_pic": "https://feikuai.tv/cover.jpg",
|
|
"vod_remarks": "更新至10集",
|
|
}
|
|
],
|
|
)
|
|
self.assertNotIn("pagecount", result)
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_search_content_parses_cards_and_blank_keyword(self, mock_request_html):
|
|
blank = self.spider.searchContent("", False, "1")
|
|
self.assertEqual(blank, {"page": 1, "limit": 0, "total": 0, "list": []})
|
|
mock_request_html.assert_not_called()
|
|
|
|
mock_request_html.return_value = """
|
|
<div class="module-card-item module-item">
|
|
<a class="module-card-item-poster" href="/voddetail/67890.html"></a>
|
|
<div class="module-item-pic"><img data-original="/search.jpg" /></div>
|
|
<div class="module-card-item-title"><strong>搜索命中</strong></div>
|
|
<div class="module-item-note">HD</div>
|
|
</div>
|
|
"""
|
|
result = self.spider.searchContent("繁花", False, "2")
|
|
self.assertEqual(
|
|
mock_request_html.call_args.args[0],
|
|
"https://feikuai.tv/label/search_ajax.html?wd=%E7%B9%81%E8%8A%B1&by=time&order=desc&page=2",
|
|
)
|
|
self.assertEqual(result["list"][0]["vod_id"], "/voddetail/67890.html")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_detail_content_merges_online_and_pan_groups(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<h1>示例影片</h1>
|
|
<div class="module-item-pic"><img data-original="/detail.jpg" /></div>
|
|
<div class="module-info-introduction-content">这里是简介</div>
|
|
<div class="module-tab-items-box">
|
|
<div class="module-tab-item"><span>线路A</span></div>
|
|
<div class="module-tab-item"><span>线路B</span></div>
|
|
</div>
|
|
<div class="module-list tab-list">
|
|
<a class="module-play-list-link" href="/vodplay/1-1-1.html">第1集</a>
|
|
<a class="module-play-list-link" href="/vodplay/1-1-2.html">第2集</a>
|
|
</div>
|
|
<div class="module-list tab-list">
|
|
<a class="module-play-list-link" href="/vodplay/1-2-1.html">第1集</a>
|
|
</div>
|
|
<div class="module-list">
|
|
<div class="tab-content">
|
|
<h4>夸克资源@分享一</h4>
|
|
<p>https://pan.quark.cn/s/demo1</p>
|
|
</div>
|
|
<div class="tab-content">
|
|
<h4>百度合集@分享二</h4>
|
|
<p>https://pan.baidu.com/s/demo2</p>
|
|
</div>
|
|
</div>
|
|
"""
|
|
result = self.spider.detailContent(["/voddetail/1.html"])
|
|
vod = result["list"][0]
|
|
self.assertEqual(vod["vod_id"], "/voddetail/1.html")
|
|
self.assertEqual(vod["vod_name"], "示例影片")
|
|
self.assertEqual(vod["vod_pic"], "https://feikuai.tv/detail.jpg")
|
|
self.assertEqual(vod["vod_content"], "这里是简介")
|
|
self.assertEqual(vod["vod_play_from"], "线路A$$$线路B$$$quark$$$baidu")
|
|
self.assertEqual(
|
|
vod["vod_play_url"],
|
|
"第1集$/vodplay/1-1-1.html#第2集$/vodplay/1-1-2.html$$$第1集$/vodplay/1-2-1.html$$$夸克资源$https://pan.quark.cn/s/demo1$$$百度合集$https://pan.baidu.com/s/demo2",
|
|
)
|
|
|
|
def test_base64decode_decodes_fixture(self):
|
|
self.assertEqual(
|
|
self.spider._base64decode("aHR0cHM6Ly9jZG4uZXhhbXBsZS5jb20vdjIubTN1OA=="),
|
|
"https://cdn.example.com/v2.m3u8",
|
|
)
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_player_content_supports_encrypt_1_and_encrypt_2(self, mock_request_html):
|
|
mock_request_html.side_effect = [
|
|
'<script>player_aaaa={"url":"https%3A//cdn.example.com/v1.m3u8","encrypt":"1"}</script>',
|
|
'<script>player_aaaa={"url":"aHR0cHM6Ly9jZG4uZXhhbXBsZS5jb20vdjIubTN1OA==","encrypt":"2"}</script>',
|
|
]
|
|
direct = self.spider.playerContent("feikuai", "/vodplay/1-1-1.html", {})
|
|
encoded = self.spider.playerContent("feikuai", "/vodplay/1-1-2.html", {})
|
|
self.assertEqual(direct["parse"], 0)
|
|
self.assertEqual(direct["url"], "https://cdn.example.com/v1.m3u8")
|
|
self.assertEqual(encoded["parse"], 0)
|
|
self.assertEqual(encoded["url"], "https://cdn.example.com/v2.m3u8")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_player_content_handles_live_nested_player_payload(self, mock_request_html):
|
|
mock_request_html.return_value = """
|
|
<script type="text/javascript">
|
|
var player_aaaa={"flag":"play","encrypt":2,"trysee":0,"points":0,
|
|
"link_current":"\\/vodplay\\/236443-1-1.html","link":"\\/vodplay\\/236443-{sid}-{nid}.html",
|
|
"link_next":"","link_pre":"",
|
|
"vod_data":{"vod_name":"\\u6076\\u5973\\u5b66\\u9662","vod_actor":"","vod_director":"Jamie Grefe","vod_class":"\\u5267\\u60c5"},
|
|
"url":"JTY4JTc0JTc0JTcwJTczJTNBJTJGJTJGJTYzJTY0JTZFJTJFJTc5JTdBJTdBJTc5JTc2JTY5JTcwJTJEJTMyJTM5JTJFJTYzJTZGJTZEJTJGJTMyJTMwJTMyJTM2JTMwJTM0JTMyJTM1JTJGJTMyJTMzJTMyJTM5JTMxJTVGJTM0JTM1JTM5JTM2JTM3JTYzJTM3JTMzJTJGJTY5JTZFJTY0JTY1JTc4JTJFJTZEJTMzJTc1JTM4",
|
|
"from":"1080zyk","server":"no","note":"","id":"236443","sid":1,"nid":1}
|
|
</script>
|
|
"""
|
|
result = self.spider.playerContent("feikuai", "/vodplay/236443-1-1.html", {})
|
|
self.assertEqual(result["parse"], 0)
|
|
self.assertEqual(result["jx"], 0)
|
|
self.assertEqual(result["url"], "https://cdn.yzzyvip-29.com/20260425/23291_45967c73/index.m3u8")
|
|
|
|
@patch.object(Spider, "_request_html")
|
|
def test_player_content_falls_back_when_script_missing(self, mock_request_html):
|
|
mock_request_html.return_value = "<html><body>empty</body></html>"
|
|
result = self.spider.playerContent("feikuai", "/vodplay/1-1-3.html", {})
|
|
self.assertEqual(result["parse"], 1)
|
|
self.assertEqual(result["jx"], 1)
|
|
self.assertEqual(result["url"], "https://feikuai.tv/vodplay/1-1-3.html")
|