feat: add startup forced update check

2026-07-07 17:58:28 +08:00
parent 610a2304cc
commit 95b180e9cf
9 changed files with 527 additions and 12 deletions
@@ -0,0 +1,122 @@
+import socket
+import os
+import sys
+import unittest
+
+sys.path.insert(0, os.path.dirname(__file__))
+from _helpers import REPO_ROOT
+
+from app import update_check
+
+assert REPO_ROOT
+
+
+class UpdateCheckTests(unittest.TestCase):
+    def test_compare_versions_uses_numeric_segments(self):
+        self.assertGreater(update_check.compare_versions("0.10.0", "0.2.0"), 0)
+        self.assertEqual(0, update_check.compare_versions("1.2", "1.2.0"))
+        self.assertLess(update_check.compare_versions("v1.2.3", "1.2.4"), 0)
+
+    def test_forced_update_by_min_supported_version(self):
+        info = update_check.UpdateInfo(
+            latest_version="1.2.0",
+            min_supported_version="1.1.0",
+            force_update=False,
+        )
+
+        self.assertTrue(update_check.is_forced_update(info, "1.0.9"))
+        self.assertFalse(update_check.is_forced_update(info, "1.1.0"))
+
+    def test_forced_update_by_force_flag_and_latest_version(self):
+        forced = update_check.UpdateInfo(latest_version="1.2.0", force_update=True)
+        optional = update_check.UpdateInfo(latest_version="1.2.0", force_update=False)
+
+        self.assertTrue(update_check.is_forced_update(forced, "1.1.9"))
+        self.assertFalse(update_check.is_forced_update(optional, "1.1.9"))
+        self.assertFalse(update_check.is_forced_update(forced, "1.2.0"))
+
+    def test_check_for_update_forced_response(self):
+        def fetcher(_url, _timeout):
+            return {
+                "latest_version": "1.2.0",
+                "min_supported_version": "1.1.0",
+                "force_update": True,
+                "download_url": "https://example.test/cmshopee.zip",
+                "sha256": "abc",
+                "message": "请升级后继续使用",
+            }
+
+        result = update_check.check_for_update(
+            current_version="1.0.0",
+            url="https://example.test/version.json",
+            fetcher=fetcher,
+        )
+
+        self.assertTrue(result.checked)
+        self.assertTrue(result.forced)
+        self.assertFalse(result.can_enter)
+        self.assertEqual("1.2.0", result.latest_version)
+        self.assertEqual("https://example.test/cmshopee.zip", result.download_url)
+
+    def test_check_for_update_accepts_release_wrapper_response(self):
+        def fetcher(_url, _timeout):
+            return {
+                "platform": "windows",
+                "release": {
+                    "version": "0.1.1",
+                    "download_url": "https://example.test/cmshopee-0.1.1.zip",
+                    "sha256": "abc",
+                    "release_notes": "新版说明",
+                },
+            }
+
+        result = update_check.check_for_update(
+            current_version="0.1.0",
+            url="https://example.test/releases/latest?platform=windows",
+            fetcher=fetcher,
+        )
+
+        self.assertTrue(result.checked)
+        self.assertFalse(result.forced)
+        self.assertEqual("0.1.1", result.latest_version)
+        self.assertEqual("https://example.test/cmshopee-0.1.1.zip", result.download_url)
+        self.assertEqual("abc", result.sha256)
+        self.assertEqual("新版说明", result.message)
+
+    def test_network_failure_allows_entry(self):
+        def fetcher(_url, _timeout):
+            raise socket.timeout("timeout")
+
+        result = update_check.check_for_update(
+            current_version="1.0.0",
+            url="https://example.test/version.json",
+            fetcher=fetcher,
+        )
+
+        self.assertTrue(result.checked)
+        self.assertFalse(result.forced)
+        self.assertTrue(result.can_enter)
+        self.assertIn("启动版本检查失败", result.error)
+
+    def test_invalid_response_allows_entry(self):
+        result = update_check.check_for_update(
+            current_version="1.0.0",
+            url="https://example.test/version.json",
+            fetcher=lambda _url, _timeout: {"message": "missing versions"},
+        )
+
+        self.assertTrue(result.checked)
+        self.assertFalse(result.forced)
+        self.assertTrue(result.can_enter)
+        self.assertIn("缺少 latest_version", result.error)
+
+    def test_empty_update_url_skips_check(self):
+        result = update_check.check_for_update(current_version="1.0.0", url="")
+
+        self.assertFalse(result.checked)
+        self.assertFalse(result.forced)
+        self.assertEqual("", result.error)
+
+
+if __name__ == "__main__":
+    unittest.main()