feat: scaffold cupfox spider helpers

2026-04-19 20:10:46 +08:00
parent f99824ee40
commit 6d47957195
2 changed files with 132 additions and 0 deletions
@@ -0,0 +1,50 @@
+import base64
+import unittest
+from importlib.machinery import SourceFileLoader
+from pathlib import Path
+from unittest.mock import patch
+
+
+ROOT = Path(__file__).resolve().parents[1]
+MODULE = SourceFileLoader("cupfox_spider", str(ROOT / "茶杯狐.py")).load_module()
+Spider = MODULE.Spider
+
+
+class TestCupfoxSpider(unittest.TestCase):
+    def setUp(self):
+        Spider._instance = None
+        self.spider = Spider()
+        self.spider.init()
+
+    def test_encode_and_decode_ids_keep_short_paths(self):
+        self.assertEqual(
+            self.spider._encode_detail_id("/movie/test-slug.html"),
+            "detail/test-slug",
+        )
+        self.assertEqual(
+            self.spider._decode_detail_id("detail/test-slug"),
+            "https://www.cupfox.ai/movie/test-slug.html",
+        )
+        self.assertEqual(
+            self.spider._encode_play_id("/play/abc123.html"),
+            "play/abc123",
+        )
+        self.assertEqual(
+            self.spider._decode_play_id("play/abc123"),
+            "https://www.cupfox.ai/play/abc123.html",
+        )
+
+    def test_merge_set_cookie_and_cookie_header(self):
+        jar = {}
+        self.spider._merge_set_cookie(jar, ["foo=1; Path=/", "bar=2; HttpOnly"])
+        self.assertEqual(jar, {"foo": "1", "bar": "2"})
+        self.assertEqual(self.spider._cookie_header(jar), "foo=1; bar=2")
+
+    def test_extract_firewall_token(self):
+        html = '<script>var token = encrypt("abcXYZ");</script>'
+        self.assertEqual(self.spider._extract_firewall_token(html), "abcXYZ")
+
+    @patch("cupfox_spider.random.randint", side_effect=[0, 1, 2, 3])
+    def test_firewall_encrypt_returns_base64_text(self, _mock_randint):
+        encoded = self.spider._cupfox_firewall_encrypt("PX")
+        self.assertEqual(base64.b64decode(encoded).decode("utf-8"), "PwXh7w")