fixture.client.spec.ts 78 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748
  1. import { afterEach, describe, expect, it, vi } from 'vitest'
  2. import type {
  3. RpcRequest,
  4. RpcResponse,
  5. RpcResult,
  6. SessionEvent,
  7. SessionId,
  8. } from '../src/client/api.ts'
  9. import { RpcId } from '../src/client/api.ts'
  10. import { decodeStorageRecord } from '@deepseek-ai/dsh-session/chunk-rows'
  11. import type { ChunkRow } from '@deepseek-ai/dsh-session/chunk-rows'
  12. import {
  13. createFixtureConnectionRpc,
  14. createFixtureFaces,
  15. type FixtureOptions,
  16. } from '../src/client/fixture.ts'
  17. import type {
  18. ClientConnectionRpc, ConnectionRpcResult,
  19. } from '../src/rpc.ts'
  20. import type { DirectoryListing } from '@deepseek-ai/dsh-host-directory-picker/types'
  21. import type { ModelCatalog } from '@deepseek-ai/dsh-api-session-controller/types'
  22. import type { ModelSelection } from '@deepseek-ai/dsh-api-session-controller/types'
  23. const sid = (id: string): SessionId => id as SessionId
  24. type WorkspaceId = string & { readonly __fixtureWorkspaceId: 'WorkspaceId' }
  25. const req = <P>(payload: P): RpcRequest<P> => ({ rpcId: RpcId(`t-${Math.abs(Math.sin(reqCount++)).toString(36).slice(2, 10)}`), payload })
  26. let reqCount = 0
  27. interface FixtureSessionSummary {
  28. sessionId: SessionId
  29. updatedAt: number
  30. running: boolean
  31. blank: boolean
  32. parentSessionId?: SessionId
  33. origin?: 'subagent'
  34. cwd?: string
  35. agentPreset?: string
  36. }
  37. interface FixtureHistoryEntry {
  38. readonly type: 'event'
  39. readonly event: SessionEvent
  40. }
  41. type FixtureChunkRowEvent = {
  42. [Kind in ChunkRow['type']]: {
  43. readonly type: `chunkrow/${Kind}`
  44. readonly seq: number
  45. readonly time: number
  46. readonly data: Extract<ChunkRow, { readonly type: Kind }>['data']
  47. }
  48. }[ChunkRow['type']]
  49. interface FixtureHistoryChunkRun {
  50. readonly type: 'chunks'
  51. readonly event: FixtureChunkRowEvent
  52. }
  53. type FixtureHistoryRecord = FixtureHistoryEntry | FixtureHistoryChunkRun
  54. interface FixturePage {
  55. readonly records: readonly FixtureHistoryRecord[]
  56. readonly hasMore: boolean
  57. }
  58. function historyEvents(records: readonly FixtureHistoryRecord[]): SessionEvent[] {
  59. return records.flatMap(record => record.type === 'event'
  60. ? [record.event]
  61. : decodeStorageRecord(chunkRow(record.event)))
  62. }
  63. function chunkRow(event: FixtureChunkRowEvent): ChunkRow {
  64. switch (event.type) {
  65. case 'chunkrow/text-chunks':
  66. return { type: 'text-chunks', seq0: event.seq, time0: event.time, data: event.data }
  67. case 'chunkrow/reasoning-chunks':
  68. return { type: 'reasoning-chunks', seq0: event.seq, time0: event.time, data: event.data }
  69. case 'chunkrow/tool-call-chunks':
  70. return { type: 'tool-call-chunks', seq0: event.seq, time0: event.time, data: event.data }
  71. }
  72. }
  73. type FixtureFollowFrame =
  74. | {
  75. readonly type: 'snapshot'
  76. readonly cursor: number
  77. readonly records: readonly FixtureHistoryRecord[]
  78. readonly hasMore: boolean
  79. readonly projections: {
  80. readonly asOfSeq: number
  81. readonly values: Readonly<Record<string, unknown>>
  82. }
  83. }
  84. | FixtureHistoryEntry
  85. type FixtureControlFrame =
  86. | {
  87. readonly type: 'baseline'
  88. readonly value: {
  89. readonly queues: Readonly<Record<string, readonly unknown[]>>
  90. readonly jobs: Readonly<Record<string, readonly unknown[]>>
  91. readonly approvals: readonly unknown[]
  92. readonly questions: readonly unknown[]
  93. readonly projections: Readonly<Record<string, {
  94. readonly asOfSeq: number
  95. readonly values: Readonly<Record<string, unknown>>
  96. }>>
  97. }
  98. }
  99. | {
  100. readonly type: 'projection'
  101. readonly sessionId: SessionId
  102. readonly key: string
  103. readonly value: unknown
  104. readonly seq: number
  105. }
  106. interface FixtureSessionRequests {
  107. list: { readonly cursor?: string }
  108. search: { readonly query: string }
  109. create: {
  110. readonly workspaceId?: WorkspaceId
  111. readonly cwd?: string
  112. readonly sessionId?: SessionId
  113. readonly agentPreset?: string
  114. }
  115. history: {
  116. readonly sessionId: SessionId
  117. readonly beforeSeq?: number
  118. readonly maxMessages?: number
  119. }
  120. selectModel: {
  121. readonly sessionId: SessionId
  122. readonly provider: string
  123. readonly model: string
  124. readonly reasoningEffort?: string
  125. }
  126. prompt: {
  127. readonly sessionId: SessionId
  128. readonly mode: 'queue' | 'steer'
  129. readonly content: readonly ({ readonly type: 'text'; readonly text: string } | {
  130. readonly type: 'image'
  131. readonly mediaType: 'image/png' | 'image/jpeg' | 'image/webp' | 'image/gif'
  132. readonly data: string
  133. readonly name?: string
  134. })[]
  135. }
  136. cancel: { readonly sessionId: SessionId }
  137. rename: { readonly sessionId: SessionId; readonly title: string }
  138. }
  139. interface FixtureSessionValues {
  140. list: { readonly items: FixtureSessionSummary[] }
  141. search: { readonly items: readonly { readonly sessionId: SessionId; readonly snippet: string }[]; readonly hasMore: boolean }
  142. create: { readonly sessionId: SessionId }
  143. history: FixturePage
  144. selectModel: { readonly selected: ModelSelection }
  145. prompt: { readonly accepted: true }
  146. cancel: Record<never, never>
  147. rename: { readonly title: string; readonly seq: number }
  148. }
  149. type FixtureSessionApi = {
  150. [K in keyof FixtureSessionRequests]: (
  151. request: RpcRequest<FixtureSessionRequests[K]>,
  152. signal?: AbortSignal,
  153. ) => Promise<RpcResponse<FixtureSessionValues[K]>>
  154. }
  155. type FixtureSessionClient = {
  156. [K in keyof FixtureSessionRequests]: (
  157. request: FixtureSessionRequests[K],
  158. signal?: AbortSignal,
  159. ) => Promise<RpcResponse<FixtureSessionValues[K]>>
  160. }
  161. interface FixtureSessionRemote {
  162. modelCatalog(): Promise<ConnectionRpcResult<ModelCatalog>>
  163. follow(sessionId: SessionId, signal: AbortSignal): AsyncIterable<FixtureFollowFrame>
  164. control(signal: AbortSignal): AsyncIterable<FixtureControlFrame>
  165. }
  166. interface FixtureWorkspaceView {
  167. readonly workspaceId: WorkspaceId
  168. readonly path: string
  169. readonly title: string
  170. readonly sessionIds: readonly SessionId[]
  171. readonly createdAt: string
  172. readonly updatedAt: string
  173. }
  174. interface FixtureWorkspaceRequests {
  175. create: { readonly path: string }
  176. rename: { readonly workspaceId: WorkspaceId; readonly title: string }
  177. delete: { readonly workspaceId: WorkspaceId }
  178. insertBefore: { readonly workspaceId: WorkspaceId; readonly beforeWorkspaceId?: WorkspaceId }
  179. insertSessionBefore: {
  180. readonly workspaceId: WorkspaceId
  181. readonly sessionId: SessionId
  182. readonly beforeSessionId?: SessionId
  183. }
  184. archiveSession: { readonly sessionId: SessionId }
  185. }
  186. interface FixtureWorkspaceValues {
  187. create: { readonly workspace: FixtureWorkspaceView; readonly created: boolean }
  188. rename: { readonly workspace: FixtureWorkspaceView }
  189. delete: { readonly deleted: true }
  190. insertBefore: { readonly workspaceIds: readonly WorkspaceId[] }
  191. insertSessionBefore: { readonly workspace: FixtureWorkspaceView }
  192. archiveSession: { readonly archivedSessionIds: readonly SessionId[] }
  193. }
  194. type FixtureWorkspaceApi = {
  195. [K in keyof FixtureWorkspaceRequests]: (
  196. request: RpcRequest<FixtureWorkspaceRequests[K]>,
  197. signal?: AbortSignal,
  198. ) => Promise<RpcResponse<FixtureWorkspaceValues[K]>>
  199. }
  200. type FixtureWorkspaceClient = {
  201. [K in keyof FixtureWorkspaceRequests]: (
  202. request: FixtureWorkspaceRequests[K],
  203. signal?: AbortSignal,
  204. ) => Promise<RpcResponse<FixtureWorkspaceValues[K]>>
  205. }
  206. type FixtureWorkspaceFrame =
  207. | {
  208. readonly type: 'baseline'
  209. readonly value: {
  210. readonly items: readonly FixtureWorkspaceView[]
  211. readonly archivedSessionIds: readonly SessionId[]
  212. }
  213. }
  214. | { readonly type: 'upsert'; readonly workspace: FixtureWorkspaceView }
  215. | { readonly type: 'remove'; readonly workspaceId: WorkspaceId }
  216. | { readonly type: 'order'; readonly workspaceIds: readonly WorkspaceId[] }
  217. | { readonly type: 'archived'; readonly archivedSessionIds: readonly SessionId[] }
  218. interface FixtureWorkspaceRemote {
  219. follow(signal: AbortSignal): AsyncIterable<FixtureWorkspaceFrame>
  220. }
  221. interface FixtureRemoteEventNotificationFrame {
  222. readonly type: 'emit'
  223. readonly event: string
  224. readonly args: readonly unknown[]
  225. }
  226. interface FixtureRemoteEventRequestFrame {
  227. readonly type: 'waterfall'
  228. readonly event: string
  229. readonly eventId: string
  230. readonly agentId: SessionId
  231. readonly request: Readonly<Record<string, unknown>>
  232. }
  233. interface FixtureRemoteEventCancellationFrame {
  234. readonly type: 'cancel'
  235. readonly eventId: string
  236. }
  237. type FixtureRemoteEventFrame =
  238. | FixtureRemoteEventNotificationFrame
  239. | FixtureRemoteEventRequestFrame
  240. | FixtureRemoteEventCancellationFrame
  241. interface FixtureRemoteEventResult {
  242. readonly clientId: string
  243. readonly eventId: string
  244. readonly outcome:
  245. | { readonly kind: 'next' }
  246. | { readonly kind: 'result'; readonly value?: unknown }
  247. | {
  248. readonly kind: 'rejected'
  249. readonly error: {
  250. readonly name: string
  251. readonly message: string
  252. readonly code?: string
  253. readonly details?: unknown
  254. }
  255. }
  256. }
  257. interface FixtureRemoteEventStream extends AsyncIterable<FixtureRemoteEventFrame> {
  258. readonly clientId: Promise<string>
  259. }
  260. type FixtureTestApi = {
  261. /** The directory-picking Remote namespace as the fixture serves it. */
  262. readonly directoryPickerRemote: {
  263. pick: () => Promise<ConnectionRpcResult<string | null>>
  264. list: (path?: string) => Promise<ConnectionRpcResult<DirectoryListing>>
  265. createDirectory: (path: string, name: string) => Promise<ConnectionRpcResult<string>>
  266. }
  267. readonly sessions: FixtureSessionApi
  268. readonly sessionRemote: FixtureSessionRemote
  269. readonly workspace: FixtureWorkspaceApi
  270. readonly workspaceRemote: FixtureWorkspaceRemote
  271. readonly credentialRemote: FixtureCredentialRemote
  272. readonly settingsRemote: FixtureSettingsRemote
  273. readonly remoteEvents: (signal: AbortSignal) => FixtureRemoteEventStream
  274. readonly answerRemoteEvent: (result: FixtureRemoteEventResult) => Promise<unknown>
  275. }
  276. /** Keep existing fixture assertions compact while driving only the new Session Remote endpoints. */
  277. function createFixtureApi(options: FixtureOptions = {}): FixtureTestApi {
  278. const { rpc } = createFixtureFaces(options)
  279. return {
  280. directoryPickerRemote: {
  281. pick: () => rpc.call('/api', 'directoryPicker/pick', { args: {} }) as
  282. Promise<ConnectionRpcResult<string | null>>,
  283. list: (path?: string) => rpc.call('/api', 'directoryPicker/list', { args: { path } }) as
  284. Promise<ConnectionRpcResult<DirectoryListing>>,
  285. createDirectory: (path: string, name: string) =>
  286. rpc.call('/api', 'directoryPicker/createDirectory', { args: { path, name } }) as
  287. Promise<ConnectionRpcResult<string>>,
  288. },
  289. sessions: createSessionApi(rpc),
  290. sessionRemote: createSessionRemote(rpc),
  291. workspace: createWorkspaceApi(rpc),
  292. workspaceRemote: createWorkspaceRemote(rpc),
  293. credentialRemote: createCredentialRemote(rpc),
  294. settingsRemote: createSettingsRemote(rpc),
  295. remoteEvents: (signal: AbortSignal) => openFixtureRemoteEvents(rpc, signal),
  296. answerRemoteEvent: (result: FixtureRemoteEventResult) =>
  297. rpc.call('/api', '$events/result', { args: result }),
  298. }
  299. }
  300. /** The fixture's Credentials Remote endpoints over the shared RPC carrier. */
  301. interface FixtureCredentialRemote {
  302. describe(refs: readonly string[]): Promise<ConnectionRpcResult<unknown>>
  303. set(ref: string, value: string): Promise<ConnectionRpcResult<unknown>>
  304. unset(ref: string): Promise<ConnectionRpcResult<unknown>>
  305. }
  306. /** The settings Remote reads the fixture serves, addressed like the credential half. */
  307. interface FixtureSettingsRemote {
  308. describe(): Promise<ConnectionRpcResult<unknown>>
  309. update(ns: string, patch: unknown, expectedRevision?: number): Promise<ConnectionRpcResult<unknown>>
  310. replace(ns: string, section: unknown, expectedRevision?: number): Promise<ConnectionRpcResult<unknown>>
  311. }
  312. function createSettingsRemote(rpc: ClientConnectionRpc): FixtureSettingsRemote {
  313. return {
  314. describe: () => rpc.call('/api', 'settings/describe', { args: {} }),
  315. update: (ns, patch, expectedRevision) => rpc.call('/api', 'settings/update', {
  316. args: { ns, patch, expectedRevision },
  317. }),
  318. replace: (ns, section, expectedRevision) => rpc.call('/api', 'settings/replace', {
  319. args: { ns, section, expectedRevision },
  320. }),
  321. }
  322. }
  323. function createCredentialRemote(rpc: ClientConnectionRpc): FixtureCredentialRemote {
  324. return {
  325. describe: refs => rpc.call('/api', 'credentials/describe', { args: { refs } }),
  326. set: (ref, value) => rpc.call('/api', 'credentials/set', { args: { ref, value } }),
  327. unset: ref => rpc.call('/api', 'credentials/unset', { args: { ref } }),
  328. }
  329. }
  330. function openFixtureRemoteEvents(
  331. rpc: ClientConnectionRpc,
  332. signal: AbortSignal,
  333. ): FixtureRemoteEventStream {
  334. const ready = Promise.withResolvers<string>()
  335. const source = (async function* (): AsyncGenerator<FixtureRemoteEventFrame> {
  336. const stream = rpc.open?.('/api', '$events', { args: {} }, signal)
  337. if (stream === undefined) throw new Error('fixture forwarded-event stream is unavailable')
  338. let opened = false
  339. for await (const value of stream) {
  340. if (!opened) {
  341. expect(value).toMatchObject({ type: 'ready' })
  342. const clientId: unknown = Reflect.get(value as object, 'clientId')
  343. if (typeof clientId !== 'string') throw new Error('fixture forwarded-event stream omitted its Client id')
  344. ready.resolve(clientId)
  345. opened = true
  346. continue
  347. }
  348. yield value as FixtureRemoteEventFrame
  349. }
  350. })()
  351. return Object.assign(source, { clientId: ready.promise })
  352. }
  353. function createSessionApi(rpc: ClientConnectionRpc): FixtureSessionApi {
  354. const call = async <K extends keyof FixtureSessionRequests>(
  355. endpoint: K,
  356. request: RpcRequest<FixtureSessionRequests[K]>,
  357. signal?: AbortSignal,
  358. ): Promise<RpcResponse<FixtureSessionValues[K]>> => {
  359. const page = endpoint === 'history'
  360. ? request.payload as FixtureSessionRequests['history']
  361. : undefined
  362. const args = endpoint === 'list'
  363. ? { _request: request.payload }
  364. : endpoint === 'history'
  365. ? {
  366. request: {
  367. address: { kind: 'session', sessionId: page?.sessionId },
  368. ...page?.beforeSeq === undefined ? {} : { beforeSeq: page.beforeSeq },
  369. ...page?.maxMessages === undefined ? {} : { maxMessages: page.maxMessages },
  370. },
  371. }
  372. : { request: request.payload }
  373. const remoteEndpoint = endpoint === 'history' ? 'page' : endpoint
  374. const result = await rpc.call('/api', `session/${remoteEndpoint}`, { args }, signal)
  375. return {
  376. rpcId: request.rpcId,
  377. result: result as unknown as RpcResult<FixtureSessionValues[K]>,
  378. }
  379. }
  380. return {
  381. list: (request, signal) => call('list', request, signal),
  382. search: (request, signal) => call('search', request, signal),
  383. create: (request, signal) => call('create', request, signal),
  384. history: (request, signal) => call('history', request, signal),
  385. selectModel: (request, signal) => call('selectModel', request, signal),
  386. prompt: (request, signal) => call('prompt', request, signal),
  387. cancel: (request, signal) => call('cancel', request, signal),
  388. rename: (request, signal) => call('rename', request, signal),
  389. }
  390. }
  391. function createSessionClient(rpc: ClientConnectionRpc): FixtureSessionClient {
  392. const api = createSessionApi(rpc)
  393. return {
  394. list: (request, signal) => api.list(req(request), signal),
  395. search: (request, signal) => api.search(req(request), signal),
  396. create: (request, signal) => api.create(req(request), signal),
  397. history: (request, signal) => api.history(req(request), signal),
  398. selectModel: (request, signal) => api.selectModel(req(request), signal),
  399. prompt: (request, signal) => api.prompt(req(request), signal),
  400. cancel: (request, signal) => api.cancel(req(request), signal),
  401. rename: (request, signal) => api.rename(req(request), signal),
  402. }
  403. }
  404. function createSessionRemote(rpc: ClientConnectionRpc): FixtureSessionRemote {
  405. const open = <F>(endpoint: string, args: object, signal: AbortSignal): AsyncIterable<F> => {
  406. const stream = rpc.open?.('/api', endpoint, { args }, signal)
  407. if (stream === undefined) throw new Error(`fixture ${endpoint} stream is unavailable`)
  408. return stream as AsyncIterable<F>
  409. }
  410. return {
  411. modelCatalog: () => rpc.call('/api', 'session/modelCatalog', { args: {} }) as
  412. Promise<ConnectionRpcResult<ModelCatalog>>,
  413. follow: (sessionId, signal) => open<FixtureFollowFrame>('session/follow', {
  414. request: { address: { kind: 'session', sessionId } },
  415. }, signal),
  416. control: signal => open<FixtureControlFrame>('session/control', {}, signal),
  417. }
  418. }
  419. function createWorkspaceApi(rpc: ClientConnectionRpc): FixtureWorkspaceApi {
  420. const call = async <K extends keyof FixtureWorkspaceRequests>(
  421. endpoint: K,
  422. request: RpcRequest<FixtureWorkspaceRequests[K]>,
  423. signal?: AbortSignal,
  424. ): Promise<RpcResponse<FixtureWorkspaceValues[K]>> => {
  425. const result = await rpc.call('/api', `workspace/${endpoint}`, {
  426. args: { request: request.payload },
  427. }, signal)
  428. return {
  429. rpcId: request.rpcId,
  430. result: result as unknown as RpcResult<FixtureWorkspaceValues[K]>,
  431. }
  432. }
  433. return {
  434. create: (request, signal) => call('create', request, signal),
  435. rename: (request, signal) => call('rename', request, signal),
  436. delete: (request, signal) => call('delete', request, signal),
  437. insertBefore: (request, signal) => call('insertBefore', request, signal),
  438. insertSessionBefore: (request, signal) => call('insertSessionBefore', request, signal),
  439. archiveSession: (request, signal) => call('archiveSession', request, signal),
  440. }
  441. }
  442. function createWorkspaceClient(rpc: ClientConnectionRpc): FixtureWorkspaceClient {
  443. const api = createWorkspaceApi(rpc)
  444. return {
  445. create: (request, signal) => api.create(req(request), signal),
  446. rename: (request, signal) => api.rename(req(request), signal),
  447. delete: (request, signal) => api.delete(req(request), signal),
  448. insertBefore: (request, signal) => api.insertBefore(req(request), signal),
  449. insertSessionBefore: (request, signal) => api.insertSessionBefore(req(request), signal),
  450. archiveSession: (request, signal) => api.archiveSession(req(request), signal),
  451. }
  452. }
  453. function createWorkspaceRemote(rpc: ClientConnectionRpc): FixtureWorkspaceRemote {
  454. return {
  455. follow(signal) {
  456. const stream = rpc.open?.('/api', 'workspace/follow', { args: {} }, signal)
  457. if (stream === undefined) throw new Error('fixture workspace/follow stream is unavailable')
  458. return stream as AsyncIterable<FixtureWorkspaceFrame>
  459. },
  460. }
  461. }
  462. interface TimingHooks {
  463. setHistoryDelay(ms: number): void
  464. failNextHistory(): void
  465. appendUser(id: string, msg: string): void
  466. appendTitle(id: string, title: string): void
  467. startReasoningChunkStorm(id: string, chunkCount: number, chunksPerInterval: number, intervalMs: number): string
  468. reasoningChunkStormState(): {
  469. sessionId: string
  470. chunkCount: number
  471. chunksPerInterval: number
  472. intervalMs: number
  473. emitted: number
  474. marker: string
  475. emitting: boolean
  476. } | null
  477. beginModelRetry(id: string): void
  478. scheduleModelRetry(id: string, retry?: number, delayMs?: number): void
  479. cancelModelRetryDuringBackoff(id: string, delayMs?: number): void
  480. completeModelRetry(id: string): void
  481. appendSilent(id: string, msg: string): void
  482. breakStreams(): void
  483. }
  484. const timing = (): TimingHooks => (globalThis as Record<string, unknown>).__fxTiming as TimingHooks
  485. /** Collect value-stream frames until the predicate or a soft cap; abort ends the stream. */
  486. async function collectValues<F>(stream: AsyncIterable<F>, abort: AbortController, done: (frames: F[]) => boolean): Promise<F[]> {
  487. const frames: F[] = []
  488. for await (const frame of stream) {
  489. frames.push(frame)
  490. if (done(frames) || frames.length > 500) {
  491. abort.abort()
  492. break
  493. }
  494. }
  495. return frames
  496. }
  497. async function readControlBaseline(remote: FixtureSessionRemote): Promise<Extract<FixtureControlFrame, { type: 'baseline' }>> {
  498. const abort = new AbortController()
  499. for await (const frame of remote.control(abort.signal)) {
  500. if (frame.type !== 'baseline') continue
  501. abort.abort()
  502. return frame
  503. }
  504. throw new Error('fixture control baseline missing')
  505. }
  506. function isRemoteEventRequest(frame: FixtureRemoteEventFrame): frame is FixtureRemoteEventRequestFrame {
  507. return frame.type === 'waterfall'
  508. }
  509. function isRemoteEventCancellation(frame: FixtureRemoteEventFrame): frame is FixtureRemoteEventCancellationFrame {
  510. return frame.type === 'cancel'
  511. }
  512. async function readResidentRemoteEvents(
  513. api: FixtureTestApi,
  514. count: number,
  515. ): Promise<FixtureRemoteEventRequestFrame[]> {
  516. const abort = new AbortController()
  517. const frames = await collectValues(
  518. api.remoteEvents(abort.signal),
  519. abort,
  520. seen => seen.filter(isRemoteEventRequest).length >= count,
  521. )
  522. return frames.filter(isRemoteEventRequest)
  523. }
  524. async function nextRemoteEvent(
  525. iterator: AsyncIterator<FixtureRemoteEventFrame>,
  526. predicate: (frame: FixtureRemoteEventFrame) => boolean,
  527. ): Promise<FixtureRemoteEventFrame> {
  528. for (;;) {
  529. const item = await iterator.next()
  530. if (item.done) throw new Error('fixture Remote Event stream ended before the expected frame')
  531. if (predicate(item.value)) return item.value
  532. }
  533. }
  534. async function readOpeningCursor(remote: FixtureSessionRemote, sessionId: SessionId): Promise<number> {
  535. const abort = new AbortController()
  536. for await (const frame of remote.follow(sessionId, abort.signal)) {
  537. if (frame.type !== 'snapshot') continue
  538. abort.abort()
  539. return frame.cursor
  540. }
  541. throw new Error('fixture follow opening cursor missing')
  542. }
  543. async function readWorkspaceBaseline(
  544. remote: FixtureWorkspaceRemote,
  545. ): Promise<Extract<FixtureWorkspaceFrame, { type: 'baseline' }>['value']> {
  546. const abort = new AbortController()
  547. for await (const frame of remote.follow(abort.signal)) {
  548. if (frame.type !== 'baseline') continue
  549. abort.abort()
  550. return frame.value
  551. }
  552. throw new Error('fixture Workspace baseline missing')
  553. }
  554. describe('createFixtureApi', () => {
  555. it('serves the session list sorted by updatedAt desc and echoes rpcIds on every unary', async () => {
  556. const api = createFixtureApi()
  557. const request = req({})
  558. const response = await api.sessions.list(request)
  559. expect(response.rpcId).toBe(request.rpcId)
  560. if (!response.result.ok) throw new Error('list failed')
  561. expect(response.result.value.items.map(s => s.sessionId)).toEqual(['fx-alpha', 'fx-beta', 'fx-gamma'])
  562. expect(response.result.value.items[1]?.parentSessionId).toBe('fx-alpha') // lineage material
  563. })
  564. it('searches current message text with literal unicode61-style token phrases', async () => {
  565. const api = createFixtureApi()
  566. const signal = new AbortController().signal
  567. const phrase = await api.sessions.search(req({ query: 'FIXTURE 历史消息' }), signal)
  568. expect(phrase.result).toMatchObject({
  569. ok: true,
  570. value: {
  571. items: [{ sessionId: 'fx-alpha' }],
  572. hasMore: false,
  573. },
  574. })
  575. if (!phrase.result.ok) throw new Error('search failed')
  576. expect(phrase.result.value.items[0]?.snippet).toContain('fixture 历史消息')
  577. timing().appendUser(
  578. 'fx-alpha',
  579. `${'leading context '.repeat(20)}late café token${' trailing context'.repeat(20)}`,
  580. )
  581. const late = await api.sessions.search(req({ query: 'LATE CAFE TOKEN' }), signal)
  582. if (!late.result.ok) throw new Error('late search failed')
  583. const lateSnippet = late.result.value.items[0]?.snippet ?? ''
  584. expect(lateSnippet).toContain('late café token')
  585. expect(lateSnippet.startsWith('…')).toBe(true)
  586. expect(lateSnippet.endsWith('…')).toBe(true)
  587. expect(Array.from(lateSnippet).length).toBeLessThanOrEqual(120)
  588. timing().appendUser('fx-alpha', 'Greek final sigma: ος')
  589. const finalSigma = await api.sessions.search(req({ query: 'ΟΣ' }), signal)
  590. if (!finalSigma.result.ok) throw new Error('final sigma search failed')
  591. expect(finalSigma.result.value.items[0]?.snippet).toContain('ος')
  592. const substring = await api.sessions.search(req({ query: 'ixtur' }), signal)
  593. expect(substring.result).toEqual({
  594. ok: true,
  595. value: { items: [], hasMore: false },
  596. })
  597. const punctuationOnly = await api.sessions.search(req({ query: '*' }), signal)
  598. expect(punctuationOnly.result).toEqual({
  599. ok: true,
  600. value: { items: [], hasMore: false },
  601. })
  602. const reasoningOnly = await api.sessions.search(req({ query: '思考过程' }), signal)
  603. expect(reasoningOnly.result).toEqual({
  604. ok: true,
  605. value: { items: [], hasMore: false },
  606. })
  607. const aborted = new AbortController()
  608. aborted.abort()
  609. await expect(api.sessions.search(req({ query: 'fixture' }), aborted.signal))
  610. .resolves.toMatchObject({ result: { ok: false, error: { code: 'gateway/cancelled' } } })
  611. })
  612. it('pages history backwards on message-boundary cuts with seq-contiguous stitching', async () => {
  613. const api = createFixtureApi()
  614. const tail = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 10 }))
  615. if (!tail.result.ok) throw new Error('history failed')
  616. const tailPage = tail.result.value
  617. expect(tailPage.hasMore).toBe(true)
  618. const tailEvents = historyEvents(tailPage.records)
  619. expect(tailEvents[0]?.type).toBe('turn/start') // cut lands on a turn boundary
  620. const boundary = tailEvents[0]?.seq ?? 0
  621. expect(boundary).toBeGreaterThan(0)
  622. const older = await api.sessions.history(req({ sessionId: sid('fx-alpha'), beforeSeq: boundary, maxMessages: 10 }))
  623. if (!older.result.ok) throw new Error('older failed')
  624. const olderTail = historyEvents(older.result.value.records).at(-1)
  625. expect((olderTail?.seq ?? -1) + 1).toBe(boundary) // pages stitch with no hole/overlap
  626. // Out-of-range beforeSeq clamps instead of exploding.
  627. const clamped = await api.sessions.history(req({ sessionId: sid('fx-alpha'), beforeSeq: -5, maxMessages: 10 }))
  628. if (!clamped.result.ok) throw new Error('clamped failed')
  629. expect(clamped.result.value.records).toEqual([])
  630. // Unknown session: empty page, not an error (history of a bare id).
  631. const empty = await api.sessions.history(req({ sessionId: sid('no-such'), maxMessages: 10 }))
  632. if (!empty.result.ok) throw new Error('empty failed')
  633. expect(empty.result.value).toEqual({ records: [], hasMore: false })
  634. })
  635. it('serves raw history entries with replayable tool-result metadata', async () => {
  636. const api = createFixtureApi()
  637. const response = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 200 }))
  638. if (!response.result.ok) throw new Error('history failed')
  639. const records = response.result.value.records
  640. const results = historyEvents(records)
  641. .filter(event => event.type === 'tool/result')
  642. expect(results.find(event => event.data.turn === 64)).toMatchObject({
  643. data: {
  644. meta: {
  645. diffs: [
  646. { path: 'src/config.ts', oldText: 'const timeout = 30', newText: 'const timeout = 60' },
  647. { path: 'src/config.ts', oldText: 'retries: 1', newText: 'retries: 3' },
  648. ],
  649. },
  650. },
  651. })
  652. expect(results.find(event => event.data.turn === 67)).toMatchObject({
  653. data: { meta: { shape: 'matches', truncated: true, total: 42 } },
  654. })
  655. expect(results.find(event => event.data.turn === 69)).toMatchObject({
  656. data: { meta: { path: 'packages/client/ui-primitives/src/ReadBlock.tsx', offset: 41, totalLines: 180 } },
  657. })
  658. const webSearch = results.find(event => event.data.turn === 70)
  659. expect(webSearch).toHaveProperty('data.meta.truncated', true)
  660. expect(webSearch).toHaveProperty('data.meta.sources', expect.arrayContaining([
  661. expect.objectContaining({ url: 'https://github.com/deepseek-ai/deepseek-harness' }),
  662. ]))
  663. expect(results.find(event => event.data.turn === 71)).toMatchObject({
  664. data: { meta: { url: 'https://www.deepseek.com/blog/harness-architecture', statusCode: 200 } },
  665. })
  666. const terminal = results.find(event => event.data.turn === 66)
  667. expect(terminal).toHaveProperty('data.message.content.0.content.0.type', 'text')
  668. expect(terminal).toHaveProperty(
  669. 'data.message.content.0.content.0.text',
  670. expect.stringContaining('\n[exit code: 1]'),
  671. )
  672. })
  673. it('serves grouped models and keeps a selection for later history and fixture requests', async () => {
  674. const api = createFixtureApi()
  675. const sessionId = sid('fx-alpha')
  676. const catalog = await api.sessionRemote.modelCatalog()
  677. if (!catalog.ok) throw new Error('models failed')
  678. expect(catalog.value.groups.map(group => group.name)).toEqual(['DeepSeek', 'OpenAI'])
  679. expect(catalog.value.groups[0]?.models.map(model => model.id))
  680. .toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
  681. const selected = await api.sessions.selectModel(req({
  682. sessionId,
  683. provider: 'openai',
  684. model: 'gpt-5',
  685. }))
  686. if (!selected.result.ok) throw new Error('selection failed')
  687. expect(selected.result.value.selected).toEqual({ provider: 'openai', model: 'gpt-5' })
  688. const history = await api.sessions.history(req({ sessionId }))
  689. if (!history.result.ok) throw new Error('history failed')
  690. const prompt = await api.sessions.prompt(req({
  691. sessionId,
  692. mode: 'queue',
  693. content: [{ type: 'text', text: 'report model' }],
  694. }))
  695. expect(prompt.result.ok).toBe(true)
  696. await new Promise(resolve => setTimeout(resolve, 600))
  697. const after = await api.sessions.history(req({ sessionId }))
  698. if (!after.result.ok) throw new Error('history failed')
  699. expect(JSON.stringify(after.result.value.records)).toContain('openai/gpt-5')
  700. })
  701. it('serves configured DeepSeek readiness and keeps credential values write-only', async () => {
  702. const api = createFixtureApi()
  703. const settings = await api.settingsRemote.describe()
  704. if (!settings.ok) throw new Error('settings describe failed')
  705. expect((settings.value as { namespaces: unknown[] }).namespaces).toMatchObject([{
  706. ns: 'llm-deepseek',
  707. value: { apiKeyEnv: 'DEEPSEEK_API_KEY' },
  708. secrets: [{ path: ['apiKey'], set: false }],
  709. }])
  710. for (const result of [
  711. await api.settingsRemote.update('llm-deepseek', {}, undefined),
  712. await api.settingsRemote.replace('llm-deepseek', {}, undefined),
  713. ]) {
  714. expect(result).toMatchObject({
  715. ok: false,
  716. error: { code: 'settings/rejected', message: 'fixture: the minimal readiness settings descriptor is read-only' },
  717. })
  718. }
  719. const describe = async (refs: readonly string[]): Promise<Record<string, unknown>> => {
  720. const result = await api.credentialRemote.describe(refs)
  721. if (!result.ok) throw new Error('credential describe failed')
  722. return result.value as Record<string, unknown>
  723. }
  724. expect(await describe(['DEEPSEEK_API_KEY', 'TEST_API_KEY'])).toEqual({
  725. DEEPSEEK_API_KEY: { configured: true, source: 'file', writable: true },
  726. TEST_API_KEY: { configured: false, writable: true },
  727. })
  728. await api.credentialRemote.set('TEST_API_KEY', 'write-only-fixture-secret')
  729. expect((await describe(['TEST_API_KEY'])).TEST_API_KEY).toEqual({
  730. configured: true,
  731. source: 'file',
  732. writable: true,
  733. })
  734. await api.credentialRemote.unset('TEST_API_KEY')
  735. expect((await describe(['TEST_API_KEY'])).TEST_API_KEY).toEqual({ configured: false, writable: true })
  736. })
  737. it('emits the todo/write snapshot at the real tool boundary: between tool/call and tool/result, timestamps monotonic', async () => {
  738. const api = createFixtureApi()
  739. const tail = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 10 }))
  740. if (!tail.result.ok) throw new Error('history failed')
  741. const events = historyEvents(tail.result.value.records)
  742. const todoAt = events.findIndex(e => e.type === 'todo/write')
  743. expect(todoAt).toBeGreaterThan(0)
  744. // Production ordering (the tool appends mid-execution): call → snapshot → result.
  745. expect(events[todoAt - 1]?.type).toBe('tool/call')
  746. expect(events[todoAt + 1]?.type).toBe('tool/result')
  747. const times = events.slice(todoAt - 1, todoAt + 2).map(e => e.time)
  748. expect(times[0]).toBeLessThanOrEqual(times[1] ?? 0)
  749. expect(times[1]).toBeLessThanOrEqual(times[2] ?? 0)
  750. // The sample is a parallel plan: this fixture chooses the parallel policy,
  751. // so the surfaces fed from here face more than one active item.
  752. const snapshot = events[todoAt] as { data: { todos: { status: string }[] } }
  753. expect(snapshot.data.todos.filter(t => t.status === 'in_progress')).toHaveLength(2)
  754. })
  755. it('create adds a session and announces it through the Host Remote event stream', async () => {
  756. const api = createFixtureApi()
  757. const abort = new AbortController()
  758. const seen: FixtureRemoteEventNotificationFrame[] = []
  759. const consuming = (async () => {
  760. for await (const frame of api.remoteEvents(abort.signal)) {
  761. if (frame.type !== 'emit' || frame.event !== 'api-session/added') continue
  762. seen.push(frame)
  763. abort.abort()
  764. break
  765. }
  766. })()
  767. await new Promise(resolve => setTimeout(resolve, 10)) // let the stream register
  768. const created = await api.sessions.create(req({}))
  769. if (!created.result.ok) throw new Error('create failed')
  770. await consuming
  771. if (!created.result.ok) throw new Error('create failed')
  772. const createdId = created.result.value.sessionId
  773. expect(seen).toHaveLength(1)
  774. const added = seen[0]
  775. expect(added).toMatchObject({
  776. event: 'api-session/added',
  777. args: [{ sessionId: createdId, blank: true, cwd: '/tmp/fixture' }],
  778. })
  779. const list = await api.sessions.list(req({}))
  780. if (!list.result.ok) throw new Error('list failed')
  781. expect(list.result.value.items.some(s => s.sessionId === createdId)).toBe(true)
  782. })
  783. it('prompt replays a full streamed turn and cancel mid-replay freezes with (已中断)', async () => {
  784. const api = createFixtureApi()
  785. const created = await api.sessions.create(req({}))
  786. if (!created.result.ok) throw new Error('create failed')
  787. const id = created.result.value.sessionId
  788. const followAbort = new AbortController()
  789. const controlAbort = new AbortController()
  790. const controlFrames: FixtureControlFrame[] = []
  791. const followPromise = collectValues(
  792. api.sessionRemote.follow(id, followAbort.signal),
  793. followAbort,
  794. frames => frames.some(frame => frame.type === 'event' && frame.event.type === 'turn/end'),
  795. )
  796. const controlPromise = (async () => {
  797. for await (const frame of api.sessionRemote.control(controlAbort.signal)) controlFrames.push(frame)
  798. })()
  799. await new Promise(resolve => setTimeout(resolve, 10))
  800. // Unknown session → session/not-found with the id echoed in details.
  801. const missing = await api.sessions.prompt(req({ sessionId: sid('ghost'), mode: 'queue' as const, content: [{ type: 'text' as const, text: 'x' }] }))
  802. expect(missing.result).toMatchObject({ ok: false, error: { code: 'session/not-found', details: { sessionId: 'ghost' } } })
  803. // Real prompt: replay starts (running flips true), cancel freezes it.
  804. const accepted = await api.sessions.prompt(req({ sessionId: id, mode: 'queue' as const, content: [{ type: 'text' as const, text: 'render markdown' }] }))
  805. expect(accepted.result).toMatchObject({ ok: true, value: { accepted: true } })
  806. await new Promise(resolve => setTimeout(resolve, 120)) // a couple of typewriter ticks
  807. await api.sessions.cancel(req({ sessionId: id }))
  808. const frames = await followPromise
  809. const types = frames.flatMap(frame => frame.type === 'event' ? [frame.event.type] : [])
  810. expect(types).toContain('turn/start')
  811. expect(types).toContain('user/message')
  812. expect(types).toContain('assistant/chunk')
  813. expect(types).toContain('assistant/message')
  814. expect(types.at(-1)).toBe('turn/end')
  815. // Capacity is durable log state, not a transient frame: the prompt path
  816. // records request/context and the projection carries it to the client.
  817. expect(types).toContain('request/context')
  818. await vi.waitFor(() => {
  819. expect(controlFrames.some(frame =>
  820. frame.type === 'projection'
  821. && frame.key === 'contextBreakdown'
  822. && (frame.value as { messageTokens?: number }).messageTokens! > 0)).toBe(true)
  823. })
  824. expect(controlFrames.some(frame =>
  825. frame.type === 'projection'
  826. && frame.key === 'tokenUsage'
  827. && (frame.value as { outputTokens?: number }).outputTokens === 8)).toBe(true)
  828. expect(controlFrames.some(frame =>
  829. frame.type === 'projection'
  830. && frame.key === 'contextPressure'
  831. && (frame.value as { contextWindow?: number }).contextWindow === 128_000)).toBe(true)
  832. const finalize = frames.find(frame => frame.type === 'event' && frame.event.type === 'assistant/message')
  833. if (finalize?.type !== 'event') throw new Error('assistant final event missing')
  834. expect(JSON.stringify(finalize?.event.data)).toContain('(已中断)')
  835. controlAbort.abort()
  836. await controlPromise
  837. // Idle cancel: no replay in flight, must not explode; running flips false.
  838. const idleCancel = await api.sessions.cancel(req({ sessionId: id }))
  839. expect(idleCancel.result).toMatchObject({ ok: true })
  840. })
  841. it('steer during a replay lands a user/message inside the current turn and the replay continues', async () => {
  842. const api = createFixtureApi()
  843. const created = await api.sessions.create(req({}))
  844. if (!created.result.ok) throw new Error('create failed')
  845. const id = created.result.value.sessionId
  846. const abort = new AbortController()
  847. const framesPromise = collectValues(api.sessionRemote.follow(id, abort.signal), abort,
  848. frames => frames.some(frame => frame.type === 'event' && frame.event.type === 'turn/end'))
  849. await new Promise(resolve => setTimeout(resolve, 10))
  850. await api.sessions.prompt(req({ sessionId: id, mode: 'queue' as const, content: [{ type: 'text' as const, text: '短' }] }))
  851. await api.sessions.prompt(req({ sessionId: id, mode: 'steer' as const, content: [{ type: 'text' as const, text: '插话' }] }))
  852. const frames = await framesPromise
  853. const types = frames.flatMap(frame => frame.type === 'event' ? [frame.event.type] : [])
  854. expect(JSON.stringify(frames)).toContain('插话')
  855. expect(types.at(-1)).toBe('turn/end') // steer did not restart the turn
  856. })
  857. it('control replays projections while resident Remote Events retain ids across reconnects', async () => {
  858. const api = createFixtureApi()
  859. const first = await readControlBaseline(api.sessionRemote)
  860. const second = await readControlBaseline(api.sessionRemote)
  861. expect(first.value.approvals).toEqual([])
  862. expect(first.value.questions).toEqual([])
  863. const alpha = first.value.projections['fx-alpha']
  864. expect(alpha?.asOfSeq).toBeGreaterThan(0)
  865. expect(alpha?.values).toMatchObject({
  866. title: 'Fixture 历史会话',
  867. plan: { active: false, pending: false },
  868. goal: null,
  869. imageLimits: { maxImagesPerMessage: 20, maxImageBytes: 5 * 1024 * 1024 },
  870. })
  871. expect((alpha?.values['contextBreakdown'] as { messageTokens: number }).messageTokens).toBeGreaterThan(0)
  872. expect((alpha?.values['sessionStats'] as { steps: number }).steps).toBeGreaterThan(0)
  873. expect(second.value.projections['fx-alpha']).toEqual(alpha)
  874. const firstEvents = await readResidentRemoteEvents(api, 2)
  875. const secondEvents = await readResidentRemoteEvents(api, 2)
  876. const firstApproval = firstEvents.find(frame => frame.event === 'approval/request')
  877. const firstQuestion = firstEvents.find(frame => frame.event === 'user-questions/request')
  878. const secondApproval = secondEvents.find(frame => frame.event === 'approval/request')
  879. const secondQuestion = secondEvents.find(frame => frame.event === 'user-questions/request')
  880. expect(firstApproval).toMatchObject({
  881. type: 'waterfall',
  882. request: { toolName: 'dangerous_tool' },
  883. agentId: 'fx-alpha',
  884. })
  885. expect(firstQuestion).toMatchObject({
  886. type: 'waterfall',
  887. agentId: 'fx-alpha',
  888. })
  889. expect(Array.isArray(firstQuestion?.request.questions)).toBe(true)
  890. expect(secondApproval?.eventId).toBe(firstApproval?.eventId)
  891. expect(secondQuestion?.eventId).toBe(firstQuestion?.eventId)
  892. expect(await readOpeningCursor(api.sessionRemote, sid('fx-alpha'))).toBeGreaterThan(0)
  893. })
  894. it('steer with no replay in flight falls through to a fresh queued turn; non-text blocks stringify empty', async () => {
  895. const api = createFixtureApi()
  896. const created = await api.sessions.create(req({}))
  897. if (!created.result.ok) throw new Error('create failed')
  898. const abort = new AbortController()
  899. const framesPromise = collectValues(
  900. api.sessionRemote.follow(created.result.value.sessionId, abort.signal),
  901. abort,
  902. frames => frames.some(frame => frame.type === 'event' && frame.event.type === 'turn/end'),
  903. )
  904. await new Promise(resolve => setTimeout(resolve, 10))
  905. // steer while idle + a non-text content block (covers the '' arm of the text join).
  906. await api.sessions.prompt(req({
  907. sessionId: created.result.value.sessionId, mode: 'steer' as const,
  908. content: [{ type: 'text' as const, text: '短' }, { type: 'image', data: 'x' } as never],
  909. }))
  910. const frames = await framesPromise
  911. const types = frames.flatMap(frame => frame.type === 'event' ? [frame.event.type] : [])
  912. expect(types[0]).toBe('turn/start') // idle steer degraded to a queued turn, not an in-turn insert
  913. })
  914. it('gamma interval flip emits a Remote status event and its empty follow source opens at -1', async () => {
  915. vi.useFakeTimers()
  916. try {
  917. const api = createFixtureApi()
  918. const abort = new AbortController()
  919. const hostSeen: FixtureRemoteEventFrame[] = []
  920. const consuming = (async () => {
  921. for await (const frame of api.remoteEvents(abort.signal)) hostSeen.push(frame)
  922. })()
  923. await vi.advanceTimersByTimeAsync(5001) // interval fires: fx-gamma flips running=true (no log exists)
  924. expect(hostSeen).toContainEqual({
  925. type: 'emit',
  926. event: 'api-session/status',
  927. args: [sid('fx-gamma'), true],
  928. })
  929. expect(await readOpeningCursor(api.sessionRemote, sid('fx-gamma'))).toBe(-1)
  930. abort.abort()
  931. await vi.advanceTimersByTimeAsync(10)
  932. await consuming
  933. } finally {
  934. vi.useRealTimers()
  935. }
  936. })
  937. it('answers a resident question through its Remote Event id and stops replaying it', async () => {
  938. const api = createFixtureApi()
  939. const abort = new AbortController()
  940. const stream = api.remoteEvents(abort.signal)
  941. const iterator = stream[Symbol.asyncIterator]()
  942. const question = await nextRemoteEvent(
  943. iterator,
  944. frame => isRemoteEventRequest(frame) && frame.event === 'user-questions/request',
  945. )
  946. if (!isRemoteEventRequest(question)) throw new Error('fixture question Remote Event missing')
  947. const clientId = await stream.clientId
  948. await expect(api.answerRemoteEvent({
  949. clientId,
  950. eventId: 'unrelated',
  951. outcome: { kind: 'result', value: {} },
  952. })).resolves.toEqual({ ok: true, value: undefined })
  953. await expect(api.answerRemoteEvent({
  954. clientId,
  955. eventId: question.eventId,
  956. outcome: { kind: 'result', value: { answers: {} } },
  957. })).resolves.toEqual({ ok: true, value: undefined })
  958. const cancelled = await nextRemoteEvent(
  959. iterator,
  960. frame => isRemoteEventCancellation(frame) && frame.eventId === question.eventId,
  961. )
  962. expect(cancelled).toEqual({ type: 'cancel', eventId: question.eventId })
  963. abort.abort()
  964. await iterator.return?.()
  965. await expect(api.answerRemoteEvent({
  966. clientId,
  967. eventId: question.eventId,
  968. outcome: { kind: 'result', value: { answers: {} } },
  969. })).resolves.toMatchObject({ ok: false, error: { code: 'gateway/invocation-unavailable' } })
  970. const remaining = await readResidentRemoteEvents(api, 1)
  971. expect(remaining.map(frame => frame.event)).toEqual(['approval/request'])
  972. const cancelledApi = createFixtureApi()
  973. const cancelAbort = new AbortController()
  974. const cancelStream = cancelledApi.remoteEvents(cancelAbort.signal)
  975. const cancelIterator = cancelStream[Symbol.asyncIterator]()
  976. const cancelQuestion = await nextRemoteEvent(
  977. cancelIterator,
  978. frame => isRemoteEventRequest(frame) && frame.event === 'user-questions/request',
  979. )
  980. if (!isRemoteEventRequest(cancelQuestion)) throw new Error('fixture cancellation question missing')
  981. await expect(cancelledApi.answerRemoteEvent({
  982. clientId: await cancelStream.clientId,
  983. eventId: cancelQuestion.eventId,
  984. outcome: {
  985. kind: 'rejected',
  986. error: { name: 'UserQuestionError', message: 'skip', code: 'ASK_CANCELLED' },
  987. },
  988. })).resolves.toEqual({ ok: true, value: undefined })
  989. cancelAbort.abort()
  990. await cancelIterator.return?.()
  991. const afterCancellation = await readResidentRemoteEvents(cancelledApi, 1)
  992. expect(afterCancellation.map(frame => frame.event)).toEqual(['approval/request'])
  993. })
  994. it('answers a resident approval and broadcasts cancellation to its active delivery', async () => {
  995. const api = createFixtureApi()
  996. const abort = new AbortController()
  997. const stream = api.remoteEvents(abort.signal)
  998. const iterator = stream[Symbol.asyncIterator]()
  999. const approval = await nextRemoteEvent(
  1000. iterator,
  1001. frame => isRemoteEventRequest(frame) && frame.event === 'approval/request',
  1002. )
  1003. if (!isRemoteEventRequest(approval)) throw new Error('fixture approval Remote Event missing')
  1004. await expect(api.answerRemoteEvent({
  1005. clientId: await stream.clientId,
  1006. eventId: approval.eventId,
  1007. outcome: { kind: 'result', value: 'allowed-once' },
  1008. })).resolves.toEqual({ ok: true, value: undefined })
  1009. const cancelled = await nextRemoteEvent(
  1010. iterator,
  1011. frame => isRemoteEventCancellation(frame) && frame.eventId === approval.eventId,
  1012. )
  1013. expect(cancelled).toEqual({ type: 'cancel', eventId: approval.eventId })
  1014. abort.abort()
  1015. await iterator.return?.()
  1016. await expect(api.answerRemoteEvent({
  1017. clientId: await stream.clientId,
  1018. eventId: approval.eventId,
  1019. outcome: { kind: 'next' },
  1020. })).resolves.toMatchObject({ ok: false, error: { code: 'gateway/invocation-unavailable' } })
  1021. const remaining = await readResidentRemoteEvents(api, 1)
  1022. expect(remaining.map(frame => frame.event)).toEqual(['user-questions/request'])
  1023. })
  1024. it('createDirectory under the root mints /name whose listing and crumbs share the identity', async () => {
  1025. const api = createFixtureApi()
  1026. const created = await api.directoryPickerRemote.createDirectory('/', 'srv')
  1027. if (!created.ok) throw new Error('create failed')
  1028. expect(created.value).toBe('/srv')
  1029. const listed = await api.directoryPickerRemote.list('/srv')
  1030. if (!listed.ok) throw new Error('list failed')
  1031. expect(listed.value.crumbs).toEqual([
  1032. { name: '/', path: '/', hidden: false },
  1033. { name: 'srv', path: '/srv', hidden: false },
  1034. ])
  1035. const root = await api.directoryPickerRemote.list('/')
  1036. if (!root.ok) throw new Error('root list failed')
  1037. expect(root.value.entries).toContainEqual({ name: 'srv', path: '/srv', hidden: false })
  1038. })
  1039. it('workspace/follow serves the resident baseline and create reuses on path collision', async () => {
  1040. const api = createFixtureApi()
  1041. const baseline = await readWorkspaceBaseline(api.workspaceRemote)
  1042. expect(baseline.items).toEqual([
  1043. expect.objectContaining({
  1044. workspaceId: 'fx-ws-fixture', path: '/tmp/fixture', title: 'fixture',
  1045. sessionIds: ['fx-alpha', 'fx-beta', 'fx-gamma'],
  1046. }),
  1047. expect.objectContaining({
  1048. workspaceId: 'fx-ws-home', path: '/home/fixture/Documents/project', title: 'project',
  1049. sessionIds: [],
  1050. }),
  1051. ])
  1052. // path collision → the existing entity comes back, created:false, no frame.
  1053. const reused = await api.workspace.create(req({ path: '/tmp/fixture' }))
  1054. if (!reused.result.ok) throw new Error('reuse failed')
  1055. expect(reused.result.value).toMatchObject({ created: false, workspace: { workspaceId: 'fx-ws-fixture' } })
  1056. })
  1057. it('workspace.create on a fresh path mints a new entity and pushes an upsert', async () => {
  1058. const api = createFixtureApi()
  1059. const abort = new AbortController()
  1060. const consuming = collectValues(
  1061. api.workspaceRemote.follow(abort.signal),
  1062. abort,
  1063. frames => frames.some(frame => frame.type === 'upsert'
  1064. && frame.workspace.path === '/tmp/fixture-workspaces/nova'),
  1065. )
  1066. await new Promise(resolve => setTimeout(resolve, 10))
  1067. const created = await api.workspace.create(req({ path: '/tmp/fixture-workspaces/nova' }))
  1068. if (!created.result.ok) throw new Error('create failed')
  1069. expect(created.result.value.created).toBe(true)
  1070. expect(created.result.value.workspace).toMatchObject({
  1071. path: '/tmp/fixture-workspaces/nova', title: 'nova', sessionIds: [],
  1072. })
  1073. const frames = await consuming
  1074. expect(frames.at(-1)).toEqual({ type: 'upsert', workspace: created.result.value.workspace })
  1075. // A basename-less path serves as its own title.
  1076. const rootPath = await api.workspace.create(req({ path: '/' }))
  1077. if (!rootPath.result.ok) throw new Error('rootPath failed')
  1078. expect(rootPath.result.value.workspace.title).toBe('/')
  1079. })
  1080. it('workspace.rename covers not-found, conflict, no-op, and the changed frame', async () => {
  1081. const api = createFixtureApi()
  1082. const abort = new AbortController()
  1083. const consuming = collectValues(
  1084. api.workspaceRemote.follow(abort.signal),
  1085. abort,
  1086. frames => frames.filter(frame => frame.type === 'upsert').length >= 2,
  1087. )
  1088. await new Promise(resolve => setTimeout(resolve, 10))
  1089. const wsid = 'fx-ws-fixture' as WorkspaceId
  1090. const missing = await api.workspace.rename(req({ workspaceId: 'fx-ws-void' as WorkspaceId, title: 'x' }))
  1091. expect(missing.result).toMatchObject({ ok: false, error: { code: 'workspace/not-found', details: { workspaceId: 'fx-ws-void' } } })
  1092. await api.workspace.create(req({ path: '/tmp/fixture-workspaces/occupied' }))
  1093. const conflict = await api.workspace.rename(req({ workspaceId: wsid, title: ' occupied ' }))
  1094. expect(conflict.result).toMatchObject({ ok: false, error: { code: 'workspace/name-conflict', details: { name: 'occupied' } } })
  1095. const noop = await api.workspace.rename(req({ workspaceId: wsid, title: ' fixture ' }))
  1096. if (!noop.result.ok) throw new Error('no-op rename failed')
  1097. expect(noop.result.value.workspace.title).toBe('fixture')
  1098. const renamed = await api.workspace.rename(req({ workspaceId: wsid, title: 'renamed' }))
  1099. if (!renamed.result.ok) throw new Error('rename failed')
  1100. expect(renamed.result.value.workspace.title).toBe('renamed')
  1101. const frames = await consuming
  1102. // Only the create and the effective rename emit frames; the no-op stays silent.
  1103. const upserts = frames.filter(frame => frame.type === 'upsert')
  1104. expect(upserts).toHaveLength(2)
  1105. expect(upserts[1]).toMatchObject({ workspace: { workspaceId: wsid, title: 'renamed' } })
  1106. })
  1107. it('session.rename covers not-found, blank title, and the accepted append + title frame', async () => {
  1108. const api = createFixtureApi()
  1109. const followAbort = new AbortController()
  1110. const controlAbort = new AbortController()
  1111. const followPromise = collectValues(
  1112. api.sessionRemote.follow(sid('fx-alpha'), followAbort.signal),
  1113. followAbort,
  1114. frames => frames.some(frame => frame.type === 'event'
  1115. && (frame.event as { type: string }).type === 'session/title'),
  1116. )
  1117. const controlPromise = collectValues(
  1118. api.sessionRemote.control(controlAbort.signal),
  1119. controlAbort,
  1120. frames => frames.some(frame => frame.type === 'projection' && frame.key === 'title' && frame.value === '重命名'),
  1121. )
  1122. await new Promise(resolve => setTimeout(resolve, 10))
  1123. const missing = await api.sessions.rename(req({ sessionId: sid('fx-void'), title: 'x' }))
  1124. expect(missing.result).toMatchObject({ ok: false, error: { code: 'session/not-found', details: { sessionId: 'fx-void' } } })
  1125. const blank = await api.sessions.rename(req({ sessionId: sid('fx-alpha'), title: ' ' }))
  1126. expect(blank.result).toMatchObject({ ok: false, error: { code: 'session/title-invalid', details: { sessionId: 'fx-alpha' } } })
  1127. const renamed = await api.sessions.rename(req({ sessionId: sid('fx-alpha'), title: ' 重命名 ' }))
  1128. if (!renamed.result.ok) throw new Error('rename failed')
  1129. expect(renamed.result.value.title).toBe('重命名')
  1130. const acceptedSeq = renamed.result.value.seq
  1131. // The response seq addresses the appended title event (the client plane
  1132. // has no session/title in its event union — titles ride the projection —
  1133. // so the event is located by seq and its payload checked structurally).
  1134. const history = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 100 }))
  1135. if (!history.result.ok) throw new Error('history failed')
  1136. const appended = historyEvents(history.result.value.records).find(event => event.seq === acceptedSeq)
  1137. expect(appended).toMatchObject({
  1138. type: 'session/title',
  1139. data: { title: '重命名', messageSeqs: [], source: { kind: 'user' } },
  1140. })
  1141. const followed = await followPromise
  1142. expect(followed.some(frame => frame.type === 'event'
  1143. && frame.event.seq === acceptedSeq
  1144. && (frame.event as { readonly type: string }).type === 'session/title')).toBe(true)
  1145. const frames = await controlPromise
  1146. const titleFrames = frames.filter(frame =>
  1147. frame.type === 'projection'
  1148. && frame.key === 'title'
  1149. && frame.sessionId === sid('fx-alpha')
  1150. && frame.value === '重命名')
  1151. expect(titleFrames).toHaveLength(1)
  1152. expect(titleFrames[0]).toMatchObject({ seq: acceptedSeq })
  1153. })
  1154. it('workspace.insertSessionBefore moves, appends, no-ops, and rejects invalid ids', async () => {
  1155. const api = createFixtureApi()
  1156. const wsid = 'fx-ws-fixture' as WorkspaceId
  1157. const missing = await api.workspace.insertSessionBefore(req({ workspaceId: 'fx-ws-void' as WorkspaceId, sessionId: sid('fx-alpha') }))
  1158. expect(missing.result).toMatchObject({ ok: false, error: { code: 'workspace/not-found' } })
  1159. const ghost = await api.workspace.insertSessionBefore(req({ workspaceId: wsid, sessionId: sid('fx-ghost') }))
  1160. expect(ghost.result).toMatchObject({ ok: false, error: { code: 'workspace/move-invalid', details: { sessionId: 'fx-ghost' } } })
  1161. const badAnchor = await api.workspace.insertSessionBefore(req({ workspaceId: wsid, sessionId: sid('fx-alpha'), beforeSessionId: sid('fx-ghost') }))
  1162. expect(badAnchor.result).toMatchObject({ ok: false, error: { code: 'workspace/move-invalid', details: { beforeSessionId: 'fx-ghost' } } })
  1163. const moved = await api.workspace.insertSessionBefore(req({ workspaceId: wsid, sessionId: sid('fx-gamma'), beforeSessionId: sid('fx-beta') }))
  1164. if (!moved.result.ok) throw new Error('move failed')
  1165. expect(moved.result.value.workspace.sessionIds).toEqual(['fx-alpha', 'fx-gamma', 'fx-beta'])
  1166. const appended = await api.workspace.insertSessionBefore(req({ workspaceId: wsid, sessionId: sid('fx-alpha') }))
  1167. if (!appended.result.ok) throw new Error('append failed')
  1168. expect(appended.result.value.workspace.sessionIds).toEqual(['fx-gamma', 'fx-beta', 'fx-alpha'])
  1169. const before = appended.result.value.workspace.updatedAt
  1170. const noop = await api.workspace.insertSessionBefore(req({ workspaceId: wsid, sessionId: sid('fx-alpha') }))
  1171. if (!noop.result.ok) throw new Error('no-op move failed')
  1172. expect(noop.result.value.workspace.sessionIds).toEqual(['fx-gamma', 'fx-beta', 'fx-alpha'])
  1173. expect(noop.result.value.workspace.updatedAt).toBe(before)
  1174. })
  1175. it('workspace.delete removes only the Workspace row and emits the removal frame', async () => {
  1176. const api = createFixtureApi()
  1177. const abort = new AbortController()
  1178. const consuming = collectValues(
  1179. api.workspaceRemote.follow(abort.signal),
  1180. abort,
  1181. frames => frames.some(frame => frame.type === 'remove'),
  1182. )
  1183. await new Promise(resolve => setTimeout(resolve, 10))
  1184. const missing = await api.workspace.delete(req({ workspaceId: 'fx-ws-void' as WorkspaceId }))
  1185. expect(missing.result).toMatchObject({ ok: false, error: { code: 'workspace/not-found' } })
  1186. const deleted = await api.workspace.delete(req({ workspaceId: 'fx-ws-fixture' as WorkspaceId }))
  1187. expect(deleted.result).toEqual({ ok: true, value: { deleted: true } })
  1188. const frames = await consuming
  1189. expect(frames.at(-1)).toEqual({ type: 'remove', workspaceId: 'fx-ws-fixture' })
  1190. const baseline = await readWorkspaceBaseline(api.workspaceRemote)
  1191. expect(baseline.items.some(workspace => workspace.workspaceId === 'fx-ws-fixture')).toBe(false)
  1192. const sessions = await api.sessions.list(req({}))
  1193. if (!sessions.result.ok) throw new Error('session list failed')
  1194. expect(sessions.result.value.items.map(session => session.sessionId)).toContain('fx-alpha')
  1195. })
  1196. it('session.create({workspaceId}) lands on the account and unknown ids error', async () => {
  1197. const api = createFixtureApi()
  1198. const hostAbort = new AbortController()
  1199. const workspaceAbort = new AbortController()
  1200. const seen: FixtureRemoteEventNotificationFrame[] = []
  1201. const consuming = (async () => {
  1202. for await (const frame of api.remoteEvents(hostAbort.signal)) {
  1203. if (frame.type !== 'emit' || frame.event !== 'api-session/added') continue
  1204. seen.push(frame)
  1205. hostAbort.abort()
  1206. break
  1207. }
  1208. })()
  1209. const workspaceFrames = collectValues(
  1210. api.workspaceRemote.follow(workspaceAbort.signal),
  1211. workspaceAbort,
  1212. frames => frames.some(frame => frame.type === 'upsert'
  1213. && frame.workspace.sessionIds.length === 4),
  1214. )
  1215. await new Promise(resolve => setTimeout(resolve, 10))
  1216. const missing = await api.sessions.create(req({ workspaceId: 'fx-ws-void' as WorkspaceId }))
  1217. expect(missing.result).toMatchObject({ ok: false, error: { code: 'workspace/not-found', details: { workspaceId: 'fx-ws-void' } } })
  1218. const created = await api.sessions.create(req({ workspaceId: 'fx-ws-fixture' as WorkspaceId }))
  1219. if (!created.result.ok) throw new Error('create failed')
  1220. const id = created.result.value.sessionId
  1221. await consuming
  1222. const added = seen[0]
  1223. expect(added).toMatchObject({
  1224. event: 'api-session/added',
  1225. args: [{ sessionId: id, blank: true, cwd: '/tmp/fixture' }],
  1226. })
  1227. expect((await workspaceFrames).at(-1)).toMatchObject({
  1228. type: 'upsert',
  1229. workspace: {
  1230. workspaceId: 'fx-ws-fixture',
  1231. sessionIds: [id, 'fx-alpha', 'fx-beta', 'fx-gamma'],
  1232. },
  1233. })
  1234. })
  1235. it('supports an empty baseline, preallocated ids, independent streams, and idempotent retry', async () => {
  1236. const api = createFixtureApi({ empty: true, createFrameOrder: 'workspace-first' })
  1237. const initialSessions = await api.sessions.list(req({}))
  1238. expect(initialSessions.result).toMatchObject({ ok: true, value: { items: [] } })
  1239. expect(await readWorkspaceBaseline(api.workspaceRemote)).toEqual({
  1240. items: [],
  1241. archivedSessionIds: [],
  1242. })
  1243. const made = await api.workspace.create(req({ path: '/tmp/fixture-workspaces/nova' }))
  1244. if (!made.result.ok) throw new Error('workspace create failed')
  1245. const hostAbort = new AbortController()
  1246. const workspaceAbort = new AbortController()
  1247. const hostFrames = collectValues(
  1248. api.remoteEvents(hostAbort.signal),
  1249. hostAbort,
  1250. frames => frames.length === 1,
  1251. )
  1252. const workspaceFrames = collectValues(
  1253. api.workspaceRemote.follow(workspaceAbort.signal),
  1254. workspaceAbort,
  1255. frames => frames.some(frame => frame.type === 'upsert'
  1256. && frame.workspace.sessionIds.includes(sid('fx-preallocated'))),
  1257. )
  1258. await new Promise(resolve => setTimeout(resolve, 10))
  1259. const preallocated = sid('fx-preallocated')
  1260. const created = await api.sessions.create(req({
  1261. workspaceId: made.result.value.workspace.workspaceId,
  1262. sessionId: preallocated,
  1263. }))
  1264. expect(created.result).toEqual({ ok: true, value: { sessionId: preallocated } })
  1265. expect((await workspaceFrames).at(-1)).toMatchObject({
  1266. type: 'upsert', workspace: { sessionIds: [preallocated] },
  1267. })
  1268. const added = (await hostFrames)[0]
  1269. expect(added).toMatchObject({
  1270. event: 'api-session/added',
  1271. args: [{
  1272. sessionId: preallocated,
  1273. blank: true,
  1274. cwd: made.result.value.workspace.path,
  1275. }],
  1276. })
  1277. const retried = await api.sessions.create(req({
  1278. workspaceId: made.result.value.workspace.workspaceId,
  1279. sessionId: preallocated,
  1280. }))
  1281. expect(retried.result).toEqual({ ok: true, value: { sessionId: preallocated } })
  1282. const listed = await api.sessions.list(req({}))
  1283. if (!listed.result.ok) throw new Error('session list failed')
  1284. expect(listed.result.value.items.filter(item => item.sessionId === preallocated)).toHaveLength(1)
  1285. const conflict = await api.sessions.create(req({ sessionId: preallocated, cwd: '/elsewhere' }))
  1286. expect(conflict.result).toMatchObject({
  1287. ok: false,
  1288. error: { code: 'session/conflict', details: { sessionId: preallocated, requestedCwd: '/elsewhere' } },
  1289. })
  1290. })
  1291. it('attaches an existing ungrouped Session to a matching Workspace', async () => {
  1292. const api = createFixtureApi()
  1293. const sessionId = sid('fx-existing-ungrouped')
  1294. await expect(api.sessions.create(req({ sessionId, cwd: '/tmp/fixture' }))).resolves.toMatchObject({
  1295. result: { ok: true, value: { sessionId } },
  1296. })
  1297. await expect(api.sessions.create(req({
  1298. sessionId,
  1299. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1300. }))).resolves.toMatchObject({ result: { ok: true, value: { sessionId } } })
  1301. const workspaces = await readWorkspaceBaseline(api.workspaceRemote)
  1302. expect(workspaces.items[0]?.sessionIds).toContain(sessionId)
  1303. })
  1304. it('reports a conflict without an existing cwd detail for an unrecorded cwd', async () => {
  1305. const api = createFixtureApi()
  1306. const listed = await api.sessions.list(req({}))
  1307. if (!listed.result.ok) throw new Error('session list failed')
  1308. const existing = listed.result.value.items.find(item => item.sessionId === sid('fx-alpha'))
  1309. if (existing === undefined) throw new Error('fixture Session missing')
  1310. delete existing.cwd
  1311. const conflict = await api.sessions.create(req({ sessionId: existing.sessionId }))
  1312. expect(conflict.result).toEqual({
  1313. ok: false,
  1314. error: {
  1315. code: 'session/conflict',
  1316. message: `session ${existing.sessionId} already uses no cwd`,
  1317. details: { sessionId: existing.sessionId, requestedCwd: '/tmp/fixture' },
  1318. },
  1319. })
  1320. })
  1321. it('publishes an ungrouped Session when Workspace attachment fails', async () => {
  1322. const api = createFixtureApi({ failWorkspaceAttach: true })
  1323. const sessionId = sid('fx-partial')
  1324. const created = await api.sessions.create(req({
  1325. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1326. sessionId,
  1327. }))
  1328. expect(created.result).toMatchObject({
  1329. ok: false,
  1330. error: { code: 'session/workspace-attach-failed', details: { sessionId, workspaceId: 'fx-ws-fixture' } },
  1331. })
  1332. const listed = await api.sessions.list(req({}))
  1333. const workspaces = await readWorkspaceBaseline(api.workspaceRemote)
  1334. if (!listed.result.ok) throw new Error('list failed')
  1335. expect(listed.result.value.items.filter(item => item.sessionId === sessionId)).toHaveLength(1)
  1336. expect(workspaces.items[0]?.sessionIds).not.toContain(sessionId)
  1337. const retried = await api.sessions.create(req({
  1338. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1339. sessionId,
  1340. }))
  1341. expect(retried.result).toMatchObject({ ok: false, error: { code: 'session/workspace-attach-failed' } })
  1342. const afterRetry = await api.sessions.list(req({}))
  1343. if (!afterRetry.result.ok) throw new Error('list failed')
  1344. expect(afterRetry.result.value.items.filter(item => item.sessionId === sessionId)).toHaveLength(1)
  1345. })
  1346. it('reconciles a dropped create response and can reject a prompt before acceptance', async () => {
  1347. const sessionId = sid('fx-lost-response')
  1348. const dropped = createFixtureApi({ dropSessionCreateResponse: true })
  1349. await expect(Promise.resolve().then(() => dropped.sessions.create(req({
  1350. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1351. sessionId,
  1352. })))).rejects.toThrow(/dropped session\.create response/)
  1353. const listed = await dropped.sessions.list(req({}))
  1354. const workspaces = await readWorkspaceBaseline(dropped.workspaceRemote)
  1355. if (!listed.result.ok) throw new Error('list failed')
  1356. expect(listed.result.value.items.some(item => item.sessionId === sessionId)).toBe(true)
  1357. expect(workspaces.items[0]?.sessionIds).toContain(sessionId)
  1358. await expect(dropped.sessions.create(req({
  1359. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1360. sessionId,
  1361. }))).resolves.toMatchObject({ result: { ok: true, value: { sessionId } } })
  1362. const rejecting = createFixtureApi({ empty: true, rejectPrompt: true })
  1363. const real = await rejecting.sessions.create(req({ sessionId: sid('fx-rejected') }))
  1364. if (!real.result.ok) throw new Error('session create failed')
  1365. const prompt = await rejecting.sessions.prompt(req({
  1366. sessionId: real.result.value.sessionId,
  1367. mode: 'queue' as const,
  1368. content: [{ type: 'text' as const, text: 'keep me' }],
  1369. }))
  1370. expect(prompt.result).toMatchObject({ ok: false, error: { code: 'session/agent-busy' } })
  1371. const imagePrompt = await rejecting.sessions.prompt(req({
  1372. sessionId: real.result.value.sessionId,
  1373. mode: 'queue' as const,
  1374. content: [{ type: 'image' as const, mediaType: 'image/png' as const, data: 'iVBORw0KGgo=' }],
  1375. }))
  1376. expect(imagePrompt.result).toMatchObject({
  1377. ok: false,
  1378. error: { code: 'session/attachment-invalid', details: { reason: 'IMAGE_DIMENSION_TOO_LARGE' } },
  1379. })
  1380. })
  1381. it('timing hooks: history delay + one-shot failure, silent append, and breakStreams end open generators', async () => {
  1382. const api = createFixtureApi()
  1383. const hooks = timing()
  1384. // One-shot transport failure after transit delay.
  1385. hooks.setHistoryDelay(5)
  1386. hooks.failNextHistory()
  1387. await expect(api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 5 }))).rejects.toThrow(/simulated history transport failure/)
  1388. hooks.setHistoryDelay(0)
  1389. // The failure was one-shot: the next call succeeds.
  1390. const ok = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 5 }))
  1391. expect(ok.result.ok).toBe(true)
  1392. // A durable append without a live frame creates a detectable seq gap.
  1393. const gapAbort = new AbortController()
  1394. const gapIterator = api.sessionRemote.follow(sid('fx-alpha'), gapAbort.signal)[Symbol.asyncIterator]()
  1395. const opening = await gapIterator.next()
  1396. if (opening.done || opening.value.type !== 'snapshot') throw new Error('follow opening snapshot missing')
  1397. hooks.appendSilent('fx-alpha', '静默丢帧')
  1398. hooks.appendUser('fx-alpha', '正常直播')
  1399. await expect(gapIterator.next()).rejects.toThrow(/stream skipped seq/)
  1400. // Reopening replaces the window with a complete snapshot containing both durable events.
  1401. const followAbort = new AbortController()
  1402. const controlAbort = new AbortController()
  1403. const followed: FixtureFollowFrame[] = []
  1404. const controlled: FixtureControlFrame[] = []
  1405. const following = (async () => {
  1406. for await (const frame of api.sessionRemote.follow(sid('fx-alpha'), followAbort.signal)) {
  1407. followed.push(frame)
  1408. }
  1409. })()
  1410. const controlling = (async () => {
  1411. for await (const frame of api.sessionRemote.control(controlAbort.signal)) controlled.push(frame)
  1412. })()
  1413. await new Promise(resolve => setTimeout(resolve, 10))
  1414. await vi.waitFor(() => {
  1415. const snapshot = followed.find(frame => frame.type === 'snapshot')
  1416. const events = snapshot === undefined ? [] : historyEvents(snapshot.records)
  1417. expect(events.some(event => JSON.stringify(event.data).includes('静默丢帧'))).toBe(true)
  1418. expect(events.some(event => JSON.stringify(event.data).includes('正常直播'))).toBe(true)
  1419. })
  1420. hooks.appendTitle('fx-alpha', 'Fixture 修订标题')
  1421. hooks.beginModelRetry('fx-alpha')
  1422. hooks.scheduleModelRetry('fx-alpha')
  1423. hooks.completeModelRetry('fx-alpha')
  1424. hooks.beginModelRetry('fx-alpha')
  1425. hooks.cancelModelRetryDuringBackoff('fx-alpha')
  1426. await vi.waitFor(() => {
  1427. expect(followed.some(frame => frame.type === 'event' && (frame.event as { type: string }).type === 'llm/retry')).toBe(true)
  1428. expect(followed.some(frame => frame.type === 'event' && JSON.stringify(frame.event.data).includes('重试后的完整回复'))).toBe(true)
  1429. expect(followed.some(frame => frame.type === 'event'
  1430. && frame.event.type === 'turn/end'
  1431. && frame.event.data.reason.kind === 'aborted')).toBe(true)
  1432. expect(controlled.some(frame => frame.type === 'projection'
  1433. && frame.key === 'title'
  1434. && frame.value === 'Fixture 修订标题')).toBe(true)
  1435. })
  1436. expect(followed.some(frame => frame.type === 'event' && (frame.event as { type: string }).type === 'session/title')).toBe(true)
  1437. // Paging and resumed follow agree on the recovered durable event.
  1438. const repull = await api.sessions.history(req({ sessionId: sid('fx-alpha'), maxMessages: 5 }))
  1439. if (!repull.result.ok) throw new Error('repull failed')
  1440. expect(JSON.stringify(repull.result.value.records)).toContain('静默丢帧')
  1441. // breakStreams force-ends follow and control without client aborts.
  1442. await new Promise(resolve => setTimeout(resolve, 10))
  1443. hooks.breakStreams()
  1444. await following
  1445. await controlling
  1446. expect(followAbort.signal.aborted).toBe(false)
  1447. expect(controlAbort.signal.aborted).toBe(false)
  1448. })
  1449. it('paces the opt-in reasoning stress hook from an external interval', async () => {
  1450. vi.useFakeTimers()
  1451. vi.setSystemTime(0)
  1452. const api = createFixtureApi()
  1453. const hooks = timing()
  1454. expect(hooks.reasoningChunkStormState()).toBeNull()
  1455. expect(() => hooks.startReasoningChunkStorm('fx-alpha', 0, 1, 16)).toThrow(/chunk count/)
  1456. expect(() => hooks.startReasoningChunkStorm('fx-alpha', 1, 0, 16)).toThrow(/chunks per interval/)
  1457. expect(() => hooks.startReasoningChunkStorm('fx-alpha', 1, 1, 0)).toThrow(/reasoning interval/)
  1458. const abort = new AbortController()
  1459. try {
  1460. const streamed = collectValues(api.sessionRemote.follow(sid('fx-alpha'), abort.signal), abort, frames => frames.some(frame => (
  1461. frame.type === 'event'
  1462. && frame.event.type === 'assistant/chunk'
  1463. && frame.event.data.chunk.type === 'reasoning-delta'
  1464. && frame.event.data.chunk.text.includes('REASONING_STRESS_COMPLETE')
  1465. )))
  1466. const marker = hooks.startReasoningChunkStorm('fx-alpha', 3, 2, 16)
  1467. expect(() => hooks.startReasoningChunkStorm('fx-alpha', 1, 1, 16)).toThrow(/already running/)
  1468. expect(hooks.reasoningChunkStormState()).toMatchObject({ emitted: 0, emitting: true, marker })
  1469. await vi.advanceTimersByTimeAsync(0)
  1470. expect(hooks.reasoningChunkStormState()).toMatchObject({ emitted: 2, emitting: true })
  1471. await vi.advanceTimersByTimeAsync(16)
  1472. expect(hooks.reasoningChunkStormState()).toEqual({
  1473. sessionId: 'fx-alpha', chunkCount: 3, chunksPerInterval: 2, intervalMs: 16,
  1474. emitted: 3, marker, emitting: false,
  1475. })
  1476. const frames = await streamed
  1477. const deltas = frames.flatMap(frame => (
  1478. frame.type === 'event'
  1479. && frame.event.type === 'assistant/chunk'
  1480. && frame.event.data.chunk.type === 'reasoning-delta'
  1481. ? [frame.event.data.chunk.text]
  1482. : []
  1483. ))
  1484. expect(deltas).toEqual(['推理', '推理', `\n${marker}`])
  1485. } finally {
  1486. abort.abort()
  1487. vi.useRealTimers()
  1488. }
  1489. })
  1490. })
  1491. describe('fixture Connection RPC', () => {
  1492. afterEach(() => {
  1493. vi.restoreAllMocks()
  1494. vi.unstubAllGlobals()
  1495. })
  1496. it('covers the migrated Remote dispatch table', async () => {
  1497. const rpc = createFixtureConnectionRpc()
  1498. const sessions = createSessionClient(rpc)
  1499. const workspaces = createWorkspaceClient(rpc)
  1500. expect((await sessions.search(
  1501. { query: 'fixture' },
  1502. new AbortController().signal,
  1503. )).result.ok).toBe(true)
  1504. const created = await sessions.create({})
  1505. if (!created.result.ok) throw new Error('create failed')
  1506. const id = created.result.value.sessionId
  1507. expect((await sessions.history({ sessionId: id })).result.ok).toBe(true)
  1508. expect((await sessions.prompt({ sessionId: id, mode: 'queue', content: [{ type: 'text', text: '嗨' }] })).result.ok).toBe(true)
  1509. expect((await sessions.cancel({ sessionId: id })).result.ok).toBe(true)
  1510. expect((await readWorkspaceBaseline(createWorkspaceRemote(rpc))).items).not.toHaveLength(0)
  1511. const workspace = await workspaces.create({ path: '/tmp/fixture-workspaces/via-client' })
  1512. if (!workspace.result.ok) throw new Error('workspace create failed')
  1513. expect(workspace.result.value.workspace.title).toBe('via-client')
  1514. const wsid = workspace.result.value.workspace.workspaceId
  1515. const renamed = await workspaces.rename({ workspaceId: wsid, title: 'via-client-2' })
  1516. if (!renamed.result.ok) throw new Error('workspace rename failed')
  1517. expect(renamed.result.value.workspace.title).toBe('via-client-2')
  1518. const attached = await sessions.create({ workspaceId: wsid })
  1519. if (!attached.result.ok) throw new Error('attached create failed')
  1520. const moved = await workspaces.insertSessionBefore({ workspaceId: wsid, sessionId: attached.result.value.sessionId })
  1521. if (!moved.result.ok) throw new Error('workspace move failed')
  1522. expect(moved.result.value.workspace.sessionIds).toEqual([attached.result.value.sessionId])
  1523. })
  1524. it('folds the goal lifecycle over the Goal Remotes', async () => {
  1525. const rpc = createFixtureConnectionRpc()
  1526. const sessions = createSessionClient(rpc)
  1527. const created = await sessions.create({})
  1528. if (!created.result.ok) throw new Error('create failed')
  1529. const id = created.result.value.sessionId
  1530. const goal = (endpoint: string, args: Record<string, unknown>) =>
  1531. rpc.call('/api', endpoint, { args: { agentId: id, ...args } })
  1532. // create → edit → pause → resume → complete → clear; each mutation advances the CAS
  1533. // revision by one (state rides the projection frames).
  1534. const goalCreated = await goal('goals/create', { request: { objective: 'ship it' } })
  1535. if (!goalCreated.ok) throw new Error('goal create failed')
  1536. const { id: goalId, revision } = (goalCreated.value as { ref: { id: string; revision: number } }).ref
  1537. expect(revision).toBe(1)
  1538. const ref = (at: number) => ({ id: goalId, revision: at })
  1539. expect((await goal('goals/edit', { ref: ref(1), request: { objective: 'ship it v2' } })).ok).toBe(true)
  1540. expect((await goal('goals/pause', { ref: ref(2) })).ok).toBe(true)
  1541. expect((await goal('goals/resume', { ref: ref(3) })).ok).toBe(true)
  1542. // A stale ref loses the CAS check.
  1543. expect((await goal('goals/pause', { ref: ref(1) })).ok).toBe(false)
  1544. expect((await goal('goals/complete', { ref: ref(4) })).ok).toBe(true)
  1545. // complete → complete is an invalid transition.
  1546. expect((await goal('goals/complete', { ref: ref(5) })).ok).toBe(false)
  1547. expect(await goal('goals/clear', { ref: ref(5) })).toEqual({ ok: true, value: ref(6) })
  1548. const goalHistory = await sessions.history({ sessionId: id })
  1549. if (!goalHistory.result.ok) throw new Error('goal history failed')
  1550. const goalEvents = historyEvents(goalHistory.result.value.records).map(event => event as unknown as {
  1551. type: string
  1552. data: {
  1553. operation?: string
  1554. source?: { kind?: string; round?: number }
  1555. }
  1556. })
  1557. const goalChanges = goalEvents.filter(event => event.type === 'goal/change')
  1558. expect(goalChanges.map(event => event.data.operation))
  1559. .toEqual(['create', 'edit', 'pause', 'resume', 'complete', 'clear'])
  1560. expect(goalEvents.some(event => event.type === 'user/message'
  1561. && event.data.source?.kind === 'goal' && event.data.source.round === 0)).toBe(false)
  1562. })
  1563. it('maps empty, prompt-reject, and workspace-first query scenarios', async () => {
  1564. vi.stubGlobal('location', {
  1565. search: '?fixture=empty&fixturePrompt=reject&fixtureFrames=workspace-first',
  1566. })
  1567. const rpc = createFixtureConnectionRpc()
  1568. const sessions = createSessionClient(rpc)
  1569. const workspaces = createWorkspaceClient(rpc)
  1570. const workspaceRemote = createWorkspaceRemote(rpc)
  1571. await expect(sessions.list({})).resolves.toMatchObject({ result: { ok: true, value: { items: [] } } })
  1572. const made = await workspaces.create({ path: '/tmp/fixture-workspaces/query-workspace' })
  1573. if (!made.result.ok) throw new Error('workspace create failed')
  1574. const hostAbort = new AbortController()
  1575. const workspaceAbort = new AbortController()
  1576. const hostFrames = collectValues(
  1577. openFixtureRemoteEvents(rpc, hostAbort.signal),
  1578. hostAbort,
  1579. frames => frames.length === 1,
  1580. )
  1581. const workspaceFrames = collectValues(
  1582. workspaceRemote.follow(workspaceAbort.signal),
  1583. workspaceAbort,
  1584. frames => frames.some(frame => frame.type === 'upsert'
  1585. && frame.workspace.sessionIds.includes(sid('fx-query-session'))),
  1586. )
  1587. await new Promise(resolve => setTimeout(resolve, 10))
  1588. const sessionId = sid('fx-query-session')
  1589. const created = await sessions.create({
  1590. workspaceId: made.result.value.workspace.workspaceId,
  1591. sessionId,
  1592. })
  1593. expect(created.result).toMatchObject({ ok: true, value: { sessionId } })
  1594. expect((await workspaceFrames).at(-1)).toMatchObject({
  1595. type: 'upsert',
  1596. workspace: { sessionIds: [sessionId] },
  1597. })
  1598. expect((await hostFrames)[0]).toMatchObject({
  1599. event: 'api-session/added',
  1600. })
  1601. const rejected = await sessions.prompt({
  1602. sessionId,
  1603. mode: 'queue',
  1604. content: [{ type: 'text', text: 'retain' }],
  1605. })
  1606. expect(rejected.result).toMatchObject({ ok: false, error: { code: 'session/agent-busy' } })
  1607. })
  1608. it('maps attach-failure and dropped-response query scenarios', async () => {
  1609. vi.stubGlobal('location', { search: '?fixture&fixtureAttach=fail' })
  1610. const partial = createFixtureConnectionRpc()
  1611. const partialResult = await createSessionClient(partial).create({
  1612. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1613. sessionId: sid('fx-query-partial'),
  1614. })
  1615. expect(partialResult.result).toMatchObject({
  1616. ok: false,
  1617. error: { code: 'session/workspace-attach-failed', details: { sessionId: 'fx-query-partial' } },
  1618. })
  1619. vi.stubGlobal('location', { search: '?fixture&fixtureSessionCreate=drop-response' })
  1620. const dropped = createFixtureConnectionRpc()
  1621. await expect(createSessionClient(dropped).create({
  1622. workspaceId: 'fx-ws-fixture' as WorkspaceId,
  1623. sessionId: sid('fx-query-dropped'),
  1624. })).rejects.toThrow(/dropped session\.create response/)
  1625. })
  1626. })