import unittest from base64 import b64encode from importlib.machinery import SourceFileLoader from pathlib import Path from unittest.mock import patch from Cryptodome.Cipher import AES from Cryptodome.Util.Padding import pad 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_default_host_prefers_czzy89(self): self.assertEqual(self.spider.hosts[0], "https://www.czzy89.com") self.assertEqual(self.spider.current_host, "https://www.czzy89.com") 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 = """

更新至10集

""" 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集", } ], ) @patch.object(Spider, "fetch") def test_request_html_falls_back_to_second_host(self, mock_fetch): class FakeResponse: def __init__(self, text, status_code=200): self.text = text self.status_code = status_code self.encoding = "utf-8" mock_fetch.side_effect = [ Exception("first host down"), FakeResponse("ok"), ] html, host = self.spider._request_html("/movie_bt/page/1", expect_xpath="//body") self.assertIn("ok", html) self.assertEqual(host, "https://www.cz01.org") self.assertEqual(self.spider.current_host, "https://www.cz01.org") @patch.object(Spider, "_request_html") def test_category_content_builds_media_list(self, mock_request_html): mock_request_html.return_value = ( """

""", "https://www.czzy89.com", ) 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.czzy89.com/cover.jpg") @patch.object(Spider, "_request_html") def test_search_content_reuses_media_card_parser(self, mock_request_html): mock_request_html.return_value = ( """

""", "https://www.czzy89.com", ) result = self.spider.searchContent("繁花", False, "1") self.assertEqual(result["list"][0]["vod_id"], "/movie/search-hit.html") self.assertEqual(result["list"][0]["vod_name"], "搜索影片") def test_parse_detail_page_splits_direct_and_pan_sources(self): html = """

繁花

年份：2024

地区：中国大陆

导演：王家卫

主演：胡歌

一段剧情简介

第1集第2集

阿里云盘

""" detail = self.spider._parse_detail_page(html, "https://www.czzy89.com", "/movie/fanhua.html") vod = detail["list"][0] self.assertEqual(vod["vod_name"], "繁花") self.assertEqual(vod["vod_year"], "2024") self.assertEqual(vod["vod_play_from"], "厂长资源$$$网盘资源") self.assertIn("第1集$https://www.czzy89.com/v_play/1.html", vod["vod_play_url"]) self.assertIn("阿里云盘$https://www.alipan.com/s/demo", vod["vod_play_url"]) @patch.object(Spider, "_request_html") def test_detail_content_reads_from_vod_id_path(self, mock_request_html): mock_request_html.return_value = ( """

示例影片

立即播放

""", "https://www.czzy89.com", ) result = self.spider.detailContent(["/movie/example.html"]) self.assertEqual(result["list"][0]["vod_id"], "/movie/example.html") self.assertEqual(result["list"][0]["vod_name"], "示例影片") def test_extract_iframe_src(self): html = '' self.assertEqual( self.spider._extract_iframe_src(html, "https://www.czzy89.com"), "https://www.czzy89.com/player-v2/test", ) def test_extract_player_url_prefers_mysvg_then_art_url(self): self.assertEqual( self.spider._extract_player_url_from_iframe("var mysvg='https://video.example/a.m3u8';"), "https://video.example/a.m3u8", ) self.assertEqual( self.spider._extract_player_url_from_iframe("art.url='https://video.example/b.m3u8';"), "https://video.example/b.m3u8", ) def test_extract_player_url_decodes_data_payload(self): original = "https://video.example/data.m3u8" middle = len(original) // 2 obfuscated = original[:middle] + "ABCDEFG" + original[middle:] encoded = "".join("{:02x}".format(ord(ch)) for ch in obfuscated)[::-1] html = 'var config = {"data":"%s"};' % encoded self.assertEqual(self.spider._extract_player_url_from_iframe(html), original) def test_extract_player_url_decrypts_player_payload(self): payload = '{"url":"https://video.example/encrypted.m3u8"}' key = b"VFBTzdujpR9FWBhe" iv = b"1234567890abcdef" cipher = AES.new(key, AES.MODE_CBC, iv) encrypted = b64encode(cipher.encrypt(pad(payload.encode("utf-8"), AES.block_size))).decode("utf-8") html = 'var player="%s";var rand="%s";' % (encrypted, iv.decode("utf-8")) self.assertEqual( self.spider._extract_player_url_from_iframe(html), "https://video.example/encrypted.m3u8", ) def test_extract_player_url_supports_wp_nonce_fallback(self): html = """ """ self.assertEqual( self.spider._extract_player_url_from_iframe(html), "https://video.example/wp.m3u8", ) @patch.object(Spider, "_request_html") def test_player_content_returns_pan_link_directly(self, mock_request_html): result = self.spider.playerContent("网盘资源", "https://www.alipan.com/s/demo", {}) self.assertEqual(result["parse"], 0) self.assertEqual(result["url"], "https://www.alipan.com/s/demo") mock_request_html.assert_not_called() @patch.object(Spider, "_request_html") def test_player_content_resolves_iframe_chain(self, mock_request_html): mock_request_html.side_effect = [ ('', "https://www.czzy89.com"), ("var mysvg='https://media.example/final.m3u8';", "https://www.czzy89.com"), ] result = self.spider.playerContent("厂长资源", "https://www.czzy89.com/v_play/1.html", {}) self.assertEqual(result["parse"], 0) self.assertEqual(result["url"], "https://media.example/final.m3u8") self.assertIn("Referer", result["header"]) if __name__ == "__main__": unittest.main()