managedBrowsers.test.ts 54 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812
  1. // @vitest-environment node
  2. import {afterEach,expect,it,vi} from 'vitest'
  3. import Koa from 'koa'
  4. import {bodyParser} from '@koa/bodyparser'
  5. import request from 'supertest'
  6. import {createHash,randomUUID} from 'node:crypto'
  7. import {mkdtempSync,rmSync,writeFileSync,symlinkSync,mkdirSync,realpathSync} from 'node:fs'
  8. import {readFile} from 'node:fs/promises'
  9. import {tmpdir} from 'node:os'
  10. import {join} from 'node:path'
  11. import {ManagedBrowsers,type BrowserTurn,type BrowserTurnOptions} from '../../src/server/managedBrowsers'
  12. import {WorkspaceStore} from '../../src/server/workspaceStore'
  13. import {WorkspaceNodes} from '../../src/server/workspaceGateway'
  14. import {loadServerConfig} from '../../src/server/config'
  15. import {saveHostTools} from '../../src/server/hostToolSettings'
  16. import {HttpError} from '../../src/server/errors'
  17. import type {LocalAuthStore} from '../../src/server/localAuth'
  18. import type {RunnerHub} from '../../src/server/runnerHub'
  19. import type {BrowserScope,BrowserState} from '../../src/runner/browser/types'
  20. import type {BrowserInstallation,ManagedBrowserCard} from '../../src/shared/managedBrowser'
  21. import {browserOwnedFile,browserVmFile} from '../../src/server/browserFiles'
  22. import {UploadStore} from '../../src/server/uploads'
  23. import {FileTransferFiles} from '../../src/shared/fileTransferEndpoint.mjs'
  24. const digest=(value:string|Buffer)=>createHash('sha256').update(value).digest('hex')
  25. const defer=<T=void>()=>{let resolve!:(value:T)=>void;const promise=new Promise<T>(done=>{resolve=done});return {promise,resolve}}
  26. function fixture(enabled=true){
  27. const home=realpathSync(mkdtempSync(join(tmpdir(),'yaoyao-managed-browser-'))),store=new WorkspaceStore(home)
  28. if(enabled)saveHostTools(home,{managedBrowser:true})
  29. const versions=new Map<string,number>(),blockedSources=new Set<string>(),inactiveOwners=new Set<string>()
  30. const auth={require:(ctx:any)=>({id:ctx.get('x-fixture-owner')||'owner'}),pushAuthorizationVersion:(owner:string)=>versions.get(owner)??1,isUserActive:(owner:string)=>!inactiveOwners.has(owner)} as unknown as LocalAuthStore
  31. const nodes=new WorkspaceNodes(store,loadServerConfig({HERMES_YAOYAO_HOME:home}),{} as any)
  32. nodes.sourceAllowed=(owner,node,profile)=>!blockedSources.has(`${owner}:${node}:${profile}`)
  33. let installation:BrowserInstallation|undefined,retention=true
  34. const states=new Map<string,BrowserState>(),calls:any[]=[],turns:BrowserTurn[]=[]
  35. let runnerId=randomUUID(),runnerEpoch=randomUUID(),downloadReply:(payload:any)=>unknown=()=>{throw new Error('Unexpected download')}
  36. let beforeReply:((payload:any)=>Promise<void>)|undefined
  37. let afterReply:((payload:any)=>Promise<void>)|undefined
  38. const stateFor=(scope:BrowserScope)=>{
  39. const key=runnerId+':'+runnerEpoch+':'+JSON.stringify(scope);let state=states.get(key)
  40. if(!state){state={open:false,generation:1,profile:scope.profile,tabs:[],downloads:[]};states.set(key,state)}
  41. return state
  42. }
  43. let service:ManagedBrowsers
  44. const hub={browserRunner:()=>({id:runnerId,browserEpoch:runnerEpoch}),supportsBrowserRetention:()=>retention,browser:vi.fn(async(owner:string,agent:any,payload:any,authorize:()=>void)=>{
  45. authorize()
  46. expect(service.allowed(runnerId,payload.scope,payload.grantId)).toBe(true)
  47. calls.push({owner,agentId:agent.id,runnerId,...structuredClone(payload)})
  48. const state=stateFor(payload.scope)
  49. if(beforeReply)await beforeReply(payload)
  50. let result:unknown
  51. if(payload.op==='prepare'){
  52. if(installation?.status!=='failed'||payload.retry)installation={status:'installing',message:'正在下载浏览器',updatedAt:Date.now()}
  53. }else if(payload.op==='open'){
  54. if(!state.open){state.open=true;state.generation++;state.tabs=[{id:'tab',title:'Example',url:'about:blank',active:true}]}
  55. }else if(payload.op==='pause'||payload.op==='resume'||payload.op==='park')state.generation++
  56. else if(payload.op==='close'){state.open=false;state.tabs=[];state.generation++}
  57. else if(payload.op==='download')result=downloadReply(payload)
  58. else if(payload.op==='execute'){
  59. if(payload.operation.generation!==state.generation)throw new HttpError(409,'代次改变','browser_generation_expired')
  60. if(payload.operation.action.kind==='screenshot')result={frameId:randomUUID(),generation:state.generation,data:'c2NyZWVu',width:800,height:600,createdAt:Date.now()}
  61. else {
  62. if(payload.operation.action.kind==='navigate'&&state.tabs[0])state.tabs[0].url=payload.operation.action.url
  63. result={ok:true,action:payload.operation.action}
  64. }
  65. }
  66. const reply=result??structuredClone({...state,...(installation?{available:installation.status==='ready',installation}:{})})
  67. if(afterReply)await afterReply(payload)
  68. authorize()
  69. return reply
  70. })} as unknown as RunnerHub
  71. service=new ManagedBrowsers(store,auth,nodes,hub)
  72. const app=new Koa()
  73. app.use(async(ctx,next)=>{try{await next()}catch(error){ctx.status=error instanceof HttpError?error.status:500;ctx.body={code:(error as any).code,error:error instanceof Error?error.message:'Failed'}}})
  74. app.use(bodyParser());const router=service.router();app.use(router.routes());app.use(router.allowedMethods())
  75. const agent=store.createAgent('owner',{name:'浏览器 Bot',profile:'default'})
  76. const base=`/api/app/agents/${agent.id}`
  77. const turn=(options:Partial<BrowserTurnOptions>={},owner='owner',target=agent)=>{
  78. const controller=new AbortController(),publish=vi.fn(async(name:string,bytes:Buffer)=>({name,size:bytes.length})),upload=vi.fn(async()=>{throw new HttpError(403,'附件未授权','file_forbidden')})
  79. const value=service.openTurn(owner,target,{workId:randomUUID(),signal:controller.signal,authorize:()=>{},publish,upload,...options})
  80. turns.push(value);return {value,controller,publish,upload}
  81. }
  82. return {home,store,nodes,service,agent,base,app,hub,calls,versions,blockedSources,inactiveOwners,turn,
  83. setInstallation:(status:BrowserInstallation['status'],message='浏览器环境',error?:string)=>{installation={status,message,error,updatedAt:Date.now()}},
  84. setRetention:(value:boolean)=>{retention=value},
  85. runnerClosed:(scope:BrowserScope)=>{const state=stateFor(scope);state.open=false;state.tabs=[];state.generation++},
  86. setRunner:(id:string)=>{runnerId=id},runner:()=>runnerId,
  87. setEpoch:(id:string)=>{runnerEpoch=id},
  88. setDownload:(reply:(payload:any)=>unknown)=>{downloadReply=reply},
  89. setBeforeReply:(value:typeof beforeReply)=>{beforeReply=value},
  90. setAfterReply:(value:typeof afterReply)=>{afterReply=value},
  91. async close(){for(const turn of turns)await turn.close();service.close();nodes.close();store.close();rmSync(home,{recursive:true,force:true})}}
  92. }
  93. const controlURL=(base:string,action='')=>`${base}/computer${action}?backend=managed-browser`
  94. afterEach(()=>vi.restoreAllMocks())
  95. it('defaults off and reads an enabled browser without opening any execution environment',async()=>{
  96. const f=fixture(false)
  97. try{
  98. const off=await request(f.app.callback()).get(f.base+'/managed-browser').expect(200)
  99. expect(off.body).toMatchObject({enabled:false,available:false,open:false})
  100. expect(f.calls).toEqual([])
  101. saveHostTools(f.home,{managedBrowser:true,vm:false,serverComputer:false,scriptMachine:false})
  102. const status=await request(f.app.callback()).get(f.base+'/managed-browser').expect(200)
  103. expect(status.body).toMatchObject({enabled:true,available:true,open:false})
  104. expect(f.calls.map(call=>call.op)).toEqual(['status'])
  105. expect(f.service.allowed(f.runner(),f.calls[0].scope,f.calls[0].grantId)).toBe(false)
  106. await request(f.app.callback()).get(controlURL(f.base)).expect(200)
  107. expect(f.calls.some(call=>call.op==='open')).toBe(false)
  108. }finally{await f.close()}
  109. })
  110. it('isolates owner, source profile and grant scope instead of granting a whole runner',async()=>{
  111. const f=fixture()
  112. try{
  113. const other=f.store.createAgent('other',{name:'另一账号',profile:'default'})
  114. const first=f.turn(),second=f.turn({},'other',other)
  115. await first.value.call('managed_browser_state',{},'state-one')
  116. await second.value.call('managed_browser_state',{},'state-two')
  117. const [a,b]=f.calls
  118. expect(a.scope.ownerKey).toBe(digest('owner'));expect(b.scope.ownerKey).toBe(digest('other'))
  119. expect(f.service.allowed(f.runner(),a.scope,a.grantId)).toBe(true)
  120. expect(f.service.allowed(f.runner(),b.scope,a.grantId)).toBe(false)
  121. expect(f.service.allowed(randomUUID(),a.scope,a.grantId)).toBe(false)
  122. await request(f.app.callback()).get(f.base+'/managed-browser').set('x-fixture-owner','other').expect(404)
  123. f.blockedSources.add('owner:local:default')
  124. expect(f.service.allowed(f.runner(),a.scope,a.grantId)).toBe(false)
  125. await expect(first.value.call('managed_browser_state',{},'denied')).rejects.toMatchObject({code:'agent_source_forbidden'})
  126. expect((await f.service.state('owner',f.agent.id)).available).toBe(false)
  127. }finally{await f.close()}
  128. })
  129. it('keeps control tokens private, rejects cross-owner credentials and stale generations',async()=>{
  130. const f=fixture()
  131. try{
  132. const requestId=randomUUID(),take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).expect(200)
  133. const credentials={controlId:take.body.controlId,token:take.body.token}
  134. expect(credentials.token.length).toBeGreaterThan(32)
  135. const status=await request(f.app.callback()).get(controlURL(f.base)).expect(200)
  136. expect(status.body.mode).toBe('human');expect(status.body).not.toHaveProperty('token')
  137. expect(JSON.stringify(f.calls)).not.toContain(credentials.token)
  138. const repeat=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).expect(200)
  139. expect(repeat.body.token).toBe(credentials.token)
  140. await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(409)
  141. await request(f.app.callback()).post(controlURL(f.base,'/renew')).send({...credentials,token:'x'.repeat(43)}).expect(403)
  142. await request(f.app.callback()).post(controlURL(f.base,'/renew')).set('x-fixture-owner','other').send(credentials).expect(403)
  143. await request(f.app.callback()).post(f.base+'/managed-browser/action').send({...credentials,requestId:randomUUID(),generation:take.body.generation-1,action:{kind:'navigate',url:'https://example.com'}}).expect(409)
  144. const frame=await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  145. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...credentials,requestId:randomUUID(),generation:take.body.generation,frameId:randomUUID(),action:{kind:'click',x:1,y:1}}).expect(409)
  146. // Another viewer can read a frame without invalidating the human's recent
  147. // frame; retries of the same input must not press the key twice.
  148. await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  149. const input={...credentials,requestId:randomUUID(),generation:frame.body.generation,frameId:frame.body.id,action:{kind:'key',key:'a',modifiers:['ctrl']}}
  150. await request(f.app.callback()).post(controlURL(f.base,'/input')).send(input).expect(200)
  151. expect(f.calls.at(-1).operation.action).toEqual({kind:'key',key:'Control+a'})
  152. await request(f.app.callback()).post(controlURL(f.base,'/input')).send(input).expect(200)
  153. expect(f.calls.filter(call=>call.operation?.action.key==='Control+a')).toHaveLength(1)
  154. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...input,action:{kind:'text',text:'different'}}).expect(409)
  155. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...credentials,requestId:randomUUID(),generation:frame.body.generation,frameId:frame.body.id,action:{kind:'text',text:'stale'}}).expect(409)
  156. f.versions.set('owner',2)
  157. await request(f.app.callback()).post(controlURL(f.base,'/renew')).send(credentials).expect(410)
  158. }finally{await f.close()}
  159. })
  160. it('forwards bounded screenshot drags once and exposes the same action to Bot tools',async()=>{
  161. const f=fixture()
  162. try{
  163. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  164. const credentials={controlId:take.body.controlId,token:take.body.token}
  165. const frame=await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  166. const action={kind:'drag',fromX:10,fromY:20,toX:500,toY:300}
  167. const body={...credentials,requestId:randomUUID(),generation:frame.body.generation,frameId:frame.body.id,action}
  168. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...body,requestId:randomUUID(),action:{...action,toX:800}}).expect(400)
  169. expect(f.calls.some(call=>call.operation?.action.kind==='drag')).toBe(false)
  170. await request(f.app.callback()).post(controlURL(f.base,'/input')).send(body).expect(200)
  171. await request(f.app.callback()).post(controlURL(f.base,'/input')).send(body).expect(200)
  172. expect(f.calls.filter(call=>call.operation?.action.kind==='drag')).toHaveLength(1)
  173. expect(f.calls.at(-1).operation.action).toEqual(action)
  174. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...body,requestId:randomUUID()}).expect(409)
  175. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({...credentials,notes:''}).expect(200)
  176. const turn=f.turn()
  177. await turn.value.call('managed_browser_action',{action},'bot-drag')
  178. expect(f.calls.filter(call=>call.operation?.action.kind==='drag')).toHaveLength(2)
  179. expect(f.calls.findLast(call=>call.operation?.action.kind==='drag').operation.action).toEqual(action)
  180. await expect(turn.value.call('managed_browser_action',{action:{kind:'pointer',phase:'start',gestureId:randomUUID(),x:10,y:20}},'bot-held-pointer')).rejects.toMatchObject({code:'browser_input_unsupported'})
  181. }finally{await f.close()}
  182. })
  183. it('retains frames during a held gesture and allows authenticated cancellation after a frame expires',async()=>{
  184. const f=fixture()
  185. try{
  186. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  187. const credentials={controlId:take.body.controlId,token:take.body.token},gestureId=randomUUID()
  188. const frame=await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  189. const body={...credentials,generation:frame.body.generation,frameId:frame.body.id}
  190. const pointer=(phase:string)=>({kind:'pointer',gestureId,phase,x:10,y:20})
  191. for(const phase of ['start','move','end'])await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...body,requestId:randomUUID(),action:pointer(phase)}).expect(200)
  192. expect(f.calls.filter(call=>call.operation?.action.kind==='pointer').map(call=>call.operation.action.phase)).toEqual(['start','move','end'])
  193. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...body,requestId:randomUUID(),action:pointer('move')}).expect(409)
  194. const cancel={...body,frameId:randomUUID(),requestId:randomUUID(),action:pointer('cancel')}
  195. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...cancel,token:'x'.repeat(43)}).expect(403)
  196. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...cancel,requestId:randomUUID(),generation:body.generation-1}).expect(409)
  197. await request(f.app.callback()).post(controlURL(f.base,'/input')).send(cancel).expect(200)
  198. expect(f.calls.at(-1).operation.action.phase).toBe('cancel')
  199. await request(f.app.callback()).post(f.base+'/managed-browser/action').send({...body,requestId:randomUUID(),action:pointer('start')}).expect(400)
  200. }finally{await f.close()}
  201. })
  202. it('expires a human grant without admitting further input',async()=>{
  203. const f=fixture()
  204. try{
  205. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  206. const credentials={controlId:take.body.controlId,token:take.body.token}
  207. vi.spyOn(Date,'now').mockReturnValue(take.body.expiresAt+1)
  208. await request(f.app.callback()).post(controlURL(f.base,'/renew')).send(credentials).expect(410)
  209. expect(f.service.allowed(f.runner(),f.calls[0].scope,credentials.controlId)).toBe(false)
  210. }finally{await f.close()}
  211. })
  212. it('joins concurrent retries of a human click and requires a new frame for the next action',async()=>{
  213. const f=fixture(),gate=defer(),entered=defer()
  214. try{
  215. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  216. const frame=await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  217. const input={controlId:take.body.controlId,token:take.body.token,requestId:randomUUID(),generation:frame.body.generation,frameId:frame.body.id,action:{kind:'click',x:15,y:25}}
  218. f.setBeforeReply(async payload=>{if(payload.operation?.action.kind==='coordinate'){entered.resolve();await gate.promise}})
  219. const first=request(f.app.callback()).post(controlURL(f.base,'/input')).send(input).then(value=>value)
  220. await entered.promise
  221. const retry=request(f.app.callback()).post(controlURL(f.base,'/input')).send(input).then(value=>value)
  222. await new Promise(resolve=>setTimeout(resolve,20));gate.resolve()
  223. expect((await Promise.all([first,retry])).map(reply=>reply.status)).toEqual([200,200])
  224. expect(f.calls.filter(call=>call.operation?.action.kind==='coordinate')).toHaveLength(1)
  225. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...input,requestId:randomUUID()}).expect(409)
  226. const current=await request(f.app.callback()).get(controlURL(f.base,'/frame')).expect(200)
  227. await request(f.app.callback()).post(controlURL(f.base,'/input')).send({...input,requestId:randomUUID(),frameId:current.body.id}).expect(200)
  228. expect(f.calls.filter(call=>call.operation?.action.kind==='coordinate')).toHaveLength(2)
  229. }finally{gate.resolve();await f.close()}
  230. })
  231. it('waits for one confirmed pause before returning duplicate takeover requests',async()=>{
  232. const f=fixture(),gate=defer(),entered=defer(),requestId=randomUUID()
  233. let firstSettled=false,secondSettled=false
  234. f.setBeforeReply(async payload=>{if(payload.op==='pause'){entered.resolve();await gate.promise}})
  235. try{
  236. const first=request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).then(value=>{firstSettled=true;return value})
  237. await entered.promise
  238. const second=request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).then(value=>{secondSettled=true;return value})
  239. await new Promise(resolve=>setTimeout(resolve,20))
  240. expect(firstSettled).toBe(false);expect(secondSettled).toBe(false)
  241. gate.resolve()
  242. const replies=await Promise.all([first,second])
  243. expect(replies.map(reply=>reply.status)).toEqual([200,200])
  244. expect(replies[0]!.body.controlId).toBe(replies[1]!.body.controlId)
  245. expect(replies[0]!.body.token).toBe(replies[1]!.body.token)
  246. expect(f.calls.filter(call=>call.op==='pause')).toHaveLength(1)
  247. }finally{gate.resolve();await f.close()}
  248. })
  249. it('joins duplicate takeovers that arrived while the previous task was parking',async()=>{
  250. const f=fixture(),gate=defer(),entered=defer(),requestId=randomUUID()
  251. try{
  252. const task=f.turn();await task.value.call('managed_browser_open',{},'open')
  253. f.setBeforeReply(async payload=>{if(payload.op==='park'){entered.resolve();await gate.promise}})
  254. const closing=task.value.close();await entered.promise
  255. const first=request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).then(value=>value)
  256. const second=request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId}).then(value=>value)
  257. await new Promise(resolve=>setTimeout(resolve,20));gate.resolve();await closing
  258. const replies=await Promise.all([first,second])
  259. expect(replies.map(reply=>reply.status)).toEqual([200,200])
  260. expect(replies[0]!.body.controlId).toBe(replies[1]!.body.controlId)
  261. expect(replies[0]!.body.token).toBe(replies[1]!.body.token)
  262. expect(f.calls.filter(call=>call.op==='pause')).toHaveLength(1)
  263. }finally{gate.resolve();await f.close()}
  264. })
  265. it('holds task actions through takeover and resumes the same session with human notes',async()=>{
  266. const f=fixture()
  267. try{
  268. const task=f.turn()
  269. await task.value.call('managed_browser_open',{},'open')
  270. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  271. let settled=false
  272. const paused=task.value.call('managed_browser_action',{action:{kind:'text',text:'continue'}},'after-human').finally(()=>{settled=true})
  273. await new Promise(resolve=>setTimeout(resolve,20))
  274. expect(settled).toBe(false)
  275. expect(f.calls.some(call=>call.operation?.action.text==='continue')).toBe(false)
  276. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:take.body.controlId,token:take.body.token,notes:'已经登录'}).expect(200)
  277. expect(await paused).toMatchObject({humanNote:'已经登录',result:{ok:true}})
  278. expect(f.calls.filter(call=>call.op==='open')).toHaveLength(2)
  279. expect(f.calls.some(call=>call.op==='close')).toBe(false)
  280. const action=f.calls.find(call=>call.operation?.action.text==='continue')
  281. expect(action.scope.environmentId).toBe(f.agent.id)
  282. expect(action.operation.generation).toBeGreaterThan(take.body.generation)
  283. }finally{await f.close()}
  284. })
  285. it('cancels a task waiting on human control without closing the human browser',async()=>{
  286. const f=fixture()
  287. try{
  288. const task=f.turn()
  289. await task.value.call('managed_browser_open',{},'open')
  290. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  291. const waiting=task.value.call('managed_browser_snapshot',{},'waiting')
  292. const rejected=expect(waiting).rejects.toMatchObject({code:'browser_authorization_revoked'})
  293. task.controller.abort();await rejected
  294. expect(f.calls.some(call=>call.op==='close')).toBe(false)
  295. const status=await request(f.app.callback()).get(controlURL(f.base)).expect(200)
  296. expect(status.body).toMatchObject({mode:'human',canResume:false,controlId:take.body.controlId})
  297. }finally{await f.close()}
  298. })
  299. it('assembles verified download chunks once and publishes through the existing artifact callback',async()=>{
  300. const f=fixture(),bytes=Buffer.alloc(512*1024+17,7),id=randomUUID()
  301. f.setDownload(({offset})=>({size:bytes.length,sha256:digest(bytes),name:'report.csv',mimeType:'text/csv',offset,data:bytes.subarray(offset,offset+512*1024).toString('base64')}))
  302. try{
  303. const task=f.turn()
  304. const first=await task.value.call('managed_browser_export',{downloadId:id},'download')
  305. const again=await task.value.call('managed_browser_export',{downloadId:id},'download')
  306. expect(again).toEqual(first)
  307. expect(task.publish).toHaveBeenCalledExactlyOnceWith('report.csv',bytes)
  308. expect(f.calls.filter(call=>call.op==='download').map(call=>call.offset)).toEqual([0,512*1024])
  309. }finally{await f.close()}
  310. })
  311. it.each(['offset','hash','size','base64','changed'] as const)('rejects %s corruption before exporting a download',async(problem)=>{
  312. const f=fixture(),bytes=Buffer.alloc(512*1024+1,3)
  313. f.setDownload(({offset})=>({size:problem==='size'?26*1024*1024:bytes.length,sha256:problem==='hash'?'0'.repeat(64):digest(bytes),name:problem==='changed'&&offset?'changed.txt':'file.txt',mimeType:'text/plain',offset:problem==='offset'?offset+1:offset,data:problem==='base64'?'%%%':bytes.subarray(offset,offset+512*1024).toString('base64')}))
  314. try{
  315. const task=f.turn()
  316. await expect(task.value.call('managed_browser_export',{downloadId:randomUUID()},'bad-download')).rejects.toMatchObject({code:problem==='changed'?'browser_download_changed':'browser_download_invalid'})
  317. expect(task.publish).not.toHaveBeenCalled()
  318. }finally{await f.close()}
  319. })
  320. it('resolves uploads only through the authorized attachment callback and refuses host paths',async()=>{
  321. const f=fixture()
  322. try{
  323. const snapshotId=randomUUID(),fileId=randomUUID(),upload=vi.fn(async(id:string)=>{if(id!==fileId)throw new HttpError(403,'附件未授权','file_forbidden');return {name:'allowed.txt',mimeType:'text/plain',buffer:Buffer.from('owned')}})
  324. const task=f.turn({upload})
  325. await task.value.call('managed_browser_upload',{fileId,snapshotId,ref:'e1'},'upload')
  326. expect(upload).toHaveBeenCalledExactlyOnceWith(fileId)
  327. const call=f.calls.find(call=>call.upload)
  328. expect(call.upload.data).toBe(Buffer.from('owned').toString('base64'))
  329. expect(call.operation.action).toMatchObject({kind:'upload',snapshotId,ref:'e1',fileId:call.upload.id})
  330. await expect(task.value.call('managed_browser_upload',{fileId,path:'/etc/passwd',snapshotId,ref:'e1'},'host-path')).rejects.toMatchObject({code:'browser_upload_invalid'})
  331. await expect(task.value.call('managed_browser_upload',{fileId:randomUUID(),snapshotId,ref:'e1'},'other-file')).rejects.toMatchObject({code:'file_forbidden'})
  332. await expect(task.value.call('managed_browser_upload_vm',{path:'/workspace/input.txt',snapshotId,ref:'e1'},'vm-not-granted')).rejects.toMatchObject({code:'browser_vm_unavailable'})
  333. expect(f.calls.filter(call=>call.upload)).toHaveLength(1)
  334. }finally{await f.close()}
  335. })
  336. it('allows a newly authorized profile after the prior browser task has closed',async()=>{
  337. const f=fixture()
  338. try{
  339. const first=f.turn();await first.value.call('managed_browser_open',{},'first');await first.value.close()
  340. const updated=f.store.updateAgent('owner',f.agent.id,{profile:'writer'})
  341. const second=f.turn({},'owner',updated)
  342. await expect(second.value.call('managed_browser_open',{},'second')).resolves.toMatchObject({open:true})
  343. expect((await f.service.state('owner',f.agent.id)).available).toBe(true)
  344. }finally{await f.close()}
  345. })
  346. it.each(['profile','runner'] as const)('recovers an active browser after its %s changes and fences the old task and human grant',async(source)=>{
  347. const f=fixture()
  348. try{
  349. const cards:ManagedBrowserCard[]=[]
  350. const first=f.turn({onCard:card=>{cards.push(card)}});await first.value.call('managed_browser_open',{},'first-open')
  351. const oldRunner=f.runner(),oldTask=f.calls[0]
  352. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  353. const credentials={controlId:take.body.controlId,token:take.body.token}
  354. const updated=source==='profile'?f.store.updateAgent('owner',f.agent.id,{profile:'writer'}):f.agent
  355. if(source==='runner')f.setRunner(randomUUID())
  356. expect(f.service.allowed(oldRunner,oldTask.scope,oldTask.grantId)).toBe(false)
  357. expect(f.service.allowed(oldRunner,oldTask.scope,credentials.controlId)).toBe(false)
  358. await expect(first.value.call('managed_browser_state',{},'old-task')).rejects.toMatchObject({status:410})
  359. await request(f.app.callback()).post(controlURL(f.base,'/renew')).send(credentials).expect(410)
  360. const start=f.calls.length,second=f.turn({},'owner',updated)
  361. await expect(second.value.call('managed_browser_open',{},'new-open')).resolves.toMatchObject({open:true})
  362. expect(cards.at(-1)).toMatchObject({status:'closed',message:'浏览器执行来源已改变,本轮会话已结束'})
  363. const replacement=f.calls.slice(start)
  364. expect(replacement.map(call=>call.op)).toEqual(['close','status','open','status'])
  365. expect(replacement.every(call=>call.runnerId===f.runner())).toBe(true)
  366. expect(replacement.every(call=>call.grantId!==oldTask.grantId&&call.grantId!==credentials.controlId)).toBe(true)
  367. await request(f.app.callback()).post(controlURL(f.base,'/renew')).send(credentials).expect(403)
  368. await expect(first.value.call('managed_browser_state',{},'old-task-after-reset')).rejects.toMatchObject({status:410})
  369. // Finishing the retired task must not close the replacement's same-Bot
  370. // context, even when the browser scope stayed identical on one Runner.
  371. const beforeOldClose=f.calls.length;await first.value.close()
  372. expect(f.calls).toHaveLength(beforeOldClose)
  373. expect((await f.service.state('owner',f.agent.id)).open).toBe(true)
  374. const nextTake=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  375. expect(nextTake.body.controlId).not.toBe(credentials.controlId)
  376. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:nextTake.body.controlId,token:nextTake.body.token}).expect(200)
  377. await expect(second.value.call('managed_browser_state',{},'new-task-after-human')).resolves.toMatchObject({open:true})
  378. // Returning the routing configuration to its old value must not revive a
  379. // grant belonging to the replaced Resource object.
  380. if(source==='profile')f.store.updateAgent('owner',f.agent.id,{profile:'default'})
  381. else f.setRunner(oldRunner)
  382. expect(f.service.allowed(oldRunner,oldTask.scope,oldTask.grantId)).toBe(false)
  383. expect(f.service.allowed(oldRunner,oldTask.scope,credentials.controlId)).toBe(false)
  384. }finally{await f.close()}
  385. })
  386. it('reads owned file IDs with the real upload store and rechecks their source authorization',async()=>{
  387. const f=fixture(),uploads=new UploadStore(f.home),id=randomUUID(),bytes=Buffer.from('private attachment')
  388. const tempPath=join(uploads.uploadRoot,'staged'),path=join(uploads.uploadRoot,id)
  389. writeFileSync(tempPath,bytes)
  390. uploads.commit([{id,name:'owned.txt',mimeType:'text/plain',size:bytes.length,path,tempPath,accountKey:'owner',referenced:false}])
  391. try{
  392. expect((await browserOwnedFile(f.store,f.nodes,uploads,'owner',id)).buffer).toEqual(bytes)
  393. await expect(browserOwnedFile(f.store,f.nodes,uploads,'other',id)).rejects.toMatchObject({code:'upload_not_found'})
  394. await expect(browserOwnedFile(f.store,f.nodes,uploads,'owner',path)).rejects.toMatchObject({code:'invalid_upload'})
  395. const artifactId=randomUUID(),artifactPath=join(f.home,'artifact.txt')
  396. writeFileSync(artifactPath,'generated')
  397. f.store.put('owner','file',artifactId,{id:artifactId,name:'artifact.txt',mimeType:'text/plain',size:9,path:artifactPath,sender:'agent',messageFileSource:'attachment',sourceNodeId:'local',profile:'writer',createdAt:Date.now()})
  398. expect((await browserOwnedFile(f.store,f.nodes,uploads,'owner',artifactId)).buffer.toString()).toBe('generated')
  399. f.blockedSources.add('owner:local:writer')
  400. await expect(browserOwnedFile(f.store,f.nodes,uploads,'owner',artifactId)).rejects.toMatchObject({code:'agent_source_forbidden'})
  401. const symlinkId=randomUUID(),symlinkPath=join(f.home,'linked.txt');symlinkSync(artifactPath,symlinkPath)
  402. f.store.put('owner','file',symlinkId,{id:symlinkId,name:'linked.txt',mimeType:'text/plain',size:9,path:symlinkPath,sender:'user',createdAt:Date.now()})
  403. await expect(browserOwnedFile(f.store,f.nodes,uploads,'owner',symlinkId)).rejects.toThrow()
  404. }finally{uploads.close();await f.close()}
  405. })
  406. it('copies browser bytes through the checked VM endpoint with overwrite and root confinement',async()=>{
  407. const f=fixture(),directory=join(f.home,'vm');mkdirSync(directory)
  408. const endpoint=new FileTransferFiles(directory),path=join(directory,'file.csv'),bytes=Buffer.alloc(512*1024+7,9),file={name:'file.csv',mimeType:'text/csv',buffer:bytes}
  409. const transfer=(action:Record<string,unknown>)=>endpoint.call(action)
  410. try{
  411. expect(await browserVmFile(transfer,path,()=>{},file)).toMatchObject({copied:true,target:{host:'vm',path}})
  412. expect(await readFile(path)).toEqual(bytes)
  413. await expect(browserVmFile(transfer,path,()=>{},file)).rejects.toThrow()
  414. expect((await browserVmFile(transfer,path,()=>{})) as any).toMatchObject({name:'file.csv',mimeType:'text/csv',buffer:bytes})
  415. await expect(browserVmFile(transfer,'../host-file',()=>{},file)).rejects.toMatchObject({code:'browser_vm_path_invalid'})
  416. await expect(browserVmFile(transfer,join(f.home,'outside.txt'),()=>{},file)).rejects.toThrow('路径')
  417. await browserVmFile(transfer,path,()=>{},{...file,buffer:Buffer.from('updated')},true)
  418. expect((await readFile(path)).toString()).toBe('updated')
  419. }finally{await endpoint.close();await f.close()}
  420. })
  421. it('prepares a missing environment without granting browser execution, and releases its installation grant when ready',async()=>{
  422. const f=fixture()
  423. try{
  424. f.setInstallation('missing')
  425. expect((await request(f.app.callback()).get(f.base+'/managed-browser').expect(200)).body).toMatchObject({enabled:true,available:false,installation:{status:'missing'}})
  426. expect(f.calls.some(call=>call.op==='prepare')).toBe(false)
  427. await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(409)
  428. expect(f.calls.some(call=>call.op==='open'||call.op==='pause')).toBe(false)
  429. const first=await request(f.app.callback()).post(f.base+'/managed-browser/prepare').send({}).expect(200)
  430. expect(first.body).toMatchObject({available:false,installation:{status:'installing'}})
  431. const setup=f.calls.find(call=>call.op==='prepare')
  432. expect(f.service.allowed(f.runner(),setup.scope,setup.grantId)).toBe(true)
  433. expect(f.service.allowed(f.runner(),setup.scope)).toBe(false)
  434. await request(f.app.callback()).post(f.base+'/managed-browser/prepare').send({}).expect(200)
  435. expect(f.calls.filter(call=>call.op==='prepare').every(call=>call.grantId===setup.grantId)).toBe(true)
  436. f.setInstallation('ready')
  437. expect((await request(f.app.callback()).get(f.base+'/managed-browser').expect(200)).body.available).toBe(true)
  438. expect(f.service.allowed(f.runner(),setup.scope,setup.grantId)).toBe(false)
  439. await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  440. }finally{await f.close()}
  441. })
  442. it('requires explicit installation retry and applies current owner/source permissions to preparation',async()=>{
  443. const f=fixture()
  444. try{
  445. f.setInstallation('failed','下载失败','网络不可用')
  446. expect((await request(f.app.callback()).post(f.base+'/managed-browser/prepare').send({}).expect(200)).body.installation.status).toBe('failed')
  447. const failed=f.calls.find(call=>call.op==='prepare')
  448. expect(f.service.allowed(f.runner(),failed.scope,failed.grantId)).toBe(false)
  449. expect((await request(f.app.callback()).post(f.base+'/managed-browser/prepare').send({retry:true}).expect(200)).body.installation.status).toBe('installing')
  450. const retry=f.calls.at(-1)
  451. expect(retry.retry).toBe(true)
  452. expect(f.service.allowed(f.runner(),retry.scope,retry.grantId)).toBe(true)
  453. await request(f.app.callback()).post(f.base+'/managed-browser/prepare').set('x-fixture-owner','other').send({retry:true}).expect(404)
  454. f.blockedSources.add('owner:local:default')
  455. expect(f.service.allowed(f.runner(),retry.scope,retry.grantId)).toBe(false)
  456. await request(f.app.callback()).post(f.base+'/managed-browser/prepare').send({retry:true}).expect(403)
  457. }finally{await f.close()}
  458. })
  459. it('publishes preparation and same-session takeover cards, preserving safe page history after completion',async()=>{
  460. const f=fixture(),cards:ManagedBrowserCard[]=[],preparing=defer()
  461. try{
  462. f.setInstallation('missing')
  463. const task=f.turn({onCard:card=>{cards.push(structuredClone(card))}})
  464. f.setBeforeReply(async payload=>{if(payload.op==='prepare')preparing.resolve()})
  465. const opening=task.value.call('managed_browser_open',{url:'https://user:secret@example.com/page?token=private#password'},'open')
  466. await preparing.promise
  467. await vi.waitFor(()=>expect(cards.at(-1)).toMatchObject({status:'preparing',installation:{status:'installing'}}))
  468. expect(f.calls.some(call=>call.op==='open')).toBe(false)
  469. f.setInstallation('ready')
  470. await opening
  471. const active=cards.at(-1)!
  472. expect(active).toMatchObject({agentId:f.agent.id,status:'active',title:'Example',url:'https://example.com/page'})
  473. expect(JSON.stringify(cards)).not.toMatch(/secret|private|password/)
  474. const taskGrant=f.calls.find(call=>call.op==='prepare')
  475. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  476. expect(cards.at(-1)?.message).toContain('人工控制中')
  477. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:take.body.controlId,token:take.body.token}).expect(200)
  478. const resume=f.calls.find(call=>call.op==='resume')
  479. expect(resume.grantId).toBe(taskGrant.grantId)
  480. expect(f.service.allowed(f.runner(),resume.scope,resume.grantId)).toBe(true)
  481. await task.value.close()
  482. expect(cards.at(-1)).toMatchObject({id:active.id,status:'idle',title:'Example',url:'https://example.com/page'})
  483. const takeAgain=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  484. expect(cards.at(-1)).toMatchObject({status:'active',message:'人工控制中,交还后保留页面'})
  485. await request(f.app.callback()).post(f.base+'/managed-browser/close').send({controlId:takeAgain.body.controlId,token:takeAgain.body.token}).expect(200)
  486. expect(cards.at(-1)).toMatchObject({status:'closed',url:'https://example.com/page'})
  487. }finally{await f.close()}
  488. })
  489. it('reports installation failure on the card without opening a browser or silently reinstalling',async()=>{
  490. const f=fixture(),cards:ManagedBrowserCard[]=[]
  491. try{
  492. f.setInstallation('failed','浏览器下载失败','无法连接下载服务')
  493. const task=f.turn({onCard:card=>{cards.push(card)}})
  494. await expect(task.value.call('managed_browser_open',{},'open')).rejects.toMatchObject({code:'browser_install_failed'})
  495. expect(cards.at(-1)).toMatchObject({status:'failed',message:'无法连接下载服务',installation:{status:'failed'}})
  496. expect(f.calls.find(call=>call.op==='prepare')).not.toHaveProperty('retry',true)
  497. expect(f.calls.some(call=>call.op==='open')).toBe(false)
  498. }finally{await f.close()}
  499. })
  500. it('revokes an installing task immediately on cancellation, without late opening',async()=>{
  501. const f=fixture(),prepared=defer()
  502. try{
  503. f.setInstallation('missing')
  504. const task=f.turn()
  505. f.setBeforeReply(async payload=>{if(payload.op==='prepare')prepared.resolve()})
  506. const opening=task.value.call('managed_browser_open',{},'open')
  507. const rejected=expect(opening).rejects.toMatchObject({code:'browser_authorization_revoked'})
  508. await prepared.promise
  509. task.controller.abort()
  510. await rejected
  511. const setup=f.calls.find(call=>call.op==='prepare')
  512. expect(f.service.allowed(f.runner(),setup.scope,setup.grantId)).toBe(false)
  513. f.setInstallation('ready')
  514. expect(f.calls.some(call=>call.op==='open')).toBe(false)
  515. }finally{await f.close()}
  516. })
  517. it('retains the card through a task ending under human control and parks it after giveback',async()=>{
  518. const f=fixture(),cards:ManagedBrowserCard[]=[]
  519. try{
  520. const task=f.turn({onCard:card=>{cards.push(card)}})
  521. await task.value.call('managed_browser_open',{url:'https://example.com/page'},'open')
  522. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  523. await task.value.close()
  524. expect(cards.at(-1)).toMatchObject({status:'active',message:'人工控制中,交还后保留页面'})
  525. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:take.body.controlId,token:take.body.token}).expect(200)
  526. expect(cards.at(-1)).toMatchObject({status:'idle',url:'https://example.com/page'})
  527. }finally{await f.close()}
  528. })
  529. it('keeps browser handoff and authorization cleanup working when the card observer throws',async()=>{
  530. const f=fixture(),observe=vi.fn(()=>{throw new Error('chat history unavailable')})
  531. try{
  532. const task=f.turn({onCard:observe})
  533. await expect(task.value.call('managed_browser_open',{url:'https://example.com/page'},'open')).resolves.toBeDefined()
  534. const taskGrant=f.calls.find(call=>call.op==='open')
  535. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  536. await expect(task.value.close()).resolves.toBeUndefined()
  537. expect(f.service.allowed(f.runner(),taskGrant.scope,taskGrant.grantId)).toBe(false)
  538. await expect((f.service as any).sweep()).resolves.toBeUndefined()
  539. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:take.body.controlId,token:take.body.token}).expect(200)
  540. expect(f.service.allowed(f.runner(),taskGrant.scope)).toBe(true)
  541. expect(f.calls.at(-1).op).toBe('park')
  542. expect(observe).toHaveBeenCalled()
  543. }finally{await f.close()}
  544. })
  545. it('retries an unchanged card after its persistence observer reports a failed save',async()=>{
  546. const f=fixture(),observe=vi.fn(()=>false)
  547. try{
  548. const task=f.turn({onCard:observe})
  549. await task.value.call('managed_browser_state',{},'state')
  550. const calls=observe.mock.calls.length
  551. await task.value.call('managed_browser_state',{},'state-again')
  552. expect(observe.mock.calls.length).toBeGreaterThan(calls)
  553. observe.mockReturnValue(true)
  554. await task.value.call('managed_browser_state',{},'state-saved')
  555. const saved=observe.mock.calls.length
  556. await task.value.call('managed_browser_state',{},'unchanged')
  557. expect(observe).toHaveBeenCalledTimes(saved)
  558. }finally{await f.close()}
  559. })
  560. it('retains tabs across task boundaries without retaining old task authority or waking on status',async()=>{
  561. const f=fixture(),cards:ManagedBrowserCard[]=[]
  562. try{
  563. const first=f.turn({onCard:card=>{cards.push(card)}})
  564. const opened=await first.value.call('managed_browser_open',{url:'https://example.com/session'},'first') as BrowserState
  565. const original=f.calls.find(call=>call.op==='open')
  566. await first.value.close()
  567. const park=f.calls.at(-1)
  568. expect(park.op).toBe('park');expect(park.grantId).not.toBe(original.grantId)
  569. expect(f.service.allowed(f.runner(),original.scope,original.grantId)).toBe(false)
  570. expect(f.service.allowed(f.runner(),park.scope,park.grantId)).toBe(false)
  571. expect(f.service.allowed(f.runner(),park.scope)).toBe(true)
  572. expect(cards.at(-1)).toMatchObject({status:'idle',url:'https://example.com/session'})
  573. const statusTurn=f.turn(),start=f.calls.length
  574. expect(await statusTurn.value.call('managed_browser_state',{},'read')).toMatchObject({open:true,tabs:opened.tabs})
  575. await statusTurn.value.close()
  576. expect(f.calls.slice(start).map(call=>call.op)).toEqual(['status'])
  577. const next=f.turn(),before=f.calls.length
  578. await next.value.call('managed_browser_snapshot',{},'snapshot')
  579. expect(f.calls.slice(before).map(call=>call.op)).toEqual(['resume','execute'])
  580. expect(f.calls.at(-1).operation.generation).toBeGreaterThan(opened.generation)
  581. expect(f.calls.at(-1).grantId).not.toBe(park.grantId)
  582. await expect(first.value.call('managed_browser_state',{},'old')).rejects.toMatchObject({code:'browser_authorization_revoked'})
  583. expect((await f.service.state('owner',f.agent.id)).tabs).toEqual(opened.tabs)
  584. }finally{await f.close()}
  585. })
  586. it.each(['temporary','legacy','cancel'] as const)('recycles %s sessions instead of claiming they were retained',async kind=>{
  587. const f=fixture(),cards:ManagedBrowserCard[]=[]
  588. try{
  589. if(kind==='legacy')f.setRetention(false)
  590. const agent=kind==='temporary'?{...f.agent,temporaryGoalId:randomUUID()}:f.agent
  591. if(kind==='temporary')f.store.put('owner','agent',agent.id,agent)
  592. const task=f.turn({onCard:card=>{cards.push(card)}},'owner',agent)
  593. await task.value.call('managed_browser_open',{},'open')
  594. const opening=f.calls.find(call=>call.op==='open')
  595. await task.value.close({retain:kind!=='cancel'})
  596. expect(f.calls.at(-1).op).toBe('close')
  597. expect(f.calls.some(call=>call.op==='park')).toBe(false)
  598. expect(f.service.allowed(f.runner(),opening.scope)).toBe(false)
  599. expect(cards.at(-1)?.status).toBe('closed')
  600. if(kind==='legacy')expect((await f.service.state('owner',agent.id)).supportsRetention).toBe(false)
  601. }finally{await f.close()}
  602. })
  603. it('keeps an independently controlled browser on task cancellation, then closes on giveback',async()=>{
  604. const f=fixture()
  605. try{
  606. const task=f.turn();await task.value.call('managed_browser_open',{},'open')
  607. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  608. const body={controlId:take.body.controlId,token:take.body.token},scope=f.calls[0].scope
  609. task.controller.abort();await task.value.close()
  610. expect(f.service.allowed(f.runner(),scope,take.body.controlId)).toBe(true)
  611. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send(body).expect(200)
  612. expect(f.calls.at(-1).op).toBe('close');expect(f.service.allowed(f.runner(),scope)).toBe(false)
  613. }finally{await f.close()}
  614. })
  615. it('allows explicit close only for the current human holder after the task has ended',async()=>{
  616. const f=fixture()
  617. try{
  618. const task=f.turn();await task.value.call('managed_browser_open',{},'open')
  619. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  620. const body={controlId:take.body.controlId,token:take.body.token},scope=f.calls[0].scope
  621. await request(f.app.callback()).post(f.base+'/managed-browser/close').send(body).expect(409)
  622. await task.value.close()
  623. await request(f.app.callback()).post(f.base+'/managed-browser/close').send({...body,token:'x'.repeat(43)}).expect(403)
  624. await request(f.app.callback()).post(f.base+'/managed-browser/close').set('x-fixture-owner','other').send(body).expect(403)
  625. const other=f.store.createAgent('owner',{name:'另一个 Bot',profile:'default'})
  626. await request(f.app.callback()).post(`/api/app/agents/${other.id}/managed-browser/close`).send(body).expect(403)
  627. const result=await request(f.app.callback()).post(f.base+'/managed-browser/close').send(body).expect(200)
  628. expect(result.body).toMatchObject({mode:'off',canResume:false})
  629. expect(result.body).not.toHaveProperty('controlId')
  630. expect(f.service.allowed(f.runner(),scope)).toBe(false)
  631. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send(body).expect(403)
  632. }finally{await f.close()}
  633. })
  634. it.each([false,true])('handles expired human control with task present=%s without reviving its authority',async active=>{
  635. const f=fixture()
  636. try{
  637. const task=active?f.turn():undefined
  638. if(task)await task.value.call('managed_browser_open',{},'open')
  639. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200),scope=f.calls[0].scope
  640. vi.spyOn(Date,'now').mockReturnValue(take.body.expiresAt+1)
  641. // The retained scope bridges expiry and the next server sweep; it grants no input.
  642. expect(f.service.allowed(f.runner(),scope)).toBe(true)
  643. expect(f.service.allowed(f.runner(),scope,take.body.controlId)).toBe(false)
  644. await (f.service as any).sweep()
  645. const status=await request(f.app.callback()).get(controlURL(f.base)).expect(200)
  646. expect(status.body.mode).toBe(active?'pausing':'idle')
  647. expect(f.calls.some(call=>call.op==='park')).toBe(!active)
  648. expect(f.calls.some(call=>call.op==='resume')).toBe(false)
  649. expect((await f.service.state('owner',f.agent.id)).open).toBe(true)
  650. }finally{await f.close()}
  651. })
  652. it.each(['owner','logout','source','global'] as const)('revokes retained sessions and human authority when %s access is removed',async reason=>{
  653. const f=fixture(),cards:ManagedBrowserCard[]=[]
  654. try{
  655. const task=f.turn({onCard:card=>{cards.push(card)}});await task.value.call('managed_browser_open',{},'open');await task.value.close()
  656. const scope=f.calls[0].scope
  657. if(reason==='owner')f.inactiveOwners.add('owner')
  658. else if(reason==='logout')f.versions.set('owner',2)
  659. else if(reason==='source')f.blockedSources.add('owner:local:default')
  660. else saveHostTools(f.home,{managedBrowser:false})
  661. expect(f.service.allowed(f.runner(),scope)).toBe(false)
  662. await (f.service as any).sweep()
  663. expect(cards.at(-1)?.status).toBe('closed')
  664. expect(f.calls.filter(call=>call.op==='park')).toHaveLength(1)
  665. }finally{await f.close()}
  666. })
  667. it('falls back to confirmed close when parking fails',async()=>{
  668. const f=fixture(),cards:ManagedBrowserCard[]=[]
  669. try{
  670. const task=f.turn({onCard:card=>{cards.push(card)}});await task.value.call('managed_browser_open',{},'open')
  671. f.setBeforeReply(async payload=>{if(payload.op==='park')throw new Error('park failed')})
  672. await task.value.close()
  673. expect(f.calls.slice(-2).map(call=>call.op)).toEqual(['park','close'])
  674. expect(f.service.allowed(f.runner(),f.calls[0].scope)).toBe(false)
  675. expect(cards.at(-1)?.status).toBe('closed')
  676. }finally{await f.close()}
  677. })
  678. it('updates the retained card and drops its grant after Runner idle reclamation',async()=>{
  679. const f=fixture(),cards:ManagedBrowserCard[]=[]
  680. try{
  681. const task=f.turn({onCard:card=>{cards.push(card)}});await task.value.call('managed_browser_open',{},'open');await task.value.close()
  682. const scope=f.calls[0].scope,start=f.calls.length
  683. f.runnerClosed(scope)
  684. vi.spyOn(Date,'now').mockReturnValue(Date.now()+31000)
  685. await (f.service as any).sweep()
  686. await vi.waitFor(()=>expect(cards.at(-1)?.status).toBe('closed'))
  687. expect(f.calls.slice(start).map(call=>call.op)).toEqual(['status'])
  688. expect(f.service.allowed(f.runner(),scope)).toBe(false)
  689. }finally{await f.close()}
  690. })
  691. it('parks a status-only turn when its human holder expired before normal completion',async()=>{
  692. const f=fixture()
  693. try{
  694. const first=f.turn();await first.value.call('managed_browser_open',{},'open');await first.value.close()
  695. const read=f.turn();await read.value.call('managed_browser_state',{},'read')
  696. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  697. vi.spyOn(Date,'now').mockReturnValue(take.body.expiresAt+1)
  698. await read.value.close()
  699. expect(f.calls.filter(call=>call.op==='park')).toHaveLength(2)
  700. expect(f.calls.at(-1).op).toBe('park')
  701. expect((await request(f.app.callback()).get(controlURL(f.base)).expect(200)).body.mode).toBe('idle')
  702. }finally{await f.close()}
  703. })
  704. it.each(['close','giveback'] as const)('serializes %s against the competing release action and a new takeover',async firstAction=>{
  705. const f=fixture(),entered=defer(),gate=defer()
  706. try{
  707. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  708. const credentials={controlId:take.body.controlId,token:take.body.token}
  709. const url=(action:'close'|'giveback')=>action==='close'?f.base+'/managed-browser/close':controlURL(f.base,'/giveback')
  710. f.setBeforeReply(async payload=>{if(payload.op===(firstAction==='close'?'close':'park')){entered.resolve();await gate.promise}})
  711. const first=request(f.app.callback()).post(url(firstAction)).send(credentials).then(value=>value)
  712. await entered.promise
  713. const other=request(f.app.callback()).post(url(firstAction==='close'?'giveback':'close')).send(credentials).then(value=>value)
  714. let taken=false
  715. const next=request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).then(value=>{taken=true;return value})
  716. await new Promise(resolve=>setTimeout(resolve,20));expect(taken).toBe(false)
  717. gate.resolve()
  718. expect((await first).status).toBe(200);expect((await other).status).toBe(403)
  719. const current=await next
  720. expect(current.status).toBe(200);expect(current.body.controlId).not.toBe(credentials.controlId)
  721. expect(f.calls.at(-1).op).toBe('open')
  722. }finally{gate.resolve();await f.close()}
  723. })
  724. it('waits for a background retained-state check before admitting the next task',async()=>{
  725. const f=fixture(),entered=defer(),gate=defer()
  726. try{
  727. const first=f.turn();await first.value.call('managed_browser_open',{},'open');await first.value.close()
  728. f.setBeforeReply(async payload=>{if(payload.op==='status'){entered.resolve();await gate.promise}})
  729. vi.spyOn(Date,'now').mockReturnValue(Date.now()+31000)
  730. await (f.service as any).sweep();await entered.promise
  731. const next=f.turn(),pending=next.value.call('managed_browser_snapshot',{},'next')
  732. gate.resolve()
  733. await expect(pending).resolves.toMatchObject({ok:true})
  734. expect(f.calls.slice(-2).map(call=>call.op)).toEqual(['resume','execute'])
  735. }finally{gate.resolve();await f.close()}
  736. })
  737. it('ignores a late closed-state reply instead of deleting the newly opened retained session',async()=>{
  738. const f=fixture(),entered=defer(),gate=defer();let held=false
  739. try{
  740. f.setAfterReply(async payload=>{if(payload.op==='status'&&!held){held=true;entered.resolve();await gate.promise}})
  741. const oldStatus=f.service.state('owner',f.agent.id);await entered.promise
  742. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  743. gate.resolve();await oldStatus
  744. const current=await f.service.state('owner',f.agent.id)
  745. expect(current).toMatchObject({open:true,generation:take.body.generation})
  746. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:take.body.controlId,token:take.body.token}).expect(200)
  747. expect(f.calls.at(-1).op).toBe('park')
  748. expect(f.service.allowed(f.runner(),f.calls[0].scope)).toBe(true)
  749. }finally{gate.resolve();await f.close()}
  750. })
  751. it('accepts a restarted Runner generation after fencing the previous connection epoch',async()=>{
  752. const f=fixture()
  753. try{
  754. const old=f.turn();await old.value.call('managed_browser_open',{},'old')
  755. const taskGrant=f.calls.find(call=>call.op==='open')
  756. const earlier=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  757. await request(f.app.callback()).post(controlURL(f.base,'/giveback')).send({controlId:earlier.body.controlId,token:earlier.body.token}).expect(200)
  758. const take=await request(f.app.callback()).post(controlURL(f.base,'/take')).send({requestId:randomUUID()}).expect(200)
  759. await old.value.close()
  760. f.setEpoch(randomUUID())
  761. expect(f.service.allowed(f.runner(),taskGrant.scope,taskGrant.grantId)).toBe(false)
  762. expect(f.service.allowed(f.runner(),taskGrant.scope,take.body.controlId)).toBe(false)
  763. const start=f.calls.length,newTurn=f.turn(),state=await newTurn.value.call('managed_browser_open',{},'new') as BrowserState
  764. expect(f.calls.slice(start).map(call=>call.op)).toEqual(['close','status','open','status'])
  765. expect(state.open).toBe(true);expect(state.generation).toBeLessThan(take.body.generation)
  766. await newTurn.value.close()
  767. expect((await f.service.state('owner',f.agent.id)).open).toBe(true)
  768. }finally{await f.close()}
  769. })