Files

67 lines
4.2 KiB
Python

from pathlib import Path
import subprocess,os,signal,re,json,hashlib,time
from datetime import datetime,timezone
ROOT=Path(__file__).resolve().parents[4];P=Path(__file__).resolve().parent;O=P/'logs';O.mkdir(exist_ok=True)
FILE='packages/harness/tests/claude-session.test.mjs';source=ROOT/FILE
BASE='c16e943b90ef71ac96cfa5314bbea6040634d14d'
old='realpathSync(workspace).replace(/[/.]/g, "-")';new='realpathSync(workspace).replace(/[^A-Za-z0-9]/g, "-")'
original=source.read_bytes();assert original.decode().count(old)==1
fixed=original.decode().replace(old,new,1).encode()
rows=[];temps=[]
def stamp():return datetime.now(timezone.utc).isoformat()
def verify(data):
assert subprocess.check_output(['git','rev-parse','HEAD'],cwd=ROOT,text=True).strip()==BASE
assert source.read_bytes()==data
def run(label,template,data):
verify(data);tmp=subprocess.check_output(['mktemp','-d',template],text=True).strip();temps.append(tmp)
env=os.environ.copy();env.update(TMPDIR=tmp,DOCKER_HOST='unix:///mnt/storage/scratch/rocko-r304-dead.sock',NO_COLOR='1');env.pop('MOSAIC_GITEA_CREDENTIAL_FILE',None);env.pop('MOSAIC_LIVE_PROVIDER',None)
args=['timeout','-k','10','180','node','--test','--test-reporter=tap',FILE];log=O/(label+'.tap');start=stamp();begin=time.monotonic()
print('START',label,start,tmp,flush=True)
with log.open('wb') as f:
child=subprocess.Popen(args,cwd=ROOT,env=env,stdout=f,stderr=subprocess.STDOUT,start_new_session=True)
try:rc=child.wait(timeout=195)
except subprocess.TimeoutExpired:
os.killpg(child.pid,signal.SIGKILL);child.wait();rc=124
finally:
try:os.killpg(child.pid,signal.SIGTERM)
except ProcessLookupError:pass
text=log.read_text(errors='replace');counts={k:int(m[-1]) if (m:=re.findall(r'^# '+k+r' (\d+)\s*$',text,re.M)) else None for k in ['tests','pass','fail','cancelled','skipped']}
row=dict(label=label,start=start,end=stamp(),seconds=round(time.monotonic()-begin,3),exit=rc,counts=counts,tmpdir=tmp,head=BASE,testSha256=hashlib.sha256(data).hexdigest(),skips=[x for x in text.splitlines() if re.search(r'^skip\b|# SKIP\b',x,re.I)],log=str(log.relative_to(ROOT)))
rows.append(row)
with (P/'results.jsonl').open('a') as f:f.write(json.dumps(row)+'\n')
print('END',json.dumps(row),flush=True)
assert rc not in [124,137,-9] and 'SyntaxError:' not in text,'invalid run '+label
verify(data);return row,text
def red(label,template,data):
r,t=run(label,template,data)
assert r['exit']!=0 and r['counts']['tests']==10 and r['counts']['pass']==9 and r['counts']['fail']==1 and r['counts']['cancelled']==0 and r['counts']['skipped']==0,'unexpected red counts '+label
names=re.findall(r'^not ok \d+ - (.*)$',t,re.M)
assert len(names)==1 and "CLAUDE.md files and auto-memory" in names[0] and ':298:10' in t and 'ERR_ASSERTION' in t and 'MARKER-MEMORY' in t,'unexpected red '+label
def green(label,template,data):
r,t=run(label,template,data)
assert r['exit']==0 and r['counts']['tests']==10 and r['counts']['pass']==10 and r['counts']['fail']==0 and r['counts']['cancelled']==0 and r['counts']['skipped']==0,'unexpected mapping/refusal '+label
try:
red('baseline-underscore','/mnt/storage/scratch/tmp/rk304_XXXXXX',original)
green('baseline-hyphen','/mnt/storage/scratch/tmp/rk304-XXXXXX',original)
# Both installed-CLI probes now confirm the approved mapping before editing.
source.write_bytes(fixed)
green('fixed-mktemp','/mnt/storage/scratch/tmp/rk304XXXXXX',fixed)
green('fixed-underscore','/mnt/storage/scratch/tmp/rk304_XXXXXX',fixed)
green('fixed-space','/mnt/storage/scratch/tmp/rk304 XXXXXX',fixed)
source.write_bytes(original)
try:red('M1-old-regex','/mnt/storage/scratch/tmp/rk304_XXXXXX',original)
finally:source.write_bytes(fixed)
verify(fixed);print('BUILD_GREEN',stamp(),flush=True)
finally:
found=[]
for proc in Path('/proc').glob('[0-9]*'):
try:
name=(proc/'comm').read_text().strip();cwd=str((proc/'cwd').resolve())
if any(x in name for x in ['node','chrom','claude']) and any(cwd==r or cwd.startswith(r+'/') for r in [str(ROOT),*temps]):found.append(dict(pid=int(proc.name),name=name,cwd=cwd))
except (FileNotFoundError,PermissionError,ProcessLookupError):pass
(P/'cwd-sweep.json').write_text(json.dumps(found,indent=2)+'\n');print('CWD_SWEEP',found,flush=True)