#!/usr/bin/env python3
# verify actual threads, independent vm state, parallelism, cancellation and join.
from pathlib import Path
import argparse
import subprocess

parser = argparse.ArgumentParser()
parser.add_argument('vm', nargs='?', default='bin/zy')
parser.add_argument('--case', default='')
parser.add_argument('--library', default='')
parser.add_argument('--backend', choices=('native', 'micro'), default='native')
args = parser.parse_args()
root = Path(__file__).resolve().parents[2]
expected = {'isolation': '21 16 7 16 7 4 1 1\n', 'parallel': '11 21\n', 'cancel': '0\n', 'cancel_o3': '0\n', 'invalid': '-1 -1\n', 'recycle': '66\n', 'channels': '-1 0\n', 'failed_worker': '5 7\n', 'ffi_o3': '1 5050 0\n', 'actor_identity': '1\n', 'resources': '1 24 0 0 0\n', 'close_order': '0 0 -1\n', 'resources_counts': '24 0 0 0\n', 'nested': '1 0 0 0\n'}
count = 0
for case in sorted((root / 'tests/threads').glob('*/*.zc')):
    if args.case and args.case not in str(case):
        continue
    if args.backend == 'micro' and case.stem in ('ffi_o3', 'resources'):
        print(f'skip {case.relative_to(root)} (native ffi fixture)')
        continue
    source = case.read_text()
    if args.library:
        source = source.replace('include rt, actor;', f'include rt; include \"{args.library}\"; include actor;')
    proc = subprocess.run([str(root / args.vm)], input=source, text=True, capture_output=True, cwd=root, timeout=12)
    actual = proc.stdout
    if case.stem in ('parallel', 'cancel', 'cancel_o3'):
        lines = actual.splitlines(keepends=True)
        assert len(lines) == 2, f'{case}: invalid output {actual!r}, stderr={proc.stderr!r}'
        milliseconds = int(lines[-1])
        assert 0 <= milliseconds < (580 if case.stem == 'parallel' else 500), f'{case}: elapsed {milliseconds}ms'
        actual = lines[0]
    assert proc.returncode == 0 and actual == expected[case.stem], f'{case}: exit={proc.returncode}, stdout={proc.stdout!r}, stderr={proc.stderr!r}'
    print(f'ok   {case.relative_to(root)}')
    count += 1
assert count
print(f'{count} thread context fixtures passed')