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

tests/tools/test_process_registry_write_stdin_surrogates.py

@@ -0,0 +1,54 @@
+"""Sibling regression test for #79178: background-PTY stdin must round-trip
+surrogateescape content instead of crashing on the strict UTF-8 encode."""
+import shlex
+import time
+
+import pytest
+
+from tools.process_registry import ProcessRegistry
+
+
+def test_write_stdin_pty_surrogateescape_roundtrip(tmp_path):
+    registry = ProcessRegistry()
+    out = tmp_path / "out.bin"
+    script = tmp_path / "read_stdin.py"
+    # readline(): a PTY never delivers EOF, so read one line (canonical mode
+    # delivers it after the newline we send).
+    script.write_text(
+        f"import sys\nopen({str(out)!r}, 'wb').write(sys.stdin.buffer.readline())\n"
+    )
+    session = registry.spawn_local(
+        f"python3 {shlex.quote(str(script))}",
+        cwd=str(tmp_path),
+        use_pty=True,
+    )
+    if session._pty is None:
+        registry.kill_process(session.id)
+        pytest.skip("ptyprocess not available; PTY path not exercised")
+    try:
+        result = registry.write_stdin(
+            session.id, b"\xff".decode("utf-8", "surrogateescape") + "\n"
+        )
+        assert result["status"] == "ok", result
+        # Wait for the CONTENT, and not for the file to exist. The child runs
+        # open(out,'wb').write(...). open() creates the file empty, and the
+        # bytes arrive only after the PTY delivers the line. The previous wait
+        # stopped at out.exists(), which the empty file already satisfies, so
+        # the read returned b'' when the parent won that gap.
+        #
+        # On a 144-worker runner the gap is wide enough to lose every time.
+        # This test failed both attempts in CI, and not one time only. It also
+        # loses 6 times in 25 runs on an idle 16-core machine.
+        deadline = time.monotonic() + 30
+        got = b""
+        while time.monotonic() < deadline:
+            try:
+                got = out.read_bytes()
+            except FileNotFoundError:
+                got = b""
+            if got == b"\xff\n":
+                break
+            time.sleep(0.05)
+        assert got == b"\xff\n"
+    finally:
+        registry.kill_process(session.id)