tricu

io_v2_context.py (2353B)

---

 #!/usr/bin/env python3
 """Actual read-only cwd/environment observations, independent of source handlers."""
 import argparse
 import json
 import os
 from pathlib import Path
 import subprocess
 from source_order import Runner
 from http_headers import result, tree_list, parse_tree_output
 from io_v2_text import text, problem
 ROOT=Path(__file__).resolve().parent.parent
 
 def main():
     p=argparse.ArgumentParser(description=__doc__);p.add_argument('tricu',type=Path);p.add_argument('--directory',type=Path,required=True)
     a=p.parse_args();d=a.directory.resolve();d.mkdir(parents=True,exist_ok=False);r=Runner(a.tricu,d);bundle=d/'context.arboricx'
     r.run('compile',['compile','--file',ROOT/'demos/io-v2/context.tri','--store',r.store,'--output',bundle]);pin=r.imported(bundle,'import');checks=[]
     key='TRICU_CONTEXT_TEST_VALUE'
     def run(label,name,argument,env,wanted):
         cmd=[str(r.exe),'arboricx','run','--module',pin,'--name',name,'--store',str(r.store),'--io','--format','ternary']
         if argument is not None:
             arg=d/(label+'.arg');arg.write_bytes(text(argument));cmd+=['--arg-tree-file',str(arg)]
         proc=subprocess.run(cmd,cwd=d,env=env,capture_output=True,timeout=20)
         (d/(label+'.stdout')).write_bytes(proc.stdout);(d/(label+'.stderr')).write_bytes(proc.stderr)
         assert proc.returncode==0,(label,proc.stderr)
         assert parse_tree_output(proc.stdout.decode())==wanted,label
         checks.append(label)
     env=dict(os.environ);env.pop(key,None)
     run('cwd','directory',None,env,result(True,text(str(d))))
     run('missing','variable',key,env,result(True,tree_list([text('missing')])))
     for i,value in enumerate(['','λ\n\ufeff','x'*4096]):
         run(f'value-{i}','variable',key,dict(env,**{key:value}),result(True,tree_list([text('value'),text(value)])))
     run('oversized','variable',key,dict(env,**{key:'x'*4097}),result(False,problem('invalid-bound','process/environment-variable','host context text exceeds scalar limit')))
     for i,name in enumerate(['','A=B','A\0B']):
         run(f'invalid-{i}','variable',name,env,result(False,problem('invalid-argument','process/environment-variable','')))
     (d/'report.json').write_text(json.dumps(dict(passed=checks,records=r.records),indent=2)+'\n')
     print(f'PASS {len(checks)} host-context controls')
 if __name__=='__main__':main()