135 lines
3.3 KiB
Python
135 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
import io
|
||
|
|
import contextlib
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
sys.path.insert(0, str(Path.cwd()))
|
||
|
|
|
||
|
|
from gx_cli.main import main
|
||
|
|
|
||
|
|
|
||
|
|
def test_inspect_basic():
|
||
|
|
"""Test inspecting a .gx file"""
|
||
|
|
gx_path = Path("/tmp/test_inspect_basic.gx")
|
||
|
|
|
||
|
|
# Compile
|
||
|
|
compile_code = main(["compile", "/home/dave/sample_code.py", "-o", str(gx_path)])
|
||
|
|
if compile_code != 0:
|
||
|
|
print("FAIL: compilation failed")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Inspect
|
||
|
|
f = io.StringIO()
|
||
|
|
with contextlib.redirect_stdout(f):
|
||
|
|
exit_code = main(["inspect", str(gx_path)])
|
||
|
|
|
||
|
|
output = f.getvalue()
|
||
|
|
|
||
|
|
if exit_code != 0:
|
||
|
|
print("FAIL: inspect returned non-zero")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Verify output contains expected sections
|
||
|
|
if "[Manifest]" not in output:
|
||
|
|
print("FAIL: missing [Manifest] section")
|
||
|
|
return False
|
||
|
|
|
||
|
|
if "[Segments]" not in output:
|
||
|
|
print("FAIL: missing [Segments] section")
|
||
|
|
return False
|
||
|
|
|
||
|
|
if "[Payload]" not in output:
|
||
|
|
print("FAIL: missing [Payload] section")
|
||
|
|
return False
|
||
|
|
|
||
|
|
print("PASS: Inspect shows manifest, segments, and payload")
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def test_inspect_manifest_fields():
|
||
|
|
"""Test that manifest contains expected fields"""
|
||
|
|
gx_path = Path("/tmp/test_manifest_fields.gx")
|
||
|
|
|
||
|
|
# Compile
|
||
|
|
main(["compile", "/home/dave/sample_code.py", "-o", str(gx_path)])
|
||
|
|
|
||
|
|
# Inspect and capture output
|
||
|
|
f = io.StringIO()
|
||
|
|
with contextlib.redirect_stdout(f):
|
||
|
|
main(["inspect", str(gx_path)])
|
||
|
|
|
||
|
|
output = f.getvalue()
|
||
|
|
|
||
|
|
# Check for key manifest fields
|
||
|
|
required_fields = ["version", "origin", "source_file", "source_type", "compression_model", "contributor"]
|
||
|
|
for field in required_fields:
|
||
|
|
if field not in output:
|
||
|
|
print(f"FAIL: missing manifest field: {field}")
|
||
|
|
return False
|
||
|
|
|
||
|
|
print("PASS: Manifest contains all required fields")
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def test_inspect_segments():
|
||
|
|
"""Test that segments are listed correctly"""
|
||
|
|
gx_path = Path("/tmp/test_segments.gx")
|
||
|
|
|
||
|
|
# Compile
|
||
|
|
main(["compile", "/home/dave/sample_code.py", "-o", str(gx_path)])
|
||
|
|
|
||
|
|
# Inspect
|
||
|
|
f = io.StringIO()
|
||
|
|
with contextlib.redirect_stdout(f):
|
||
|
|
main(["inspect", str(gx_path)])
|
||
|
|
|
||
|
|
output = f.getvalue()
|
||
|
|
|
||
|
|
# Should have segment entries
|
||
|
|
if "seg_0:" not in output:
|
||
|
|
print("FAIL: segment seg_0 not found")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Should show line ranges
|
||
|
|
if "lines" not in output:
|
||
|
|
print("FAIL: segment line ranges not shown")
|
||
|
|
return False
|
||
|
|
|
||
|
|
print("PASS: Segments listed with line ranges")
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def main_test():
|
||
|
|
print("[TEST SUITE] test_inspect.py")
|
||
|
|
print()
|
||
|
|
|
||
|
|
tests = [
|
||
|
|
("Basic inspect", test_inspect_basic),
|
||
|
|
("Manifest fields", test_inspect_manifest_fields),
|
||
|
|
("Segments listing", test_inspect_segments),
|
||
|
|
]
|
||
|
|
|
||
|
|
passed = 0
|
||
|
|
failed = 0
|
||
|
|
|
||
|
|
for name, test_func in tests:
|
||
|
|
print(f"Running: {name}...", end=" ")
|
||
|
|
try:
|
||
|
|
if test_func():
|
||
|
|
passed += 1
|
||
|
|
else:
|
||
|
|
failed += 1
|
||
|
|
except Exception as e:
|
||
|
|
print(f"FAIL: {e}")
|
||
|
|
failed += 1
|
||
|
|
|
||
|
|
print()
|
||
|
|
print(f"Results: {passed} passed, {failed} failed")
|
||
|
|
|
||
|
|
return 0 if failed == 0 else 1
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main_test())
|