feat: add 两个BT list parsing

2026-04-24 18:39:47 +08:00
parent 68e6fbac40
commit ff0c9bb0cf
2 changed files with 337 additions and 0 deletions
@@ -0,0 +1,139 @@
+import unittest
+from importlib.machinery import SourceFileLoader
+from pathlib import Path
+from unittest.mock import patch
+
+
+ROOT = Path(__file__).resolve().parents[1]
+MODULE = SourceFileLoader("lianggebt_spider", str(ROOT / "两个BT.py")).load_module()
+Spider = MODULE.Spider
+
+
+class TestLiangGeBTSpider(unittest.TestCase):
+    def setUp(self):
+        Spider._instance = None
+        self.spider = Spider()
+        self.spider.init()
+
+    def test_home_content_exposes_expected_categories(self):
+        content = self.spider.homeContent(False)
+        self.assertEqual(
+            [item["type_id"] for item in content["class"]],
+            [
+                "zgjun",
+                "meiju",
+                "jpsrtv",
+                "movie_bt_tags/xiju",
+                "movie_bt_tags/aiqing",
+                "movie_bt_tags/adt",
+                "movie_bt_tags/at",
+                "movie_bt_tags/donghua",
+                "movie_bt_tags/qihuan",
+                "movie_bt_tags/xuanni",
+                "movie_bt_tags/kehuan",
+                "movie_bt_tags/juqing",
+                "movie_bt_tags/kongbu",
+                "gf",
+            ],
+        )
+        self.assertNotIn("filters", content)
+
+    def test_extract_cards_supports_short_ids_and_relative_images(self):
+        html = """
+        <ul class="item">
+          <li>
+            <a href="/movie/123.html" title="示例影片">
+              <img data-original="/cover.jpg" />
+            </a>
+            <h3><a href="/movie/123.html">示例影片</a></h3>
+            <span class="status">更新至10集</span>
+          </li>
+        </ul>
+        """
+        cards = self.spider._extract_cards(html)
+        self.assertEqual(
+            cards,
+            [
+                {
+                    "vod_id": "123",
+                    "vod_name": "示例影片",
+                    "vod_pic": "https://www.bttwoo.com/cover.jpg",
+                    "vod_remarks": "更新至10集",
+                }
+            ],
+        )
+
+    @patch.object(Spider, "_request_html")
+    def test_home_video_content_requests_home_page(self, mock_request_html):
+        mock_request_html.return_value = """
+        <li>
+          <a href="/movie/200.html" title="首页影片">
+            <img src="/home.jpg" />
+          </a>
+          <h3>首页影片</h3>
+        </li>
+        """
+        result = self.spider.homeVideoContent()
+        self.assertEqual(mock_request_html.call_args.args[0], "https://www.bttwoo.com")
+        self.assertEqual(result["list"][0]["vod_id"], "200")
+
+    @patch.object(Spider, "_request_html")
+    def test_category_content_builds_expected_urls(self, mock_request_html):
+        mock_request_html.return_value = """
+        <li>
+          <a href="/movie/301.html" title="分类影片">
+            <img src="/cate.jpg" />
+          </a>
+          <h3>分类影片</h3>
+          <span class="rating">HD</span>
+        </li>
+        """
+        page_one = self.spider.categoryContent("meiju", "1", False, {})
+        page_three = self.spider.categoryContent("movie_bt_tags/xiju", "3", False, {})
+        self.assertEqual(mock_request_html.call_args_list[0].args[0], "https://www.bttwoo.com/meiju")
+        self.assertEqual(
+            mock_request_html.call_args_list[1].args[0],
+            "https://www.bttwoo.com/movie_bt_tags/xiju?paged=3",
+        )
+        self.assertEqual(page_one["page"], 1)
+        self.assertEqual(page_one["list"][0]["vod_name"], "分类影片")
+        self.assertNotIn("pagecount", page_one)
+        self.assertEqual(page_three["page"], 3)
+
+    @patch.object(Spider, "_request_html")
+    def test_search_content_builds_query_and_filters_irrelevant_results(self, mock_request_html):
+        mock_request_html.return_value = """
+        <li>
+          <a href="/movie/401.html" title="繁花">
+            <img src="/match.jpg" />
+          </a>
+          <h3>繁花</h3>
+        </li>
+        <li>
+          <a href="/movie/402.html" title="无关结果">
+            <img src="/other.jpg" />
+          </a>
+          <h3>无关结果</h3>
+        </li>
+        """
+        result = self.spider.searchContent("繁花", False, "2")
+        self.assertEqual(
+            mock_request_html.call_args.args[0],
+            "https://www.bttwoo.com/xssssearch?q=%E7%B9%81%E8%8A%B1&p=2",
+        )
+        self.assertEqual(result["page"], 2)
+        self.assertEqual(
+            result["list"],
+            [
+                {
+                    "vod_id": "401",
+                    "vod_name": "繁花",
+                    "vod_pic": "https://www.bttwoo.com/match.jpg",
+                    "vod_remarks": "",
+                }
+            ],
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()