fix(docker): discover Playwright headless_shell browser (NousResearch#35717)

Co-authored-by: Nic <nicsequenzy@gmail.com>

Author: polnikale

docker/stage2-hook.sh

@@ -309,17 +309,19 @@ fi
 #   shared libraries (libGLESv2.so, libEGL.so, ...) which inherit the
 #   executable bit from Playwright's tarball but are NOT browser binaries.
 #   We only accept files whose basename is chrome / chromium /
-#   chrome-headless-shell / chromium-browser. Compare PR #18635's earlier
-#   ``find | grep -Ei 'chrome|chromium'`` which would match the path
-#   ``.../chrome-headless-shell-linux64/libGLESv2.so`` and pick a .so.
+#   chrome-headless-shell / headless_shell / chromium-browser. Compare
+#   PR #18635's earlier ``find | grep -Ei 'chrome|chromium'`` which would
+#   match the path ``.../chrome-headless-shell-linux64/libGLESv2.so`` and
+#   pick a .so.
 # - Quietly skipped when $PLAYWRIGHT_BROWSERS_PATH doesn't exist (e.g.
 #   custom builds that strip Playwright).
 if [ -z "${AGENT_BROWSER_EXECUTABLE_PATH:-}" ] && \
         [ -n "${PLAYWRIGHT_BROWSERS_PATH:-}" ] && \
         [ -d "$PLAYWRIGHT_BROWSERS_PATH" ]; then
     browser_bin=$(find "$PLAYWRIGHT_BROWSERS_PATH" -type f -executable \
         \( -name 'chrome' -o -name 'chromium' \
-           -o -name 'chrome-headless-shell' -o -name 'chromium-browser' \) \
+           -o -name 'chrome-headless-shell' -o -name 'headless_shell' \
+           -o -name 'chromium-browser' \) \
         2>/dev/null | head -n 1)
     if [ -n "$browser_bin" ]; then
         echo "[stage2] Found agent-browser Chromium binary: $browser_bin"

tests/test_docker_stage2_browser_discovery.py

@@ -0,0 +1,19 @@
+"""Regression tests for Docker stage2 browser executable discovery."""
+
+from pathlib import Path
+
+
+def test_stage2_discovers_playwright_arm64_headless_shell() -> None:
+    """Playwright's --only-shell layout may use a headless_shell basename."""
+    script = Path("docker/stage2-hook.sh").read_text()
+
+    assert "-name 'headless_shell'" in script
+
+
+def test_stage2_discovery_stays_filename_matched() -> None:
+    """Avoid broad path grep that can pick executable shared libraries."""
+    script = Path("docker/stage2-hook.sh").read_text()
+
+    discovery_block = script.split("browser_bin=$(", 1)[1].split(")\n    if", 1)[0]
+    assert "find \"$PLAYWRIGHT_BROWSERS_PATH\" -type f -executable" in discovery_block
+    assert "grep" not in discovery_block