mirror of
https://github.com/lucannez64/darling-lol.git
synced 2026-10-26 19:47:42 +00:00
lucannez64
e07cf72a15
Ignore both invalidation methods on the shared session; deliver task completion through the owning session delegate queue. Extend the standalone probe to exercise singleton reuse and a suspended custom queue. Require stable old-client process identities and complete window data before optional recovery, and continue launch after shutdown failures.
58 lines
2.3 KiB
Python
58 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')
|
|
print('SHARED_SESSION_PASS: fixture')
|
|
print('QUEUE_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()
|