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

tests/hermes_cli/test_spawn_gateway_restart_reap.py

@@ -0,0 +1,67 @@
+"""Tests for _spawn_gateway_restart orphan-reap guard (#77276)."""
+from __future__ import annotations
+
+import subprocess
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def reset_restart_cooldown():
+    """Clear the #89034 repeat-restart cooldown between cases.
+
+    ``_spawn_gateway_restart`` now coalesces a second restart request that
+    arrives within ``GATEWAY_RESTART_COOLDOWN_SECONDS`` of the last spawn, so
+    without this the first case's spawn suppresses the second case's.
+    """
+    import hermes_cli.web_server as web_server
+
+    web_server._LAST_GATEWAY_RESTART = None
+    yield
+    web_server._LAST_GATEWAY_RESTART = None
+
+
+class TestSpawnGatewayRestartReapsOrphans:
+    """_spawn_gateway_restart must reap orphaned gateways before spawning."""
+
+    @patch("hermes_cli.web_server._gateway_subcommand", return_value=["gateway", "restart"])
+    @patch("hermes_cli.web_server._spawn_hermes_action")
+    @patch("hermes_cli.web_server._ACTION_PROCS", {})
+    def test_reap_called_before_spawn(self, mock_spawn, mock_subcmd):
+        """Orphan reap runs before the new gateway process is spawned."""
+        mock_proc = MagicMock(spec=subprocess.Popen)
+        mock_proc.poll.return_value = None
+        mock_spawn.return_value = mock_proc
+
+        from hermes_cli.web_server import _spawn_gateway_restart
+
+        with patch(
+            "hermes_cli.gateway._reap_unsupervised_gateway_orphans"
+        ) as mock_reap:
+            proc, reused = _spawn_gateway_restart()
+
+        mock_reap.assert_called_once()
+        mock_spawn.assert_called_once()
+        assert proc is mock_proc
+        assert reused is False
+
+    @patch("hermes_cli.web_server._gateway_subcommand", return_value=["gateway", "restart"])
+    @patch("hermes_cli.web_server._spawn_hermes_action")
+    @patch("hermes_cli.web_server._ACTION_PROCS", {})
+    def test_reap_failure_does_not_block_spawn(self, mock_spawn, mock_subcmd):
+        """If reap raises, the restart still proceeds."""
+        mock_proc = MagicMock(spec=subprocess.Popen)
+        mock_proc.poll.return_value = None
+        mock_spawn.return_value = mock_proc
+
+        from hermes_cli.web_server import _spawn_gateway_restart
+
+        with patch(
+            "hermes_cli.gateway._reap_unsupervised_gateway_orphans",
+            side_effect=OSError("permission denied"),
+        ):
+            proc, reused = _spawn_gateway_restart()
+
+        mock_spawn.assert_called_once()
+        assert proc is mock_proc