dekun
bfa3352122
支持手动/每日自动备份四所数据库、K线库与 env，上传 zip 一键恢复；中控默认账号 admin/admin123。 Co-authored-by: Cursor <cursoragent@cursor.com>
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
"""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()
|