|
| 1 | +#!/usr/bin/env python3 |
| 2 | +"""Missing HOME must not crash or select a root-level default authority.""" |
| 3 | +from pathlib import Path |
| 4 | +import subprocess |
| 5 | + |
| 6 | +ROOT = Path(__file__).resolve().parents[1] |
| 7 | +subprocess.run(['node', str(ROOT / 'scripts/require-docker.mjs')], check=True) |
| 8 | + |
| 9 | +import hashlib |
| 10 | +import json |
| 11 | +import os |
| 12 | +import tempfile |
| 13 | + |
| 14 | +import jsonschema |
| 15 | + |
| 16 | +CLI = str(ROOT / 'bin/git-locks') |
| 17 | +BASE = {k: v for k, v in os.environ.items() if not k.startswith('GIT_')} |
| 18 | +BASE.update(GIT_CONFIG_NOSYSTEM='1', GIT_CONFIG_GLOBAL='/dev/null') |
| 19 | +VALIDATOR = jsonschema.Draft202012Validator(json.loads((ROOT / 'schema/git-locks.schema.json').read_text())) |
| 20 | + |
| 21 | + |
| 22 | +def git(cwd, *args): |
| 23 | + return subprocess.check_output(['git', '-C', str(cwd), *args], env=BASE, |
| 24 | + stderr=subprocess.PIPE, text=True, timeout=10).strip() |
| 25 | + |
| 26 | + |
| 27 | +def inventory(base): |
| 28 | + return {str(p.relative_to(base)): hashlib.sha256(p.read_bytes()).hexdigest() |
| 29 | + for p in base.rglob('*') if p.is_file()} |
| 30 | + |
| 31 | + |
| 32 | +def run(cwd, env, *args): |
| 33 | + return subprocess.run([CLI, *args], cwd=cwd, env=env, capture_output=True, text=True, timeout=10) |
| 34 | + |
| 35 | + |
| 36 | +def exercise(base, layout, home, selector): |
| 37 | + subject = base / 'subject' |
| 38 | + subject.mkdir() |
| 39 | + cwd = subject |
| 40 | + if layout != 'outside': |
| 41 | + git(subject, 'init', '-q', '-b', 'main') |
| 42 | + if layout == 'linked': |
| 43 | + git(subject, '-c', 'user.name=test', '-c', 'user.email=test@example.invalid', |
| 44 | + 'commit', '--allow-empty', '-qm', 'fixture') |
| 45 | + cwd = base / 'linked' |
| 46 | + git(subject, 'worktree', 'add', '-q', '-b', 'linked', str(cwd)) |
| 47 | + # Enter a nested directory to also test the linked-worktree shared anchor. |
| 48 | + cwd = cwd / 'nested' |
| 49 | + cwd.mkdir() |
| 50 | + anchor = cwd if layout == 'outside' else subject |
| 51 | + env = dict(BASE) |
| 52 | + if home == 'unset': |
| 53 | + env.pop('HOME', None) |
| 54 | + elif home == 'empty': |
| 55 | + env['HOME'] = '' |
| 56 | + else: |
| 57 | + env['HOME'] = str(base / 'home') |
| 58 | + (base / 'home').mkdir() |
| 59 | + if selector == 'absolute': |
| 60 | + expected = base / 'explicit.git' |
| 61 | + env['GIT_LOCKS_STORE'] = str(expected) |
| 62 | + elif selector in ('relative', 'config'): |
| 63 | + expected = anchor / '.reservations/store.git' |
| 64 | + if selector == 'config': |
| 65 | + git(subject, 'config', 'locks.store', '.reservations/store.git') |
| 66 | + else: |
| 67 | + env['GIT_LOCKS_STORE'] = '.reservations/store.git' |
| 68 | + elif selector == 'self': |
| 69 | + env['GIT_LOCKS_STORE'] = 'self' |
| 70 | + expected = subject / '.git' |
| 71 | + elif selector == 'override': |
| 72 | + env['GIT_LOCKS_HOME'] = str(base / 'override') |
| 73 | + expected = base / 'override' / ('locks' + str(anchor)) |
| 74 | + else: |
| 75 | + if selector == 'empty-override': |
| 76 | + env['GIT_LOCKS_HOME'] = '' |
| 77 | + expected = base / 'home/.git-stunts' / ('locks' + str(anchor)) |
| 78 | + if home != 'valid' and selector in ('default', 'empty-override'): |
| 79 | + before = inventory(base) |
| 80 | + marker = base / 'command-ran' |
| 81 | + for args in (['store'], ['with', '--job', 'held', '--holder', 'alice', |
| 82 | + 'x.md', '--', 'touch', str(marker)]): |
| 83 | + result = run(cwd, env, *args) |
| 84 | + assert result.returncode == 2 and not result.stdout, result |
| 85 | + row = json.loads(result.stderr) |
| 86 | + VALIDATOR.validate(row) |
| 87 | + assert row['event'] == 'error' and row['reason'] == 'store-read', row |
| 88 | + assert 'HOME' in row['detail'], row |
| 89 | + assert inventory(base) == before, 'missing home modified the fixture' |
| 90 | + assert not marker.exists(), 'missing home launched a child' |
| 91 | + return |
| 92 | + selected = run(cwd, env, 'store') |
| 93 | + assert selected.returncode == 0 and not selected.stderr, selected |
| 94 | + row = json.loads(selected.stdout) |
| 95 | + VALIDATOR.validate(row) |
| 96 | + assert row['store'] == str(expected), (row, expected) |
| 97 | + claimed = run(cwd, env, 'claim', '--job', 'held', '--holder', 'alice', '--ttl', '300', 'x.md') |
| 98 | + assert claimed.returncode == 0 and not claimed.stderr, claimed |
| 99 | + VALIDATOR.validate(json.loads(claimed.stdout)) |
| 100 | + # Read the state from a second working directory with an explicit authority. |
| 101 | + second_cwd = subject if selector == 'self' else base |
| 102 | + second = run(second_cwd, dict(env, GIT_LOCKS_STORE=str(expected)), 'show', '--job', 'held') |
| 103 | + assert second.returncode == 0 and not second.stderr, second |
| 104 | + row = json.loads(second.stdout) |
| 105 | + VALIDATOR.validate(row) |
| 106 | + assert row['job'] == 'held' and row['paths'] == ['x.md'], row |
| 107 | + |
| 108 | + |
| 109 | +failures = [] |
| 110 | +checks = 0 |
| 111 | +for layout in ('outside', 'normal', 'linked'): |
| 112 | + selectors = ('default', 'empty-override', 'override', 'absolute', 'relative') |
| 113 | + if layout != 'outside': |
| 114 | + selectors += ('config', 'self') |
| 115 | + cases = [(home, selector) for home in ('unset', 'empty') for selector in selectors] |
| 116 | + cases += [('valid', 'default')] |
| 117 | + for home, selector in cases: |
| 118 | + with tempfile.TemporaryDirectory(prefix='locks-home-') as tmp: |
| 119 | + try: |
| 120 | + exercise(Path(tmp), layout, home, selector) |
| 121 | + checks += 1 |
| 122 | + print('PASS', layout, home, selector, flush=True) |
| 123 | + except Exception as error: |
| 124 | + failures.append((layout, home, selector, repr(error))) |
| 125 | + print('FAIL', *failures[-1], flush=True) |
| 126 | +print(f'home selection: {checks} passed; {len(failures)} failed') |
| 127 | +raise SystemExit(bool(failures)) |
0 commit comments