mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 13:16:24 +00:00
b6ce047510
* feat(workflows): recover durable workflows after Gateway restart * docs(workflows): keep recovery guidance within inherited budget * fix(workflows): retain cancellation ownership and stabilize turn-limit test * fix(workflows): log service-owned action failures
153 lines
12 KiB
JavaScript
153 lines
12 KiB
JavaScript
// Self-contained loopback browser fixture; never a deployed Gateway.
|
|
import assert from 'node:assert/strict';
|
|
import {createServer} from 'node:http';
|
|
import {readFile} from 'node:fs/promises';
|
|
const {chromium}=await import(process.env.PLAYWRIGHT_MODULE ?? '../../frontend/node_modules/@playwright/test/index.mjs');
|
|
const directory=new URL('./deerflow_extension_workflows/static/',import.meta.url);
|
|
const server=createServer(async(req,res)=>{
|
|
if(req.url==='/index.mjs'||req.url==='/style.css'){
|
|
res.setHeader('Content-Type',req.url.endsWith('.mjs')?'text/javascript':'text/css');
|
|
res.end(await readFile(new URL(req.url.slice(1),directory)));return;
|
|
}
|
|
res.setHeader('Content-Type','text/html');
|
|
res.end(`<!doctype html><html><head><title>Workflow test</title></head><body><main></main><script type="module">
|
|
import plugin from '/index.mjs';
|
|
window.calls=[];window.opened=[];window.state='running';window.hidden=false;window.failList=0;window.failGet=0;window.loseLaunch=0;window.identities=new Set();window.savedDefinitions=[];window.definitionScript='';
|
|
let saved;
|
|
window.view=plugin.surfaces[0].mount(document.querySelector('main'),{locale:'en',signal:new AbortController().signal,
|
|
openConversation:async id=>window.opened.push(id),
|
|
callBackend:async(action,payload)=>{
|
|
calls.push({action,payload});
|
|
if(action==='definitions')return {definitions:window.savedDefinitions};
|
|
if(action==='save_definition'){const item={name:payload.name,version:'saved-version'};window.savedDefinitions=[item];window.definitionScript=payload.script;return item;}
|
|
if(action==='get_definition'&&window.failDefinition)throw Error('Definition lookup failed');
|
|
if(action==='get_definition')return {...payload,script:window.definitionScript};
|
|
const bytes=new TextEncoder().encode('Exact <img src=x onerror=alert(1)> file');
|
|
const hash=Array.from(new Uint8Array(await crypto.subtle.digest('SHA-256',bytes)),b=>b.toString(16).padStart(2,'0')).join('');
|
|
if(action==='get_artifact')return {version:'artifact-v1',sha256:hash,offset:0,next_offset:null,content_base64:btoa(new TextDecoder().decode(bytes))};
|
|
if(action==='get_step'&&window.fileReview)return {kind:'file',name:'report.txt',size:bytes.length,sha256:window.badHash?'bad':hash,artifact_version:'artifact-v1'};
|
|
if(action==='get_step')return {output:{report:'Retained <img src=x onerror=alert(1)> report'},artifact_version:'artifact-v1'};
|
|
if(action==='decide'){window.state='completed';return {status:'approved'};}
|
|
if(action==='recover'){window.state='paused';return {status:'queued'};}
|
|
if(action==='resume'){window.state='completed';return {status:'resumed'};}
|
|
if(action==='launch'){saved=payload;identities.add(payload.request_id);if(window.loseLaunch-- > 0)throw Error('Launch response lost');return {run_id:'fixture-run',status:'queued'};}
|
|
if(action==='list' && window.failList-- > 0)throw Error('List refresh failed');
|
|
if(action==='get' && window.failGet-- > 0)throw Error('Inspection refresh failed');
|
|
if(action==='list')return {runs:saved&&!window.hidden?[{run_id:'fixture-run',status:window.state}]:[]};
|
|
if(action==='cancel'){window.state='cancelled';return {status:'cancelling'};}
|
|
if(action==='delete'){window.hidden=true;return {deleted:true};}
|
|
if(action==='get')return {fingerprint:'fingerprint',checkpoints:(window.state==='waiting_approval'||(window.fileReview&&window.state==='completed'))?[{step_id:'confirm',attempt:1,kind:'humanCheckpoint',status:window.state==='completed'?'approved':'pending',source_step_id:'check',artifact_version:'artifact-v1'}]:window.state==='paused'?[{step_id:'p',attempt:1,kind:'pause',status:'pending'}]:[],run_id:'fixture-run',status:window.state,script:saved.script,args:saved.args,script_hash:'script-version',args_hash:'args-version',limits:{concurrency:2,max_calls:8},steps:[{step_id:'check',agent_ref:'lead_agent',status:window.state,thread_id:'child-thread',run_id:'child-run'}],logs:['Checking'],result:'<img src=x onerror=alert(1)>'};
|
|
}});
|
|
</script></body></html>`);
|
|
});
|
|
await new Promise(resolve=>server.listen(0,'127.0.0.1',resolve));
|
|
const browser=await chromium.launch({headless:true,...(process.env.PLAYWRIGHT_CHANNEL?{channel:process.env.PLAYWRIGHT_CHANNEL}:{})});
|
|
try{
|
|
const page=await browser.newPage({viewport:{width:1280,height:1000}});
|
|
const errors=[];page.on('pageerror',e=>{errors.push(e.message);console.error(e.message);});
|
|
await page.goto(`http://127.0.0.1:${server.address().port}/`);
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.getByRole('heading',{name:'fixture-run · running',exact:true}).waitFor();
|
|
const launch=await page.evaluate(()=>calls.find(c=>c.action==='launch'));
|
|
await page.evaluate(()=>window.state='recoverable');
|
|
// Refresh the selected run through the existing list action.
|
|
await page.locator('nav button').first().click();
|
|
await page.getByRole('button',{name:'Recover workflow',exact:true}).click();
|
|
assert.equal(await page.evaluate(()=>calls.filter(c=>c.action==='recover').length),1);
|
|
await page.getByRole('button',{name:'Resume this checkpoint',exact:true}).waitFor();
|
|
await page.evaluate(()=>window.state='running');
|
|
await page.locator('nav button').first().click();
|
|
|
|
assert.equal(launch.payload.request_id.length,32);
|
|
assert.match(launch.payload.script,/await agent/);
|
|
await page.getByRole('button',{name:'Open child conversation',exact:true}).click();
|
|
assert.deepEqual(await page.evaluate(()=>opened),['child-thread']);
|
|
assert.equal(await page.locator('img').count(),0);
|
|
await page.getByRole('button',{name:'Cancel workflow',exact:true}).click();
|
|
await page.getByRole('heading',{name:'fixture-run · cancelled',exact:true}).waitFor();
|
|
await page.getByRole('button',{name:'Delete retained record',exact:true}).click();
|
|
await page.waitForFunction(()=>window.hidden);
|
|
for(const failure of ['failList','failGet','loseLaunch']){
|
|
await page.reload();
|
|
await page.waitForFunction(()=>calls.some(c=>c.action==='list'));
|
|
await page.evaluate(key=>window[key]=1,failure);
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.waitForFunction(()=>!document.querySelector('button[type=submit]').disabled);
|
|
if(failure!=='loseLaunch'){
|
|
assert.match(await page.getByRole('status').textContent(),/fixture-run/);
|
|
await page.getByRole('button',{name:'Retry inspection',exact:true}).click();
|
|
}else{
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
}
|
|
await page.getByRole('heading',{name:'fixture-run · running',exact:true}).waitFor();
|
|
// Resubmitting unchanged accepted input also keeps its launch identity.
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.waitForFunction(()=>!document.querySelector('button[type=submit]').disabled);
|
|
assert.equal(await page.evaluate(()=>identities.size),1,failure);
|
|
await page.getByLabel('Script',{exact:true}).fill('return 43;');
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.waitForFunction(()=>!document.querySelector('button[type=submit]').disabled);
|
|
assert.equal(await page.evaluate(()=>identities.size),2,failure);
|
|
}
|
|
await page.evaluate(()=>window.state='waiting_approval');
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.getByRole('button',{name:'Review exact output',exact:true}).click();
|
|
await page.getByRole('button',{name:'Approve this version',exact:true}).click();
|
|
await page.getByRole('heading',{name:'fixture-run · completed',exact:true}).waitFor();
|
|
const decision=await page.evaluate(()=>calls.find(c=>c.action==='decide').payload);
|
|
assert.equal(decision.artifact_version,'artifact-v1');assert.equal(decision.attempt,1);
|
|
await page.evaluate(()=>{window.state='waiting_approval';window.fileReview=true;window.badHash=true;});
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.getByRole('button',{name:'Review exact output',exact:true}).click();
|
|
await page.waitForFunction(()=>document.querySelector('[role=status]').textContent.includes('mismatch'));
|
|
assert.equal(await page.getByRole('button',{name:'Approve this version',exact:true}).count(),0);
|
|
await page.evaluate(()=>window.badHash=false);
|
|
await page.getByRole('button',{name:'Review exact output',exact:true}).click();
|
|
await page.getByRole('link',{name:'Download this exact version',exact:true}).waitFor();
|
|
assert.equal(await page.locator('img').count(),0);
|
|
await page.getByRole('button',{name:'Approve this version',exact:true}).click();
|
|
await page.getByRole('heading',{name:'fixture-run · completed',exact:true}).waitFor();
|
|
await page.getByRole('button',{name:'Review exact output',exact:true}).click();
|
|
const retainedDownload=page.getByRole('link',{name:'Download this exact version',exact:true});
|
|
await retainedDownload.waitFor();
|
|
assert.equal(await retainedDownload.evaluate(async a=>await (await fetch(a.href)).text()),'Exact <img src=x onerror=alert(1)> file');
|
|
assert.equal(await page.getByRole('button',{name:'Approve this version',exact:true}).count(),0);
|
|
await page.getByRole('button',{name:'fixture- · completed',exact:true}).click();
|
|
await page.getByRole('button',{name:'Review exact output',exact:true}).click();
|
|
await retainedDownload.waitFor();
|
|
assert.equal(await retainedDownload.evaluate(async a=>await (await fetch(a.href)).text()),'Exact <img src=x onerror=alert(1)> file');
|
|
await page.evaluate(()=>window.state='paused');
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.getByRole('button',{name:'Resume this checkpoint',exact:true}).click();
|
|
await page.getByRole('heading',{name:'fixture-run · completed',exact:true}).waitFor();
|
|
assert.equal(await page.evaluate(()=>calls.find(c=>c.action==='resume').payload.step_id),'p');
|
|
await page.getByLabel('Workflow name',{exact:true}).fill('inspection');
|
|
await page.getByRole('button',{name:'Save workflow version',exact:true}).click();
|
|
await page.getByRole('option',{name:'inspection · saved-versio',exact:true}).waitFor({state:'attached'});
|
|
await page.getByLabel('Saved workflow version',{exact:true}).selectOption(JSON.stringify({name:'inspection',version:'saved-version'}));
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.waitForFunction(()=>calls.some(c=>c.action==='launch'&&c.payload.definition?.version==='saved-version'));
|
|
assert.equal(await page.evaluate(()=>calls.filter(c=>c.action==='launch').at(-1).payload.script),undefined);
|
|
await page.getByLabel('Script',{exact:true}).fill('return "previous editor";');
|
|
await page.evaluate(()=>window.failDefinition=true);
|
|
await page.getByLabel('Saved workflow version',{exact:true}).selectOption(JSON.stringify({name:'inspection',version:'saved-version'}));
|
|
await page.waitForFunction(()=>document.querySelector('[role=status]').textContent.includes('Definition lookup failed'));
|
|
assert.equal(await page.getByRole('button',{name:'Launch workflow',exact:true}).isDisabled(),true);
|
|
const beforeFailedSubmit=await page.evaluate(()=>calls.filter(c=>c.action==='launch').length);
|
|
await page.locator('form').evaluate(form=>form.dispatchEvent(new Event('submit',{bubbles:true,cancelable:true})));
|
|
assert.equal(await page.evaluate(()=>calls.filter(c=>c.action==='launch').length),beforeFailedSubmit);
|
|
await page.getByLabel('Saved workflow version',{exact:true}).selectOption('');
|
|
assert.equal(await page.getByRole('button',{name:'Launch workflow',exact:true}).isDisabled(),false);
|
|
await page.getByLabel('Saved workflow version',{exact:true}).selectOption(JSON.stringify({name:'inspection',version:'saved-version'}));
|
|
await page.waitForFunction(()=>document.querySelector('button[type=submit]').disabled);
|
|
await page.getByLabel('Script',{exact:true}).fill('return "explicit edit";');
|
|
await page.getByRole('button',{name:'Launch workflow',exact:true}).click();
|
|
await page.waitForFunction(()=>calls.filter(c=>c.action==='launch').at(-1).payload.script==='return "explicit edit";');
|
|
await page.evaluate(()=>view.dispose());
|
|
assert.equal(await page.locator('.script-workflows').count(),0);
|
|
assert.deepEqual(errors,[]);
|
|
console.log('Browser checks passed: launch retries, saved versions, exact-output review, approval, resume, child navigation, cancellation, escaped output, deletion and disposal.');
|
|
}finally{
|
|
await browser.close();await new Promise(resolve=>server.close(resolve));
|
|
}
|