1 files changed, 37 insertions, 0 deletions
diff --git a/tests/drivers/thread_context_test.py b/tests/drivers/thread_context_test.py
new file mode 100644
index 0000000..fa3f144
--- /dev/null
+++ b/tests/drivers/thread_context_test.py
@@ -0,0 +1,37 @@
+#!/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')