feat: 系统设置增加备份恢复与默认登录 admin

支持手动/每日自动备份四所数据库、K线库与 env，上传 zip 一键恢复；中控默认账号 admin/admin123。 Co-authored-by: Cursor <cursoragent@cursor.com>
2026-07-02 16:39:46 +08:00
parent 55261b7812
commit bfa3352122
16 changed files with 1052 additions and 22 deletions
@@ -0,0 +1,65 @@
+"""hub_backup_lib 单元测试。"""
+from __future__ import annotations
+
+import json
+import sys
+import tempfile
+import unittest
+import zipfile
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[2]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
+from lib.hub import hub_backup_lib as backup
+
+
+class HubBackupLibTest(unittest.TestCase):
+    def test_normalize_backup_settings(self):
+        cfg = backup.normalize_backup_settings({"auto_hour": 99, "retention_days": 0})
+        self.assertEqual(cfg["auto_hour"], 23)
+        self.assertEqual(cfg["retention_days"], 1)
+
+    def test_safe_archive_name(self):
+        self.assertTrue(backup._safe_archive_name("backup_2026-07-02_163045.zip"))
+        self.assertFalse(backup._safe_archive_name("../evil.zip"))
+
+    def test_run_and_restore_roundtrip(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp) / "portal"
+            root.mkdir(parents=True)
+            settings = {
+                "backup": {
+                    "auto_enabled": False,
+                    "backup_root": str(root),
+                    "include_env": False,
+                    "include_exchange_images": False,
+                }
+            }
+            hub_settings = backup.HUB_DIR / "hub_settings.json"
+            had = hub_settings.is_file()
+            old = hub_settings.read_text(encoding="utf-8") if had else None
+            try:
+                if not had:
+                    hub_settings.write_text('{"version":1,"exchanges":[]}', encoding="utf-8")
+                result = backup.run_backup(trigger="manual", settings=settings)
+                self.assertTrue(result.get("ok"), result)
+                archive = Path(result["path"])
+                self.assertTrue(archive.is_file())
+                with zipfile.ZipFile(archive, "r") as zf:
+                    names = zf.namelist()
+                self.assertIn("manifest.json", names)
+                manifest = json.loads(
+                    zipfile.ZipFile(archive, "r").read("manifest.json").decode("utf-8")
+                )
+                self.assertEqual(manifest.get("trigger"), "manual")
+            finally:
+                if had and old is not None:
+                    hub_settings.write_text(old, encoding="utf-8")
+                elif not had and hub_settings.is_file():
+                    hub_settings.unlink(missing_ok=True)
+
+
+if __name__ == "__main__":
+    unittest.main()