mirror of
https://github.com/lucannez64/darling-lol.git
synced 2026-10-26 19:47:42 +00:00
56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
"""Check fixture traffic and private-prefix cleanup with a fake darling executable."""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import stat
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class URLSessionRunnerTests(unittest.TestCase):
|
|
def test_fixture_and_shutdown_use_only_a_private_prefix(self):
|
|
with tempfile.TemporaryDirectory() as folder:
|
|
root = Path(folder)
|
|
executable = root / 'darling'
|
|
trace = root / 'trace.jsonl'
|
|
executable.write_text(f'#!{sys.executable}\n' + f'TRACE = {str(trace)!r}\n' + '''
|
|
import json, os, sys, urllib.request
|
|
with open(TRACE, 'a') as stream:
|
|
stream.write(json.dumps({'args':sys.argv[1:], 'prefix':os.environ['DPREFIX']}) + '\\n')
|
|
if sys.argv[1] == 'shutdown':
|
|
raise SystemExit(0)
|
|
assert urllib.request.urlopen(sys.argv[2]).read() == b'darling-session-ok'
|
|
try:
|
|
urllib.request.urlopen(sys.argv[3], timeout=2)
|
|
except OSError:
|
|
pass
|
|
else:
|
|
raise AssertionError('reserved error endpoint unexpectedly accepted a connection')
|
|
print('LIFECYCLE_PASS: fixture')
|
|
print('URLSESSION_PASS: fixture')
|
|
''')
|
|
executable.chmod(0o755)
|
|
probe = root / 'probe'
|
|
probe.write_text('fake probe; never executed')
|
|
log = root / 'probe.log'
|
|
result = subprocess.run([sys.executable, str(ROOT / 'tools/riot-urlsession-check.py'),
|
|
'--probe', str(probe), '--log', str(log)],
|
|
env=dict(os.environ, PATH=str(root) + os.pathsep + os.environ['PATH']),
|
|
capture_output=True, text=True, timeout=15)
|
|
self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
|
|
calls = [json.loads(line) for line in trace.read_text().splitlines()]
|
|
self.assertEqual(len(calls), 2)
|
|
self.assertEqual(calls[1]['args'], ['shutdown'])
|
|
self.assertEqual(calls[0]['prefix'], calls[1]['prefix'])
|
|
self.assertNotEqual(Path(calls[0]['prefix']), Path.home() / '.darling')
|
|
self.assertFalse(Path(calls[0]['prefix']).exists())
|
|
self.assertEqual(stat.S_IMODE(log.stat().st_mode), 0o600)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|