mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
4b2dda3c12
Print an error message and exit with a non-zero code when we hit the timeout. This makes it clear when a test fails due to a timeout. On Darwin, run `sample` on the target process first, so that the failure includes some information about what the test was doing when the timeout occurred.
31 lines
751 B
Python
31 lines
751 B
Python
#!/uar/bin/env python3
|
|
|
|
import platform
|
|
import subprocess
|
|
import sys
|
|
|
|
sampleCommand = None
|
|
timeoutSampleTime = 0
|
|
|
|
if platform.system() == 'Darwin':
|
|
sampleCommand = '/usr/bin/sample'
|
|
timeoutSampleTime = 10
|
|
|
|
|
|
def watchdog(command, timeout):
|
|
process = subprocess.Popen(command)
|
|
try:
|
|
process.wait(timeout=timeout)
|
|
except subprocess.TimeoutExpired:
|
|
if sampleCommand:
|
|
pidstr = str(process.pid)
|
|
subprocess.run([sampleCommand, pidstr, str(timeoutSampleTime)])
|
|
process.kill()
|
|
sys.exit(
|
|
'error: command timed out after {} seconds: {}'
|
|
.format(timeout, ' '.join(sys.argv[2:])))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
watchdog(sys.argv[2:], timeout=float(sys.argv[1]))
|