51 lines
2.2 KiB
Python
51 lines
2.2 KiB
Python
"""通过受限 systemd 临时服务执行远程测试，不复用生产服务进程。"""
|
|
|
|
import argparse
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import time
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('directory')
|
|
parser.add_argument('--only', action='append', default=[])
|
|
parser.add_argument('--run', default='')
|
|
parser.add_argument('--per-test', action='store_true')
|
|
parser.add_argument('--memory-mib', type=int, choices=(64, 96, 128, 192), default=128)
|
|
args = parser.parse_args()
|
|
root = Path(args.directory).resolve(strict=True)
|
|
assert os.geteuid() == 0
|
|
assert root.parent == Path('/var/tmp') and root.name.startswith('smail-review-')
|
|
memory = dict(line.split(':', 1) for line in Path('/proc/meminfo').read_text().splitlines())
|
|
assert int(memory['MemAvailable'].split()[0]) >= (args.memory_mib + 128) * 1024, '可用内存不足，停止增加测试负载'
|
|
unit = 'smail-tests-' + time.strftime('%Y%m%d-%H%M%S')
|
|
command = [
|
|
'systemd-run', '--wait', '--pipe', '--unit=' + unit,
|
|
'--property=User=smail', '--property=Group=smail',
|
|
'--property=MemoryMax=' + str(args.memory_mib) + 'M', '--property=MemorySwapMax=0',
|
|
'--property=TasksMax=128', '--property=CPUQuota=25%',
|
|
'--property=RuntimeMaxSec=1800s', '--property=TimeoutStopSec=10s',
|
|
'--property=KillMode=control-group', '--property=OOMPolicy=stop',
|
|
'--property=PrivateTmp=yes', '--property=PrivateNetwork=yes',
|
|
'--property=BindPaths=' + str(root),
|
|
'--property=ProtectSystem=strict', '--property=ProtectHome=yes',
|
|
'--property=InaccessiblePaths=/opt/smail /run/smail',
|
|
'--property=NoNewPrivileges=yes', '--property=ReadWritePaths=' + str(root),
|
|
'--property=WorkingDirectory=' + str(root),
|
|
'python3', str(root / 'run_review_tests.py'),
|
|
]
|
|
for directory in args.only:
|
|
command.extend(['--only', directory])
|
|
if args.run:
|
|
command.extend(['--run', args.run])
|
|
if args.per_test:
|
|
command.append('--per-test')
|
|
print('测试服务：' + unit, flush=True)
|
|
return subprocess.run(command).returncode
|
|
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main())
|