import unittest
import base64
from importlib.machinery import SourceFileLoader
from pathlib import Path
from unittest.mock import patch
ROOT = Path(__file__).resolve().parents[1]
MODULE = SourceFileLoader("dbku_spider", str(ROOT / "独播库.py")).load_module()
Spider = MODULE.Spider
class TestDBKUSpider(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", "movie", "variety", "anime", "hk", "luju"])
def test_parse_list_cards_extracts_detail_url_title_cover_and_description(self):
html = """
"""
cards = self.spider._parse_list_cards(html)
self.assertEqual(
cards,
[
{
"vod_id": "https://www.dbku.tv/voddetail/123.html",
"vod_name": "示例影片",
"vod_pic": "https://img.example/dbku.jpg",
"vod_remarks": "更新至10集",
}
],
)
@patch.object(Spider, "fetch")
def test_request_html_uses_dbku_headers(self, mock_fetch):
class FakeResponse:
def __init__(self, text):
self.text = text
self.status_code = 200
self.encoding = "utf-8"
mock_fetch.return_value = FakeResponse("ok")
html = self.spider._request_html("/vodtype/1--------1---.html", expect_xpath="//body")
self.assertIn("ok", html)
called_headers = mock_fetch.call_args.kwargs["headers"]
self.assertEqual(called_headers["Referer"], "https://www.dbku.tv")
self.assertEqual(called_headers["Origin"], "https://www.dbku.tv")
@patch.object(Spider, "_request_html")
def test_category_content_builds_page_result(self, mock_request_html):
mock_request_html.return_value = """
"""
result = self.spider.categoryContent("movie", "2", False, {})
self.assertEqual(result["page"], 2)
self.assertEqual(result["list"][0]["vod_name"], "分类影片")
self.assertEqual(result["list"][0]["vod_pic"], "https://www.dbku.tv/cover.jpg")
def test_parse_search_cards_prefers_search_list_container(self):
html = """
"""
results = self.spider._parse_search_cards(html)
self.assertEqual(len(results), 1)
self.assertEqual(results[0]["vod_name"], "搜索命中")
@patch.object(Spider, "_request_html")
def test_search_content_reuses_search_parser(self, mock_request_html):
mock_request_html.return_value = """
"""
result = self.spider.searchContent("繁花", False, "1")
self.assertEqual(result["list"][0]["vod_id"], "https://www.dbku.tv/voddetail/321.html")
self.assertEqual(result["list"][0]["vod_name"], "搜索影片")
def test_parse_detail_page_extracts_meta_and_episodes(self):
html = """
独播剧
年份：2025
地区：大陆
导演：张三
主演：李四
一段剧情简介
第1集
第2集
"""
result = self.spider._parse_detail_page(html, "https://www.dbku.tv/voddetail/100.html")
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "独播剧")
self.assertEqual(vod["vod_year"], "2025")
self.assertEqual(vod["vod_play_from"], "独播库")
self.assertIn("第1集$https://www.dbku.tv/vodplay/100-1-1.html", vod["vod_play_url"])
@patch.object(Spider, "_request_html")
def test_detail_content_reads_from_vod_id_url(self, mock_request_html):
mock_request_html.return_value = """
详情影片
第1集
"""
result = self.spider.detailContent(["https://www.dbku.tv/voddetail/200.html"])
self.assertEqual(result["list"][0]["vod_id"], "https://www.dbku.tv/voddetail/200.html")
self.assertEqual(result["list"][0]["vod_name"], "详情影片")
def test_parse_player_data_reads_json_block(self):
html = ''
data = self.spider._parse_player_data(html)
self.assertEqual(data["url"], "https://video.example/a.m3u8")
def test_decode_play_url_by_encrypt_supports_modes_0_1_2_3(self):
self.assertEqual(
self.spider._decode_play_url_by_encrypt("https://video.example/raw.m3u8", 0),
"https://video.example/raw.m3u8",
)
self.assertEqual(
self.spider._decode_play_url_by_encrypt("https%3A//video.example/escape.m3u8", 1),
"https://video.example/escape.m3u8",
)
mode2 = base64.b64encode("https://video.example/base64.m3u8".encode("utf-8")).decode("utf-8")
self.assertEqual(
self.spider._decode_play_url_by_encrypt(mode2, 2),
"https://video.example/base64.m3u8",
)
mode3_raw = "ABCDEFGHhttps://video.example/trimmed.m3u8HGFEDCBA"
mode3 = base64.b64encode(mode3_raw.encode("utf-8")).decode("utf-8")
self.assertEqual(
self.spider._decode_play_url_by_encrypt("12345678" + mode3, 3),
"https://video.example/trimmed.m3u8",
)
@patch.object(Spider, "_request_html")
def test_player_content_returns_decoded_direct_url(self, mock_request_html):
mock_request_html.return_value = """
"""
result = self.spider.playerContent("独播库", "https://www.dbku.tv/vodplay/100-1-1.html", {})
self.assertEqual(result["parse"], 0)
self.assertEqual(result["url"], "https://video.example/final.m3u8")
self.assertEqual(result["header"]["Referer"], "https://www.dbku.tv/vodplay/100-1-1.html")
@patch.object(Spider, "_request_html")
def test_player_content_follows_internal_jump(self, mock_request_html):
mock_request_html.side_effect = [
'',
'',
]
result = self.spider.playerContent("独播库", "https://www.dbku.tv/vodplay/100-1-1.html", {})
self.assertEqual(result["parse"], 0)
self.assertEqual(result["url"], "https://video.example/jump-final.m3u8")
if __name__ == "__main__":
unittest.main()
