upstream/mercurial-mirror Files · mercurial/testing/ps_util.py

rust-pyo3: dagop submodule implementation...

rust-pyo3: dagop submodule implementation This is the first demonstration that the passing of objects through the Python interpreter from `rustext` to `pyo3_rustext` actually works. In the Python tests, we conflate the presence of the `pyo3_rustext` package with that of `rustext`, as we do not plan to support building one and not the other (we hope to convert fully to PyO3 soon). The skipping is actually done by the base test class. The implementation of `rank` is as trivial as it was in `hg-cpython`, yet it required to be able to convert exceptions from `vcsgraph`.

Matt Harbison - - Load All Authors

File last commit:

r53247:d4e30c15 default


                r53310:20fe0bf9

default

Download file

             ps_util.py
        
                    53 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / mercurial / testing / ps_util.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # 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.

      from __future__ import annotations

      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)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# 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.

				from __future__ import annotations

				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)