blob: 1e84934f4dedfb80e4e85b27cdacdccd1d21cbc3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
import collections
import json
import time
import os
import sys
SHUTDOWN_SIGNAL = 'SIGUSR1'
class SignalInterruptionError(Exception):
pass
def on_shutdown(s, f):
raise SignalInterruptionError()
def main():
args = sys.argv[1:]
def execve():
os.execve(args[0], args, os.environ)
jar_binary = args[args.index('--jar-binary') + 1]
java_bin_dir = os.path.dirname(jar_binary)
jstack_binary = os.path.join(java_bin_dir, 'jstack')
if not os.path.exists(jstack_binary):
sys.stderr.write("jstack is missing: {}\n".format(jstack_binary))
execve()
import signal
signum = getattr(signal, SHUTDOWN_SIGNAL, None)
if signum is None:
execve()
import subprocess
proc = subprocess.Popen(args)
signal.signal(signum, on_shutdown)
timeout = False
try:
proc.wait()
except SignalInterruptionError:
sys.stderr.write("\nGot {} signal: going to shutdown junit\n".format(signum))
# Dump stack traces
subprocess.call([jstack_binary, str(proc.pid)], stdout=sys.stderr)
# Kill junit - for more info see DEVTOOLS-7636
os.kill(proc.pid, signal.SIGKILL)
proc.wait()
timeout = True
if proc.returncode:
sys.stderr.write('java exit code: {}\n'.format(proc.returncode))
if timeout:
# In case of timeout return specific exit code
# https://a.yandex-team.ru/arc/trunk/arcadia/devtools/ya/test/const/__init__.py?rev=r8578188#L301
proc.returncode = 10
sys.stderr.write('java exit code changed to {}\n'.format(proc.returncode))
return proc.returncode
if __name__ == '__main__':
exit(main())
|
