1 files changed, 34 insertions, 0 deletions
diff --git a/tests/drivers/runtime_image_test.py b/tests/drivers/runtime_image_test.py
new file mode 100644
index 0000000..c1f4b9c
--- /dev/null
+++ b/tests/drivers/runtime_image_test.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+"""self-hosted run-only images retain enhancements and generic method bodies."""
+import argparse
+import subprocess
+import tempfile
+from pathlib import Path
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--case', default='')
+parser.add_argument('--host', default='bin/zy')
+parser.add_argument('--mkgenome', default='bin/mkgenome')
+args = parser.parse_args()
+cases = [
+    ('enhancements', 'include rt; fn main() { rt.put_n(17 % 5); rt.put_c(32); rt.put_n(17 / 5); }', b'2 3'),
+    ('array algorithms', Path('examples/learn/31-array-algorithms.zc').read_text(), Path('examples/learn/31-array-algorithms.out').read_bytes()),
+    ('owned pop', 'include rt; int*.u take() { int*[?] a = rt.new(int*[?]); int* p = rt.new(int); *p = 8; a.push(p); return a.pop(); } fn main() { let p = take(); rt.put_n(rt.isobj(p)); rt.put_c(32); rt.put_n(*p); }', b'1 8'),
+]
+prefix = b''.join(Path(f).read_bytes() + b'\n' for f in ['src/runtime/rt.zc', 'src/core/core.zc', 'src/runtime/run.zc'])
+for name, source, expected in cases:
+    if args.case and args.case not in name:
+        continue
+    with tempfile.TemporaryDirectory(prefix='tinyvm-image-') as work:
+        directory = Path(work)
+        r = subprocess.run([args.host], input=Path('src/tools/build.zc').read_bytes() + b'`' + prefix + source.encode(), capture_output=True, timeout=30)
+        assert r.returncode == 0 and len(r.stdout) >= 56 and not r.stdout.startswith(b'zc:'), (name, r.stdout[:1000], r.stderr)
+        header = subprocess.run([args.mkgenome], input=r.stdout, capture_output=True, timeout=5)
+        assert header.returncode == 0, (name, header.stderr)
+        (directory / 'genome.h').write_bytes(header.stdout)
+        for f in ['src/core/memmap.h', 'src/core/fops.h', 'src/core/bops.h', 'src/core/stack.h', 'src/micro/micro.c', 'src/micro/micro_port.h', 'src/micro/port_linux.c']:
+            (directory / Path(f).name).write_bytes(Path(f).read_bytes())
+        subprocess.run(['cc', '-O2', '-w', '-fwrapv', '-pthread', '-DZY_THREADS', str(directory / 'micro.c'), str(directory / 'port_linux.c'), '-I' + work, '-o', str(directory / 'image')], check=True, timeout=30)
+        r = subprocess.run([str(directory / 'image')], capture_output=True, timeout=10)
+        assert r.returncode == 0 and r.stdout == expected, (name, r.returncode, r.stdout, r.stderr, expected)
+        print('ok   self-hosted run-only image: ' + name)