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()