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

tests/test_tini_shim.py

@@ -0,0 +1,92 @@
+"""Unit tests for docker/tini-shim.sh argument stripping (#66679).
+
+These run without Docker: the shim's HERMES_TINI_SHIM_TARGET /
+HERMES_TINI_SHIM_WRAPPER hooks let us record the argv that would be
+handed to /init.
+"""
+from __future__ import annotations
+
+import os
+import subprocess
+from pathlib import Path
+
+import pytest
+
+REPO_ROOT = Path(__file__).resolve().parents[1]
+SHIM = REPO_ROOT / "docker" / "tini-shim.sh"
+
+
+@pytest.fixture
+def recorder(tmp_path: Path) -> tuple[Path, Path]:
+    """Fake /init + wrapper that print argv, one token per line."""
+    init = tmp_path / "fake-init"
+    wrapper = tmp_path / "fake-wrapper"
+    init.write_text("#!/bin/sh\nprintf '%s\\n' \"$@\"\n")
+    wrapper.write_text("#!/bin/sh\nprintf 'wrapper\\n'\n")
+    init.chmod(0o755)
+    wrapper.chmod(0o755)
+    return init, wrapper
+
+
+def _run_shim(
+    recorder: tuple[Path, Path],
+    args: list[str],
+) -> subprocess.CompletedProcess[str]:
+    init, wrapper = recorder
+    env = os.environ.copy()
+    env["HERMES_TINI_SHIM_TARGET"] = str(init)
+    env["HERMES_TINI_SHIM_WRAPPER"] = str(wrapper)
+    return subprocess.run(
+        ["sh", str(SHIM), *args],
+        capture_output=True,
+        text=True,
+        timeout=10,
+        env=env,
+        check=False,
+    )
+
+
+def test_shim_script_is_executable_bit_friendly() -> None:
+    assert SHIM.is_file()
+    text = SHIM.read_text()
+    assert text.startswith("#!/bin/sh")
+    assert "HERMES_TINI_SHIM_TARGET" in text
+
+
+def test_strips_g_and_double_dash(recorder: tuple[Path, Path]) -> None:
+    """Legacy `tini -g -- gateway run` must not forward `-g` to /init."""
+    r = _run_shim(recorder, ["-g", "--", "gateway", "run"])
+    assert r.returncode == 0, r.stderr
+    lines = [ln for ln in r.stdout.splitlines() if ln]
+    init, wrapper = recorder
+    assert lines[0] == str(wrapper)
+    assert lines[1:] == ["gateway", "run"]
+    assert "-g" not in lines
+    assert "--" not in lines
+
+
+
+
+
+
+def test_does_not_double_wrap_existing_wrapper(
+    recorder: tuple[Path, Path],
+) -> None:
+    _, wrapper = recorder
+    r = _run_shim(recorder, ["-g", "--", str(wrapper), "gateway", "run"])
+    assert r.returncode == 0, r.stderr
+    lines = [ln for ln in r.stdout.splitlines() if ln]
+    assert lines == [str(wrapper), "gateway", "run"]
+
+
+def test_strips_p_and_e_with_arguments(recorder: tuple[Path, Path]) -> None:
+    r = _run_shim(
+        recorder,
+        ["-p", "SIGKILL", "-e", "143", "-v", "-v", "--", "sleep", "infinity"],
+    )
+    assert r.returncode == 0, r.stderr
+    lines = [ln for ln in r.stdout.splitlines() if ln]
+    _, wrapper = recorder
+    assert lines == [str(wrapper), "sleep", "infinity"]
+    assert "SIGKILL" not in lines
+    assert "143" not in lines