test-worker.t
130 lines
| 3.1 KiB
| text/troff
|
Tads3Lexer
/ tests / test-worker.t
|
r31696 | Test UI worker interaction | ||
$ cat > t.py <<EOF | ||||
> from __future__ import absolute_import, print_function | ||||
|
r32114 | > import time | ||
|
r31696 | > from mercurial import ( | ||
|
r32041 | > error, | ||
|
r32337 | > registrar, | ||
|
r31696 | > ui as uimod, | ||
> worker, | ||||
> ) | ||||
|
r32041 | > def abort(ui, args): | ||
> if args[0] == 0: | ||||
> # by first worker for test stability | ||||
|
r36200 | > raise error.Abort(b'known exception') | ||
|
r32041 | > return runme(ui, []) | ||
|
r32043 | > def exc(ui, args): | ||
> if args[0] == 0: | ||||
> # by first worker for test stability | ||||
> raise Exception('unknown exception') | ||||
> return runme(ui, []) | ||||
|
r31696 | > def runme(ui, args): | ||
> for arg in args: | ||||
|
r36200 | > ui.status(b'run\n') | ||
|
r31696 | > yield 1, arg | ||
|
r32114 | > time.sleep(0.1) # easier to trigger killworkers code path | ||
|
r32041 | > functable = { | ||
|
r36200 | > b'abort': abort, | ||
> b'exc': exc, | ||||
> b'runme': runme, | ||||
|
r32041 | > } | ||
|
r31696 | > cmdtable = {} | ||
|
r32337 | > command = registrar.command(cmdtable) | ||
|
r36200 | > @command(b'test', [], b'hg test [COST] [FUNC]') | ||
> def t(ui, repo, cost=1.0, func=b'runme'): | ||||
|
r31696 | > cost = float(cost) | ||
|
r32041 | > func = functable[func] | ||
|
r36200 | > ui.status(b'start\n') | ||
|
r32041 | > runs = worker.worker(ui, cost, func, (ui,), range(8)) | ||
|
r31696 | > for n, i in runs: | ||
> pass | ||||
|
r36200 | > ui.status(b'done\n') | ||
|
r31696 | > EOF | ||
$ abspath=`pwd`/t.py | ||||
$ hg init | ||||
Run tests with worker enable by forcing a heigh cost | ||||
$ hg --config "extensions.t=$abspath" test 100000.0 | ||||
start | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
done | ||||
Run tests without worker by forcing a low cost | ||||
$ hg --config "extensions.t=$abspath" test 0.0000001 | ||||
start | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
run | ||||
done | ||||
|
r32041 | |||
|
r32061 | #if no-windows | ||
|
r32041 | Known exception should be caught, but printed if --traceback is enabled | ||
|
r32114 | $ hg --config "extensions.t=$abspath" --config 'worker.numcpus=8' \ | ||
> test 100000.0 abort 2>&1 | ||||
|
r32041 | start | ||
abort: known exception | ||||
|
r32042 | [255] | ||
|
r32041 | |||
|
r32114 | $ hg --config "extensions.t=$abspath" --config 'worker.numcpus=8' \ | ||
|
r41662 | > test 100000.0 abort --traceback 2>&1 | egrep '(SystemExit|Abort)' | ||
raise error.Abort(b'known exception') | ||||
mercurial.error.Abort: b'known exception' (py3 !) | ||||
Abort: known exception (no-py3 !) | ||||
|
r32113 | SystemExit: 255 | ||
|
r32043 | |||
Traceback must be printed for unknown exceptions | ||||
|
r32114 | $ hg --config "extensions.t=$abspath" --config 'worker.numcpus=8' \ | ||
|
r32113 | > test 100000.0 exc 2>&1 | grep '^Exception' | ||
Exception: unknown exception | ||||
|
r32061 | |||
|
r32112 | Workers should not do cleanups in all cases | ||
$ cat > $TESTTMP/detectcleanup.py <<EOF | ||||
> from __future__ import absolute_import | ||||
> import atexit | ||||
> import os | ||||
> import time | ||||
> oldfork = os.fork | ||||
> count = 0 | ||||
> parentpid = os.getpid() | ||||
> def delayedfork(): | ||||
> global count | ||||
> count += 1 | ||||
> pid = oldfork() | ||||
> # make it easier to test SIGTERM hitting other workers when they have | ||||
> # not set up error handling yet. | ||||
> if count > 1 and pid == 0: | ||||
> time.sleep(0.1) | ||||
> return pid | ||||
> os.fork = delayedfork | ||||
> def cleanup(): | ||||
> if os.getpid() != parentpid: | ||||
> os.write(1, 'should never happen\n') | ||||
> atexit.register(cleanup) | ||||
> EOF | ||||
$ hg --config "extensions.t=$abspath" --config worker.numcpus=8 --config \ | ||||
> "extensions.d=$TESTTMP/detectcleanup.py" test 100000 abort | ||||
start | ||||
abort: known exception | ||||
[255] | ||||
|
r32061 | #endif | ||