Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/gateway/test_gateway_process_exit.py

@@ -0,0 +1,91 @@
+from types import SimpleNamespace
+from unittest.mock import Mock
+
+import pytest
+
+import gateway.run as gateway_run
+
+
+class _ExitCalled(Exception):
+    def __init__(self, code: int):
+        super().__init__(code)
+        self.code = code
+
+
+def _raise_exit(code: int) -> None:
+    raise _ExitCalled(code)
+
+
+def test_main_terminates_via_os_exit_not_systemexit(monkeypatch):
+    """The terminating call must be os._exit, NOT sys.exit — SystemExit is
+    exactly what triggers the Py_FinalizeEx non-daemon-thread join hang this
+    fixes (#53107). If main() ever regresses to sys.exit(), SystemExit would
+    propagate instead of our os._exit sentinel and this test would fail.
+
+    Test contributed by @AgenticSpark (PR #53122, duplicate of #53121)."""
+    async def fake_start_gateway(config=None):
+        return False
+
+    stdout = SimpleNamespace(flush=Mock())
+    stderr = SimpleNamespace(flush=Mock())
+
+    monkeypatch.setattr(gateway_run, "start_gateway", fake_start_gateway)
+    monkeypatch.setattr(gateway_run.os, "_exit", _raise_exit)
+    monkeypatch.setattr(gateway_run.sys, "argv", ["gateway.run"])
+    monkeypatch.setattr(gateway_run.sys, "stdout", stdout)
+    monkeypatch.setattr(gateway_run.sys, "stderr", stderr)
+
+    # Our os._exit sentinel must be what terminates main() — not SystemExit.
+    with pytest.raises(_ExitCalled):
+        gateway_run.main()
+
+
+def test_main_routes_systemexit_through_os_exit(monkeypatch):
+    """start_gateway raises SystemExit on the clean-fatal-config (#51228),
+    planned-restart, and service-restart paths. main() must catch it and route
+    the carried code through os._exit too, so those paths are equally wedge-proof
+    (#53107) — a SystemExit propagating to interpreter finalization would join a
+    stuck non-daemon worker and hang. Verifies the explicit code (e.g. 78) is
+    preserved through the os._exit backstop."""
+    async def fake_start_gateway(config=None):
+        raise SystemExit(78)
+
+    stdout = SimpleNamespace(flush=Mock())
+    stderr = SimpleNamespace(flush=Mock())
+
+    monkeypatch.setattr(gateway_run, "start_gateway", fake_start_gateway)
+    monkeypatch.setattr(gateway_run.os, "_exit", _raise_exit)
+    monkeypatch.setattr(gateway_run.sys, "argv", ["gateway.run"])
+    monkeypatch.setattr(gateway_run.sys, "stdout", stdout)
+    monkeypatch.setattr(gateway_run.sys, "stderr", stderr)
+
+    with pytest.raises(_ExitCalled) as exc_info:
+        gateway_run.main()
+
+    # The SystemExit(78) must be converted to os._exit(78), not propagated.
+    assert exc_info.value.code == 78
+    stdout.flush.assert_called_once_with()
+    stderr.flush.assert_called_once_with()
+
+
+def test_exit_backstop_releases_pid_file_and_runtime_lock(monkeypatch):
+    """os._exit bypasses atexit, and the early SystemExit exit paths never run
+    _stop_impl — so the force-exit backstop itself must release the PID file and
+    runtime lock, or those early paths (#51228 fatal-config) would leak them.
+    Both releases are idempotent, so this is safe on every exit path."""
+    from gateway import status as gateway_status
+
+    remove_pid = Mock()
+    release_lock = Mock()
+    monkeypatch.setattr(gateway_status, "remove_pid_file", remove_pid)
+    monkeypatch.setattr(gateway_status, "release_gateway_runtime_lock", release_lock)
+    monkeypatch.setattr(gateway_run.os, "_exit", _raise_exit)
+    monkeypatch.setattr(gateway_run.sys, "stdout", SimpleNamespace(flush=Mock()))
+    monkeypatch.setattr(gateway_run.sys, "stderr", SimpleNamespace(flush=Mock()))
+
+    with pytest.raises(_ExitCalled) as exc_info:
+        gateway_run._exit_after_graceful_shutdown(78)
+
+    assert exc_info.value.code == 78
+    remove_pid.assert_called_once_with()
+    release_lock.assert_called_once_with()