162 lines
8.0 KiB
Python
162 lines
8.0 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from src.web_process_backend import WebProcessBackend, build_pm2_start_command
|
|
|
|
|
|
class WebProcessBackendTests(unittest.TestCase):
|
|
def test_build_pm2_start_command_for_python_script_uses_repo_context(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
script = base_dir / "youtube.py"
|
|
venv_python = base_dir / ".venv" / "bin" / "python"
|
|
venv_python.parent.mkdir(parents=True, exist_ok=True)
|
|
venv_python.write_text("#!/usr/bin/env python3\n", encoding="utf-8")
|
|
script.write_text("print('ok')\n", encoding="utf-8")
|
|
|
|
cmd = build_pm2_start_command(base_dir, "youtube2__lane1", "youtube.py")
|
|
|
|
self.assertIsNotNone(cmd)
|
|
assert cmd is not None
|
|
self.assertIn("LIVE_PM2_PROCESS_NAME=youtube2__lane1", cmd)
|
|
self.assertIn("--name youtube2__lane1", cmd)
|
|
self.assertIn("--cwd", cmd)
|
|
self.assertIn(str(script), cmd)
|
|
self.assertIn(str(venv_python), cmd)
|
|
|
|
def test_parse_pm2_describe_marks_missing_jlist_process_as_not_found(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend.process_infos = AsyncMock(return_value={"youtube": {"process_status": "not_found", "out_path": None, "err_path": None, "pid": 0}}) # type: ignore[method-assign]
|
|
|
|
info = asyncio.run(backend.parse_pm2_describe("youtube"))
|
|
|
|
self.assertEqual(info["process_status"], "not_found")
|
|
self.assertIsNone(info["out_path"])
|
|
self.assertIsNone(info["err_path"])
|
|
self.assertEqual(info["pid"], 0)
|
|
|
|
def test_parse_pm2_describe_reads_status_and_log_paths_from_jlist_item(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend.process_infos = AsyncMock( # type: ignore[method-assign]
|
|
return_value={
|
|
"youtube": {
|
|
"process_status": "online",
|
|
"out_path": "/home/live/.pm2/logs/youtube-out.log",
|
|
"err_path": "/home/live/.pm2/logs/youtube-error.log",
|
|
"pid": 4567,
|
|
}
|
|
}
|
|
)
|
|
|
|
info = asyncio.run(backend.parse_pm2_describe("youtube"))
|
|
|
|
self.assertEqual(info["process_status"], "online")
|
|
self.assertEqual(info["out_path"], "/home/live/.pm2/logs/youtube-out.log")
|
|
self.assertEqual(info["err_path"], "/home/live/.pm2/logs/youtube-error.log")
|
|
self.assertEqual(info["pid"], 4567)
|
|
|
|
def test_pm2_jlist_items_uses_short_ttl_cache(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
shell_out = AsyncMock(return_value=(0, '[{"name":"youtube","pm2_env":{"status":"online"}}]'))
|
|
with patch("src.web_process_backend.resolve_pm2_bin", return_value="/usr/bin/pm2"):
|
|
with patch("src.web_process_backend._shell_out", new=shell_out):
|
|
first = asyncio.run(backend._pm2_jlist_items())
|
|
second = asyncio.run(backend._pm2_jlist_items())
|
|
|
|
self.assertEqual(first[0]["name"], "youtube")
|
|
self.assertEqual(second[0]["name"], "youtube")
|
|
shell_out.assert_awaited_once()
|
|
|
|
def test_ensure_pm2_runtime_resurrects_pm2_when_dump_exists_and_jlist_is_empty(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
pm2_home = base_dir / ".pm2-home"
|
|
pm2_home.mkdir(parents=True, exist_ok=True)
|
|
(pm2_home / "pm2.pid").write_text("1234\n", encoding="utf-8")
|
|
(pm2_home / "dump.pm2").write_text('[{"name":"youtube"}]\n', encoding="utf-8")
|
|
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend._use_pm2 = True
|
|
|
|
shell_out = AsyncMock(return_value=(0, "[]"))
|
|
resurrect = AsyncMock(return_value=(0, "ok"))
|
|
with patch.dict("os.environ", {"PM2_HOME": str(pm2_home)}, clear=False):
|
|
with patch("src.web_process_backend.resolve_pm2_bin", return_value="/usr/bin/pm2"):
|
|
with patch("src.web_process_backend._shell_out", new=shell_out):
|
|
with patch("src.web_process_backend.run_pm2_result", new=resurrect):
|
|
asyncio.run(backend.ensure_pm2_runtime())
|
|
|
|
resurrect.assert_awaited_once_with("resurrect", timeout=45.0)
|
|
|
|
def test_ensure_pm2_runtime_skips_resurrect_without_saved_dump(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
pm2_home = base_dir / ".pm2-home"
|
|
pm2_home.mkdir(parents=True, exist_ok=True)
|
|
(pm2_home / "pm2.pid").write_text("1234\n", encoding="utf-8")
|
|
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend._use_pm2 = True
|
|
|
|
shell_out = AsyncMock(return_value=(0, "[]"))
|
|
resurrect = AsyncMock(return_value=(0, "ok"))
|
|
with patch.dict("os.environ", {"PM2_HOME": str(pm2_home)}, clear=False):
|
|
with patch("src.web_process_backend.resolve_pm2_bin", return_value="/usr/bin/pm2"):
|
|
with patch("src.web_process_backend._shell_out", new=shell_out):
|
|
with patch("src.web_process_backend.run_pm2_result", new=resurrect):
|
|
asyncio.run(backend.ensure_pm2_runtime())
|
|
|
|
resurrect.assert_not_awaited()
|
|
|
|
def test_start_does_not_mark_desired_running_when_shell_command_fails(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
script = base_dir / "youtube.py"
|
|
script.write_text("print('ok')\n", encoding="utf-8")
|
|
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend._use_pm2 = True
|
|
|
|
with patch.object(backend, "ensure_mode", AsyncMock()):
|
|
with patch.object(backend, "_pm2_registered_names", AsyncMock(return_value=set())):
|
|
with patch.object(backend, "clear_process_logs", AsyncMock()):
|
|
with patch("src.web_process_backend.build_pm2_start_command", return_value="pm2 start ..."):
|
|
with patch("src.web_process_backend._shell_out", new=AsyncMock(return_value=(1, "boom"))):
|
|
with patch.object(backend, "_set_pm2_desired_running", AsyncMock()) as set_desired:
|
|
with patch.object(backend, "_pm2_save", AsyncMock()):
|
|
text = asyncio.run(backend.start("youtube", "youtube.py"))
|
|
|
|
self.assertEqual(text, "boom")
|
|
set_desired.assert_not_awaited()
|
|
|
|
def test_live_ffmpeg_pids_uses_process_tree_of_target_process(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
base_dir = Path(tmp)
|
|
backend = WebProcessBackend(base_dir, base_dir / ".pm2" / "logs")
|
|
backend.ensure_mode = AsyncMock()
|
|
backend.process_infos = AsyncMock( # type: ignore[method-assign]
|
|
return_value={"youtube": {"pid": 4321, "process_status": "online"}}
|
|
)
|
|
|
|
with patch("src.web_process_backend.os.getpgid", return_value=6789):
|
|
with patch("src.web_process_backend._linux_ffmpeg_pids_for_process_tree", return_value=[9876]) as finder:
|
|
pids = asyncio.run(backend.live_ffmpeg_pids("youtube"))
|
|
|
|
self.assertEqual(pids, [9876])
|
|
finder.assert_called_once_with(4321, root_pgid=6789)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|