Интегрирован wesp в сайт

WESP_REL/tests/test_post_update_script.py

@@ -0,0 +1,103 @@
+"""Tests for scripts/post_update.sh."""
+
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import sys
+import tempfile
+import unittest
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parent.parent
+
+
+def _bash_executable() -> str:
+    if os.name == "nt":
+        for candidate in (
+            Path(r"C:\Program Files\Git\bin\bash.exe"),
+            Path(r"C:\Program Files (x86)\Git\bin\bash.exe"),
+        ):
+            if candidate.is_file():
+                return str(candidate)
+    return "bash"
+
+
+def _bash_path_style() -> str:
+    """Git Bash: /c/...; WSL: /mnt/c/..."""
+    probe = subprocess.run(
+        [_bash_executable(), "-c", 'if [ -d /mnt/c ]; then echo wsl; else echo gitbash; fi'],
+        capture_output=True,
+        text=True,
+        cwd=str(ROOT),
+    )
+    return (probe.stdout or "gitbash").strip()
+
+
+def _bash_python_executable() -> str:
+    """Путь к python для bash (WSL: /mnt/c/..., Git Bash: /c/...)."""
+    exe = Path(sys.executable).resolve()
+    if os.name != "nt":
+        return str(exe)
+    drive = exe.drive.rstrip(":").lower()
+    tail = str(exe).replace(exe.drive, "").replace("\\", "/")
+    if _bash_path_style() == "wsl":
+        return f"/mnt/{drive}{tail}"
+    return f"/{drive}{tail}"
+
+
+class PostUpdateScriptTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self._tmp = tempfile.mkdtemp(prefix="wesp-post-")
+        self.base = Path(self._tmp)
+        data = self.base / "data"
+        data.mkdir(parents=True)
+        (data / "update_state.json").write_text(
+            json.dumps(
+                {
+                    "status": "pending_restart",
+                    "target_version": "2.0.0",
+                    "previous_version": "1.0.0",
+                    "backup_path": str(self.base / "backups" / "b1"),
+                }
+            ),
+            encoding="utf-8",
+        )
+        (data / ".pending_restart.json").write_text(
+            json.dumps({"backup_path": str(self.base / "backups" / "b1"), "previous_version": "1.0.0"}),
+            encoding="utf-8",
+        )
+
+    def tearDown(self) -> None:
+        import shutil
+
+        shutil.rmtree(self._tmp, ignore_errors=True)
+
+    def _run_post(self, env: dict) -> subprocess.CompletedProcess:
+        e = os.environ.copy()
+        e.update(env)
+        # Run from real ROOT so scripts exist; point data via copying is hard — use WESP test root via symlink approach
+        # Simpler: invoke health cmd only path
+        cmd = [_bash_executable(), str(ROOT / "scripts" / "post_update.sh")]
+        return subprocess.run(cmd, capture_output=True, text=True, env=e, cwd=str(ROOT))
+
+    def test_health_success_exits_zero(self) -> None:
+        py = _bash_python_executable()
+        # Inline env: WSL/Git Bash may not inherit Windows subprocess env reliably.
+        cmd = (
+            f'WESP_SKIP_RESTART=1 WESP_HEALTH_CMD="exit 0" '
+            f'WESP_BASE="{ROOT.as_posix()}" WESP_PYTHON="{py}" '
+            f"bash scripts/post_update.sh"
+        )
+        r = subprocess.run(
+            [_bash_executable(), "-c", cmd],
+            capture_output=True,
+            text=True,
+            cwd=str(ROOT),
+        )
+        self.assertEqual(r.returncode, 0, r.stderr or r.stdout)
+
+
+if __name__ == "__main__":
+    unittest.main()