feat: add hongguo duanju spider

2026-04-25 09:16:29 +08:00
parent 7f42cdffcd
commit 3cbc11537d
2 changed files with 333 additions and 0 deletions
@@ -0,0 +1,127 @@
+import unittest
+from importlib.machinery import SourceFileLoader
+from pathlib import Path
+from unittest.mock import patch
+
+
+ROOT = Path(__file__).resolve().parents[1]
+MODULE = SourceFileLoader("hongguo_spider", str(ROOT / "红果短剧.py")).load_module()
+Spider = MODULE.Spider
+
+
+class TestHongGuoDuanJuSpider(unittest.TestCase):
+    def setUp(self):
+        Spider._instance = None
+        self.spider = Spider()
+        self.spider.init()
+
+    def test_home_content_exposes_categories(self):
+        content = self.spider.homeContent(False)
+        self.assertEqual(content["class"][0]["type_id"], "热播")
+        self.assertIn("分类", content["filters"])
+
+    @patch.object(Spider, "_get_json")
+    def test_home_video_content_uses_hot_category(self, mock_json):
+        mock_json.return_value = {
+            "data": [
+                {"id": 1, "title": "【热播】逆袭人生", "cover": "http://img/1.jpg", "totalChapterNum": 88}
+            ]
+        }
+        result = self.spider.homeVideoContent()
+        self.assertEqual(result["list"][0]["vod_id"], "1")
+        self.assertEqual(result["list"][0]["vod_name"], "逆袭人生")
+        self.assertEqual(result["list"][0]["vod_remarks"], "更新至88集")
+        self.assertIn("name=%E7%83%AD%E6%92%AD", mock_json.call_args.args[0])
+
+    @patch.object(Spider, "_get_json")
+    def test_category_content_builds_query_and_maps_items(self, mock_json):
+        mock_json.return_value = {
+            "data": [
+                {"id": "2", "title": "新剧回归", "cover": "http://img/2.jpg", "totalChapterNum": 36}
+            ]
+        }
+        result = self.spider.categoryContent("新剧", "3", False, {})
+        self.assertEqual(result["page"], 3)
+        self.assertEqual(result["list"][0]["vod_id"], "2")
+        self.assertNotIn("pagecount", result)
+        self.assertIn("name=%E6%96%B0%E5%89%A7&page=3", mock_json.call_args.args[0])
+
+    @patch.object(Spider, "_get_json")
+    def test_search_content_uses_keyword_query(self, mock_json):
+        mock_json.return_value = {
+            "data": [
+                {"id": 3, "title": "霸总归来", "cover": "", "totalChapterNum": 12}
+            ]
+        }
+        result = self.spider.searchContent("霸总", False, "1")
+        self.assertEqual(result["list"][0]["vod_name"], "霸总归来")
+        self.assertIn("name=%E9%9C%B8%E6%80%BB&page=1", mock_json.call_args.args[0])
+
+    @patch.object(Spider, "_get_json")
+    def test_search_content_returns_empty_for_blank_keyword(self, mock_json):
+        result = self.spider.searchContent("", False, "1")
+        self.assertEqual(result["list"], [])
+        mock_json.assert_not_called()
+
+    def test_clean_title_removes_hot_and_new_tags(self):
+        self.assertEqual(self.spider._clean_title("【热播】新剧名"), "新剧名")
+        self.assertEqual(self.spider._clean_title("[新剧热播] 第二部"), "第二部")
+
+    @patch.object(Spider, "_get_json")
+    def test_detail_content_builds_single_playlist(self, mock_json):
+        mock_json.return_value = {
+            "data": [
+                {"title": "【热播】逆袭人生 1", "cover": "http://img/d.jpg", "video_id": "v1"},
+                {"title": "【热播】逆袭人生 2", "cover": "http://img/d.jpg", "video_id": "v2"},
+            ]
+        }
+        result = self.spider.detailContent(["100"])
+        vod = result["list"][0]
+        self.assertEqual(vod["vod_id"], "100")
+        self.assertEqual(vod["vod_name"], "逆袭人生")
+        self.assertEqual(vod["vod_pic"], "http://img/d.jpg")
+        self.assertEqual(vod["vod_remarks"], "更新至2集")
+        self.assertEqual(vod["vod_play_from"], "红果短剧")
+        self.assertIn("逆袭人生 1$v1", vod["vod_play_url"])
+        self.assertIn("逆袭人生 2$v2", vod["vod_play_url"])
+
+    @patch.object(Spider, "_get_json")
+    def test_detail_content_returns_empty_when_id_missing(self, mock_json):
+        result = self.spider.detailContent([""])
+        self.assertEqual(result, {"list": []})
+        mock_json.assert_not_called()
+
+    def test_sort_qualities_prefers_1080p_then_sc_then_sd(self):
+        ordered = self.spider._sort_qualities(
+            [
+                {"quality": "sd", "download_url": "sd.mp4"},
+                {"quality": "1080p", "download_url": "1080.mp4"},
+                {"quality": "sc", "download_url": "sc.mp4"},
+            ]
+        )
+        self.assertEqual([item["quality"] for item in ordered], ["1080p", "sc", "sd"])
+
+    @patch.object(Spider, "_get_json")
+    def test_player_content_returns_best_quality_direct_url(self, mock_json):
+        mock_json.return_value = {
+            "data": {
+                "qualities": [
+                    {"quality": "sd", "title": "标清", "download_url": "http://video/sd.mp4"},
+                    {"quality": "1080p", "title": "超清", "download_url": "http://video/1080.mp4"},
+                ]
+            }
+        }
+        result = self.spider.playerContent("红果短剧", "v1", {})
+        self.assertEqual(result["parse"], 0)
+        self.assertEqual(result["url"], "http://video/1080.mp4")
+
+    @patch.object(Spider, "_get_json")
+    def test_player_content_returns_empty_url_on_error(self, mock_json):
+        mock_json.side_effect = Exception("boom")
+        result = self.spider.playerContent("红果短剧", "v2", {})
+        self.assertEqual(result["parse"], 0)
+        self.assertEqual(result["url"], "")
+
+
+if __name__ == "__main__":
+    unittest.main()