# HG changeset patch
# User Matt Harbison <matt_harbison@yahoo.com>
# Date 2024-10-12 20:06:37
# Node ID 625cf96215516057db773637d4dbc2c346160727
# Parent  23cc79e04c286d0431fba6572321ecddedf7e04d

tests: add a module that can perform the equivalent of `SIGKILL` on any OS

I started with this being Windows specific, but let's push all of the decision
making into this function so that it can just be called by the tests.  The
tradeoff is that this is very specific to sending `SIGKILL`- since
`signal.SIGKILL` doesn't exist on Windows, the desired signal can't be passed
from the caller.  Maybe there's a way, but let's wait until there's a need.

We don't use `killdaemons.py` unconditionally because it starts with a more
graceful `SIGTERM` on posix.

diff --git a/mercurial/testing/ps_util.py b/mercurial/testing/ps_util.py
new file mode 100644
--- /dev/null
+++ b/mercurial/testing/ps_util.py
@@ -0,0 +1,50 @@
+# This python code can be imported into tests in order to terminate a process
+# with signal.SIGKILL on posix, or a roughly equivalent procedure on Windows.
+import os
+import signal
+import subprocess
+import sys
+import tempfile
+
+from .. import (
+    encoding,
+    pycompat,
+)
+
+from ..utils import procutil
+
+
+def kill_nt(pid: int, exit_code: int):
+    fd, pidfile = tempfile.mkstemp(
+        prefix=b"sigkill-", dir=encoding.environ[b"HGTMP"], text=False
+    )
+    try:
+        os.write(fd, b'%d\n' % pid)
+    finally:
+        os.close(fd)
+
+    env = dict(encoding.environ)
+    env[b"DAEMON_EXITCODE"] = b"%d" % exit_code
+
+    # Simulate the message written to stderr for this process on non-Windows
+    # platforms, for test consistency.
+    print("Killed!", file=sys.stderr)
+
+    subprocess.run(
+        [
+            encoding.environ[b"PYTHON"],
+            b"%s/killdaemons.py"
+            % encoding.environ[b'RUNTESTDIR_FORWARD_SLASH'],
+            pidfile,
+        ],
+        env=procutil.tonativeenv(env),
+    )
+
+
+def kill(pid: int):
+    """Kill the process with the given PID with SIGKILL or equivalent."""
+    if pycompat.iswindows:
+        exit_code = 128 + 9
+        kill_nt(pid, exit_code)
+    else:
+        os.kill(pid, signal.SIGKILL)