45 lines
3.2 KiB
Python
45 lines
3.2 KiB
Python
|
"""Recovery choices keep the same task, goal history, plan and browser profile."""
|
||
|
|
import json,tempfile,threading,signal,sys
|
||
|
|
from pathlib import Path
|
||
|
|
from http.server import ThreadingHTTPServer
|
||
|
|
from runtime_flow import Provider,Run,saved
|
||
|
|
from cli_flow import response,tool
|
||
|
|
|
||
|
|
def main():
|
||
|
|
with tempfile.TemporaryDirectory(prefix='gb-recovery-') as root,ThreadingHTTPServer(('127.0.0.1',0),Provider) as server:
|
||
|
|
server.errors=[];threading.Thread(target=server.serve_forever,daemon=True).start();runs=[]
|
||
|
|
try:
|
||
|
|
replies=[response(tool('report_blocked',{'reason':'登入無法完成，兩次嘗試都停在驗證頁。','options':['由我在目前瀏覽器處理登入','先繼續寫文案']})),response(tool('write_file',{'path':'draft.txt','content':'draft without login'})),response(tool('report_done',{'message':'文案已備妥，登入仍未完成。'}))]
|
||
|
|
def callback(req):
|
||
|
|
if len(replies)==2:
|
||
|
|
last=json.loads(req['messages'][-1]['content']);assert last['status']=='replan' and last['answer']=='先繼續寫文案',last
|
||
|
|
return replies.pop(0)
|
||
|
|
server.callback=callback
|
||
|
|
run=Run(root,server);runs.append(run);run.wait_line('需要你的回覆');before=json.loads(max(Path(root,'sessions').glob('*.json'),key=lambda p:p.stat().st_mtime_ns).read_text());assert before['pending_question']['kind']=='recovery'
|
||
|
|
run.send('b');run.finish();after=saved(root)
|
||
|
|
assert before['id']==after['id'] and not after['pending_question'];assert (Path(root)/'draft.txt').exists()
|
||
|
|
print('PASS blocker -> letter choice -> alternate work in same session',flush=True)
|
||
|
|
server.callback=lambda _:response(tool('report_blocked',{'reason':'Still blocked','options':['Try a different route']}))
|
||
|
|
run=Run(root,server);runs.append(run);run.wait_line('需要你的回覆');run.send('2');run.finish(code=1)
|
||
|
|
assert saved(root)['last_verdict']=='blocked'
|
||
|
|
print('PASS explicit stop choice ends with blocked verdict',flush=True)
|
||
|
|
if '--browser' in sys.argv:
|
||
|
|
replies=[response(tool('browser_navigate',{'url':f'http://127.0.0.1:{server.server_port}/fixture'})),response(tool('browser_handoff',{'reason':'請在原頁面處理測試登入','options':['先寫文案']}))]
|
||
|
|
server.callback=lambda _:replies.pop(0)
|
||
|
|
run=Run(root,server);runs.append(run);run.wait_line('需要你的回覆');run.process.send_signal(signal.SIGINT);run.finish(code=130)
|
||
|
|
previous=saved(root);assert previous['pending_question']['kind']=='handoff'
|
||
|
|
replies=[response(tool('browser_read_page',{})),response(tool('report_done',{'message':'original browser restored and inspected'}))]
|
||
|
|
def restored(req):
|
||
|
|
assert any('login_verified' in m.get('content','') for m in req['messages']),req
|
||
|
|
return replies.pop(0)
|
||
|
|
server.callback=restored
|
||
|
|
run=Run(root,server,['run','--session',previous['id'],'continue']);runs.append(run);run.wait_line('需要你的回覆');run.send('我已經登入了');run.finish()
|
||
|
|
assert saved(root)['id']==previous['id']
|
||
|
|
print('PASS interrupted handoff reopens original browser before asking; returned control is not assumed login success',flush=True)
|
||
|
|
assert not server.errors,server.errors
|
||
|
|
finally:
|
||
|
|
for run in runs:
|
||
|
|
if run.process.poll() is None:run.process.terminate()
|
||
|
|
server.shutdown()
|
||
|
|
if __name__=='__main__':main()
|