model.client.spec.ts 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567
  1. /** Browser view ownership across slow RPCs, remounts and transport generations. */
  2. import { setImmediate } from 'node:timers/promises'
  3. import { afterEach, expect, it, vi } from 'vitest'
  4. import { RemoteStream, RemoteStreamCarrierError, type ClientRemote, type RemoteStreamOptions } from '@deepseek-ai/dsh-api-gateway/client'
  5. import { createSnapshotStore } from '@deepseek-ai/dsh-client-store'
  6. import { RemoteError, type RemoteResult } from '@deepseek-ai/dsh-typert-protocol'
  7. import type {} from '@deepseek-ai/dsh-api-terminal-controller/remote'
  8. import type { SessionId } from '@deepseek-ai/dsh-session/types'
  9. import { TerminalView, type TerminalRemote } from '../src/client/model.ts'
  10. import type { TerminalEnvironment, TerminalFrame, WebTerminalId, WebTerminalInfo } from '../src/types.ts'
  11. const sessionId = 'session' as SessionId
  12. const info: WebTerminalInfo = { id: 'terminal' as WebTerminalId, shell: { name: 'bash', path: '/bin/bash', args: ['-i'] }, title: 'bash', cwd: '/workspace', rows: 24, cols: 80, state: 'running', exitCode: null }
  13. const environment: TerminalEnvironment = {
  14. cwd: info.cwd, maxInputBytes: 1000, maxCols: 200, maxRows: 100, scrollback: 100,
  15. }
  16. const success = <T>(value: T): RemoteResult<T> => ({ ok: true, value })
  17. const cleanups: (() => void | Promise<void>)[] = []
  18. afterEach(async () => { for (const close of cleanups.splice(0).reverse()) await close() })
  19. function untilAborted(signal?: AbortSignal): Promise<void> {
  20. return new Promise((resolve) => {
  21. if (signal?.aborted) resolve()
  22. else signal?.addEventListener('abort', () => { resolve() }, { once: true })
  23. })
  24. }
  25. function failure(message: string): RemoteResult<never> {
  26. return { ok: false, error: new RemoteError('gateway/bad-request', message, {}) }
  27. }
  28. function fixture(prepareStream?: <Item>(stream: RemoteStream<Item>) => void) {
  29. const remote: TerminalRemote = {
  30. shells: vi.fn<TerminalRemote['shells']>(async () => success([info.shell])),
  31. environment: vi.fn<TerminalRemote['environment']>(async () => success(environment)), list: vi.fn<TerminalRemote['list']>(async () => success([])),
  32. create: vi.fn<TerminalRemote['create']>(async (_sessionId, request) => success({ ...info, id: request.id })),
  33. close: vi.fn<TerminalRemote['close']>(async () => success(undefined)), rename: vi.fn<TerminalRemote['rename']>(async () => success(undefined)),
  34. write: vi.fn<TerminalRemote['write']>(async () => success(undefined)), resize: vi.fn<TerminalRemote['resize']>(async () => success(undefined)),
  35. follow: vi.fn<TerminalRemote['follow']>(async function* (_sessionId, id, attachmentId, signal) {
  36. yield { type: 'snapshot', sequence: 0, screen: 'ready', info: { ...info, id, controllerId: attachmentId } }
  37. await untilAborted(signal)
  38. }),
  39. }
  40. const options: RemoteStreamOptions<unknown>[] = []
  41. const streams: { dispose(): Promise<void>; restart(): void }[] = []
  42. const generation = createSnapshotStore<ReturnType<ConstructorParameters<typeof RemoteStream>[0]['generation']['getSnapshot']>>(undefined)
  43. const gateway: Pick<ClientRemote, '$stream'> = {
  44. $stream: (option) => {
  45. options.push(option)
  46. const stream = new RemoteStream({ generation }, option)
  47. streams.push(stream)
  48. prepareStream?.(stream)
  49. return stream
  50. },
  51. }
  52. cleanups.push(async () => { await Promise.all(streams.map(stream => stream.dispose())) })
  53. const model = new TerminalView(sessionId, remote, gateway, info.id)
  54. cleanups.push(() => model.dispose())
  55. return { model, remote, gateway, options, streams, generation }
  56. }
  57. async function mount(model: TerminalView) {
  58. const detach = model.mount()
  59. await model.refresh()
  60. await model.start()
  61. return detach
  62. }
  63. function acknowledge(model: TerminalView): void {
  64. const render = model.state.getSnapshot().render
  65. if (render !== undefined) model.acknowledge(render.revision)
  66. }
  67. async function connected(model: TerminalView): Promise<void> {
  68. await mount(model)
  69. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  70. acknowledge(model)
  71. }
  72. it('does not let a late input failure downgrade a newer connection', async () => {
  73. const { model, remote } = fixture()
  74. await mount(model)
  75. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  76. acknowledge(model)
  77. const input = Promise.withResolvers<RemoteResult<void>>()
  78. vi.mocked(remote.write).mockReturnValueOnce(input.promise)
  79. model.write('x')
  80. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(1)
  81. model.connect()
  82. await expect.poll(() => vi.mocked(remote.follow).mock.calls.length).toBe(2)
  83. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  84. model.write('current input')
  85. input.reject(new Error('old attachment was replaced'))
  86. await expect.poll(() => vi.mocked(remote.write).mock.calls.some(call => call[3] === 'current input')).toBe(true)
  87. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', writable: true, error: undefined })
  88. })
  89. it('waits for render acknowledgement before publishing subsequent output', async () => {
  90. const { model, remote } = fixture()
  91. vi.mocked(remote.follow).mockImplementation(async function* (_sessionId, id, controllerId, signal) {
  92. const frames: TerminalFrame[] = [
  93. { type: 'snapshot', sequence: 0, screen: '', info: { ...info, id, controllerId } },
  94. { type: 'output', sequence: 1, data: 'first' },
  95. { type: 'output', sequence: 2, data: 'second' },
  96. ]
  97. for (const frame of frames) yield frame
  98. await untilAborted(signal)
  99. })
  100. await mount(model)
  101. await expect.poll(() => model.state.getSnapshot().render?.frame.type).toBe('snapshot')
  102. acknowledge(model)
  103. await expect.poll(() => model.state.getSnapshot().render?.frame).toMatchObject({ type: 'output', sequence: 1 })
  104. acknowledge(model)
  105. await expect.poll(() => model.state.getSnapshot().render?.frame).toMatchObject({ type: 'output', sequence: 2 })
  106. })
  107. it('serializes input and resizes, clamps geometry, and suppresses unchanged sizes', async () => {
  108. const { model, remote } = fixture()
  109. await connected(model)
  110. const input = Promise.withResolvers<RemoteResult<void>>()
  111. vi.mocked(remote.write).mockReturnValueOnce(input.promise)
  112. model.resize(80, 24)
  113. model.write('first')
  114. model.write('second')
  115. model.resize(500, 300)
  116. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(1)
  117. expect(remote.resize).not.toHaveBeenCalled()
  118. input.resolve(success(undefined))
  119. await expect.poll(() => vi.mocked(remote.resize).mock.calls.length).toBe(1)
  120. const attachmentId = vi.mocked(remote.follow).mock.calls[0]![2]
  121. const id = model.state.getSnapshot().info!.id
  122. expect(vi.mocked(remote.write).mock.calls.map(call => call[3])).toEqual(['first', 'second'])
  123. expect(remote.resize).toHaveBeenCalledWith(sessionId, id, attachmentId, 200, 100)
  124. model.resize(90, 24)
  125. await expect.poll(() => vi.mocked(remote.resize).mock.calls.length).toBe(2)
  126. })
  127. it('bounds queued input by UTF-8 bytes and releases the byte budget after settlement', async () => {
  128. const { model, remote } = fixture()
  129. vi.mocked(remote.environment).mockResolvedValue(success({ ...environment, maxInputBytes: 6 }))
  130. await connected(model)
  131. const input = Promise.withResolvers<RemoteResult<void>>()
  132. vi.mocked(remote.write).mockReturnValueOnce(input.promise)
  133. model.write('界界')
  134. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(1)
  135. model.write('a')
  136. expect(model.state.getSnapshot()).toMatchObject({ phase: 'failed', issue: 'inputFull' })
  137. input.resolve(success(undefined))
  138. model.connect()
  139. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  140. model.write('界界')
  141. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(2)
  142. expect(vi.mocked(remote.write).mock.calls.map(call => call[3])).toEqual(['界界', '界界'])
  143. })
  144. it('drops queued commands from a detached attachment and ignores its late resize failure', async () => {
  145. const { model, remote } = fixture()
  146. await connected(model)
  147. const input = Promise.withResolvers<RemoteResult<void>>()
  148. vi.mocked(remote.resize).mockReturnValueOnce(input.promise)
  149. model.resize(100, 30)
  150. await expect.poll(() => vi.mocked(remote.resize).mock.calls.length).toBe(1)
  151. model.write('stale')
  152. model.resize(110, 35)
  153. model.connect()
  154. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  155. model.write('current')
  156. input.resolve(failure('old attachment'))
  157. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(1)
  158. expect(vi.mocked(remote.write).mock.calls[0]?.[3]).toBe('current')
  159. expect(remote.resize).toHaveBeenCalledOnce()
  160. expect(model.state.getSnapshot().phase).toBe('connected')
  161. })
  162. it('publishes current input and resize failures and can reconnect for another attempt', async () => {
  163. const { model, remote } = fixture()
  164. await connected(model)
  165. vi.mocked(remote.write).mockResolvedValueOnce(failure('input refused'))
  166. model.write('x')
  167. await expect.poll(() => model.state.getSnapshot().error).toBe('input refused')
  168. model.connect()
  169. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  170. vi.mocked(remote.resize).mockResolvedValueOnce(failure('resize refused'))
  171. model.resize(100, 30)
  172. await expect.poll(() => model.state.getSnapshot().error).toBe('resize refused')
  173. })
  174. it('treats carrier loss as disconnected only for the active attachment', async () => {
  175. const { model, options } = fixture()
  176. await connected(model)
  177. const first = options[0]!
  178. first.carrierFailed?.(new RemoteStreamCarrierError('offline'))
  179. expect(model.state.getSnapshot()).toMatchObject({ phase: 'disconnected', writable: false })
  180. model.connect()
  181. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  182. first.carrierFailed?.(new RemoteStreamCarrierError('late old failure'))
  183. expect(model.state.getSnapshot().phase).toBe('connected')
  184. })
  185. it.each([
  186. ['output before snapshot', [{ type: 'output', sequence: 1, data: 'orphan' }], 'missing its screen snapshot'],
  187. ['sequence gap', [{ type: 'snapshot', sequence: 4, screen: '', info }, { type: 'output', sequence: 6, data: 'gap' }], 'sequence has a gap'],
  188. ['repeated snapshot', [{ type: 'snapshot', sequence: 4, screen: '', info }, { type: 'snapshot', sequence: 4, screen: '', info }], 'Unexpected terminal screen snapshot'],
  189. ] satisfies readonly (readonly [string, readonly TerminalFrame[], string])[])('refuses a malformed output stream: %s', async (_name, frames, message) => {
  190. const { model, remote } = fixture()
  191. vi.mocked(remote.follow).mockImplementation(async function* () { yield* frames })
  192. await mount(model)
  193. if (frames[0]?.type === 'snapshot') {
  194. await expect.poll(() => model.state.getSnapshot().render).toBeDefined()
  195. acknowledge(model)
  196. }
  197. await expect.poll(() => model.state.getSnapshot().error).toContain(message)
  198. expect(model.state.getSnapshot().writable).toBe(false)
  199. })
  200. it('retains an exited screen and closes controls when its stream ends', async () => {
  201. const { model, remote } = fixture()
  202. vi.mocked(remote.follow).mockImplementation(async function* (_sessionId, id, controllerId) {
  203. yield { type: 'snapshot', sequence: 0, screen: 'last screen', info: { ...info, id, controllerId } }
  204. yield { type: 'state', info: { ...info, id, controllerId, state: 'exited', exitCode: 3 } }
  205. })
  206. await connected(model)
  207. await expect.poll(() => model.state.getSnapshot().phase).toBe('closed')
  208. expect(model.state.getSnapshot()).toMatchObject({ writable: false, info: { state: 'exited', exitCode: 3 }, render: { frame: { screen: 'last screen' } } })
  209. })
  210. it('reports an attachment ending while the shell is still running and permits manual reconnect', async () => {
  211. const { model, remote } = fixture()
  212. vi.mocked(remote.follow).mockImplementationOnce(async function* (_sessionId, id, controllerId) {
  213. yield { type: 'snapshot', sequence: 0, screen: '', info: { ...info, id, controllerId } }
  214. })
  215. await connected(model)
  216. await expect.poll(() => model.state.getSnapshot().issue).toBe('attachmentEnded')
  217. model.connect()
  218. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  219. })
  220. it('publishes remote control transfer without replacing the retained screen', async () => {
  221. const { model, remote } = fixture()
  222. vi.mocked(remote.follow).mockImplementation(async function* (_sessionId, id, controllerId, signal) {
  223. yield { type: 'snapshot', sequence: 0, screen: 'retained', info: { ...info, id, controllerId } }
  224. yield { type: 'state', info: { ...info, id } }
  225. await untilAborted(signal)
  226. })
  227. await connected(model)
  228. await expect.poll(() => model.state.getSnapshot().writable).toBe(false)
  229. expect(model.state.getSnapshot().render?.frame).toMatchObject({ type: 'snapshot', screen: 'retained' })
  230. })
  231. /** Pause delivery after the real Gateway iterator settles, before the model observes it. */
  232. function deliveryBarrier() {
  233. const ready = Promise.withResolvers<undefined>()
  234. const release = Promise.withResolvers<undefined>()
  235. let held = false
  236. function prepare<Item>(stream: RemoteStream<Item>): void {
  237. if (held) return
  238. held = true
  239. const iterator = stream[Symbol.asyncIterator]()
  240. let first = true
  241. vi.spyOn(stream, Symbol.asyncIterator).mockReturnValue({
  242. async next() {
  243. if (!first) return iterator.next()
  244. first = false
  245. try {
  246. return await iterator.next()
  247. } finally {
  248. ready.resolve(undefined)
  249. await release.promise
  250. }
  251. },
  252. async return() { return iterator.return!() },
  253. })
  254. }
  255. cleanups.push(() => { release.resolve(undefined) })
  256. return { ready: ready.promise, release: () => { release.resolve(undefined) }, prepare }
  257. }
  258. it('ignores an already delivered screen when its attachment is replaced before the model receives it', async () => {
  259. const barrier = deliveryBarrier()
  260. const { model, remote } = fixture(barrier.prepare)
  261. await mount(model)
  262. await barrier.ready
  263. model.connect()
  264. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  265. const current = model.state.getSnapshot()
  266. barrier.release()
  267. await model.rename('settled')
  268. expect(model.state.getSnapshot().render).toBe(current.render)
  269. expect(remote.follow).toHaveBeenCalledTimes(2)
  270. })
  271. it('ignores an already settled stream error after a newer attachment has become writable', async () => {
  272. const barrier = deliveryBarrier()
  273. const { model, remote } = fixture(barrier.prepare)
  274. vi.mocked(remote.follow).mockImplementationOnce(async function* () {
  275. yield* []
  276. throw new Error('retired attachment failed')
  277. })
  278. await mount(model)
  279. await barrier.ready
  280. model.connect()
  281. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  282. barrier.release()
  283. await model.rename('settled')
  284. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', writable: true, error: undefined })
  285. })
  286. it('releases a pending screen render on carrier generation cancellation and accepts the replacement screen', async () => {
  287. const { model, remote, streams } = fixture()
  288. await mount(model)
  289. await expect.poll(() => model.state.getSnapshot().render).toBeDefined()
  290. const revision = model.state.getSnapshot().render!.revision
  291. streams[0]!.restart()
  292. await expect.poll(() => model.state.getSnapshot().render?.revision).toBe(revision + 1)
  293. expect(remote.follow).toHaveBeenCalledTimes(2)
  294. expect(model.state.getSnapshot().writable).toBe(true)
  295. })
  296. it('does not wait for the DOM callback of a screen whose generation was already cancelled at delivery', async () => {
  297. const barrier = deliveryBarrier()
  298. const { model, remote, streams } = fixture(barrier.prepare)
  299. await mount(model)
  300. await barrier.ready
  301. streams[0]!.restart()
  302. barrier.release()
  303. await expect.poll(() => model.state.getSnapshot().render?.revision).toBe(2)
  304. expect(remote.follow).toHaveBeenCalledTimes(2)
  305. expect(model.state.getSnapshot().writable).toBe(true)
  306. })
  307. it.each(['write', 'resize'] as const)('keeps a fresh attachment writable when a pending %s fails after automatic transport recovery', async (operation) => {
  308. const { model, remote, generation } = fixture()
  309. generation.set({ id: 1, host: { home: '/home/fixture' } })
  310. const disconnected = Promise.withResolvers<undefined>()
  311. vi.mocked(remote.follow).mockImplementationOnce(async function* (_sessionId, id, controllerId, signal) {
  312. yield { type: 'snapshot', sequence: 0, screen: 'before disconnect', info: { ...info, id, controllerId } }
  313. await Promise.race([disconnected.promise, untilAborted(signal)])
  314. if (signal?.aborted) return
  315. throw new RemoteStreamCarrierError('connection lost')
  316. })
  317. await connected(model)
  318. const pending = Promise.withResolvers<RemoteResult<void>>()
  319. if (operation === 'write') {
  320. vi.mocked(remote.write).mockReturnValueOnce(pending.promise)
  321. model.write('old input')
  322. } else {
  323. vi.mocked(remote.resize).mockReturnValueOnce(pending.promise)
  324. model.resize(100, 30)
  325. }
  326. await expect.poll(() => vi.mocked(remote[operation]).mock.calls.length).toBe(1)
  327. const oldAttachmentId = vi.mocked(remote.follow).mock.calls[0]![2]
  328. generation.set(undefined)
  329. disconnected.resolve(undefined)
  330. await expect.poll(() => model.state.getSnapshot().phase).toBe('disconnected')
  331. expect(model.state.getSnapshot().writable).toBe(false)
  332. expect(remote.follow).toHaveBeenCalledOnce()
  333. generation.set({ id: 2, host: { home: '/home/fixture' } })
  334. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  335. expect(remote.follow).toHaveBeenCalledTimes(2)
  336. const attachmentId = vi.mocked(remote.follow).mock.calls[1]![2]
  337. expect(attachmentId).not.toBe(oldAttachmentId)
  338. expect(model.state.getSnapshot().info?.controllerId).toBe(attachmentId)
  339. expect(model.state.getSnapshot().render?.frame).toMatchObject({ type: 'snapshot', screen: 'ready' })
  340. model.write('fresh input')
  341. pending.reject(new Error('old attachment was replaced'))
  342. await expect.poll(() => vi.mocked(remote.write).mock.calls.some(call => call[3] === 'fresh input')).toBe(true)
  343. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', writable: true, error: undefined })
  344. expect(vi.mocked(remote.write).mock.calls.at(-1)?.[2]).toBe(attachmentId)
  345. expect(remote.create).toHaveBeenCalledOnce()
  346. expect(remote.close).not.toHaveBeenCalled()
  347. })
  348. it('ignores controls before discovery and reconnects an existing process after remount', async () => {
  349. const { model, remote } = fixture()
  350. model.connect()
  351. model.write('early')
  352. model.resize(100, 30)
  353. model.acknowledge(100)
  354. expect(remote.follow).not.toHaveBeenCalled()
  355. expect(remote.write).not.toHaveBeenCalled()
  356. expect(remote.resize).not.toHaveBeenCalled()
  357. const unmount = await mount(model)
  358. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  359. unmount()
  360. model.connect()
  361. model.write('detached')
  362. model.resize(100, 30)
  363. expect(remote.follow).toHaveBeenCalledOnce()
  364. expect(remote.write).not.toHaveBeenCalled()
  365. expect(remote.resize).not.toHaveBeenCalled()
  366. model.mount()
  367. await expect.poll(() => remote.follow).toHaveBeenCalledTimes(2)
  368. await expect.poll(() => model.state.getSnapshot().writable).toBe(true)
  369. expect(remote.create).toHaveBeenCalledOnce()
  370. })
  371. it('keeps controls inert when process metadata or the attachment is unavailable', async () => {
  372. const { model, remote } = fixture()
  373. await connected(model)
  374. const state = model.state.getSnapshot()
  375. model.state.set({ ...state, info: undefined })
  376. model.write('missing metadata')
  377. model.resize(100, 30)
  378. expect(remote.write).not.toHaveBeenCalled()
  379. expect(remote.resize).not.toHaveBeenCalled()
  380. model.state.set({ ...state, environment: undefined })
  381. model.resize(100, 30)
  382. await expect.poll(() => remote.resize).toHaveBeenCalledOnce()
  383. expect(vi.mocked(remote.resize).mock.calls[0]?.slice(-2)).toEqual([100, 30])
  384. model.write('unknown budget')
  385. expect(model.state.getSnapshot().issue).toBe('inputFull')
  386. expect(remote.write).not.toHaveBeenCalled()
  387. })
  388. it('renames an existing terminal, skips unchanged names, and exposes rename failures', async () => {
  389. const { model, remote } = fixture()
  390. await connected(model)
  391. await model.rename(` ${info.title} `)
  392. expect(remote.rename).not.toHaveBeenCalled()
  393. await model.rename(' Build ')
  394. expect(model.state.getSnapshot()).toMatchObject({ title: 'Build', info: { title: 'Build' } })
  395. vi.mocked(remote.rename).mockRejectedValueOnce('rename connection lost')
  396. await model.rename('Other')
  397. expect(model.state.getSnapshot()).toMatchObject({ phase: 'failed', error: 'rename connection lost', title: 'Build' })
  398. await model.dispose()
  399. await model.rename('Ignored')
  400. expect(remote.rename).toHaveBeenCalledTimes(2)
  401. })
  402. it('classifies a discovery carrier failure as disconnected and supports explicit retry', async () => {
  403. const { model, remote } = fixture()
  404. vi.mocked(remote.environment).mockRejectedValueOnce(new RemoteStreamCarrierError('offline'))
  405. await model.refresh()
  406. expect(model.state.getSnapshot()).toMatchObject({ phase: 'disconnected', error: 'offline' })
  407. expect(remote.create).not.toHaveBeenCalled()
  408. await model.refresh()
  409. await model.start()
  410. expect(model.state.getSnapshot().info?.id).toBe(info.id)
  411. expect(remote.create).toHaveBeenCalledOnce()
  412. })
  413. it('retains state when disposal overtakes successful allocation or a failed discovery', async () => {
  414. for (const operation of ['creation', 'discovery'] as const) {
  415. const { model, remote } = fixture()
  416. const creation = Promise.withResolvers<RemoteResult<WebTerminalInfo>>()
  417. const discovery = Promise.withResolvers<RemoteResult<TerminalEnvironment>>()
  418. if (operation === 'creation') vi.mocked(remote.create).mockReturnValueOnce(creation.promise)
  419. else vi.mocked(remote.environment).mockReturnValueOnce(discovery.promise)
  420. const loading = model.refresh()
  421. const starting = operation === 'creation' ? loading.then(() => model.start()) : undefined
  422. if (operation === 'creation') await expect.poll(() => remote.create).toHaveBeenCalledOnce()
  423. await model.dispose()
  424. const before = model.state.getSnapshot()
  425. if (operation === 'creation') creation.resolve(success(info))
  426. else discovery.reject(new Error('late discovery failure'))
  427. await loading
  428. await starting
  429. expect(model.state.getSnapshot()).toBe(before)
  430. expect(remote.follow).not.toHaveBeenCalled()
  431. }
  432. })
  433. it('ignores a successful rename after the view is disposed', async () => {
  434. const { model, remote } = fixture()
  435. await connected(model)
  436. const rename = Promise.withResolvers<RemoteResult<void>>()
  437. vi.mocked(remote.rename).mockReturnValueOnce(rename.promise)
  438. const pending = model.rename('Late')
  439. await model.dispose()
  440. const before = model.state.getSnapshot()
  441. rename.resolve(success(undefined))
  442. await pending
  443. expect(model.state.getSnapshot()).toBe(before)
  444. })
  445. it('allows retry after failed process cleanup and never reconnects while close is pending', async () => {
  446. const { model, remote } = fixture()
  447. await connected(model)
  448. const closing = Promise.withResolvers<RemoteResult<void>>()
  449. vi.mocked(remote.close).mockReturnValueOnce(closing.promise)
  450. const first = model.close()
  451. expect(model.close()).toBe(first)
  452. model.connect()
  453. expect(remote.follow).toHaveBeenCalledOnce()
  454. closing.resolve(failure('close refused'))
  455. await expect(first).rejects.toThrow('close refused')
  456. expect(model.state.getSnapshot()).toMatchObject({ phase: 'failed', error: 'close refused' })
  457. await model.close()
  458. expect(model.state.getSnapshot()).toMatchObject({ phase: 'closed', writable: false })
  459. expect(remote.close).toHaveBeenCalledTimes(2)
  460. })
  461. it('does not publish a close result or reconnect after disposal', async () => {
  462. const { model, remote } = fixture()
  463. await connected(model)
  464. const closing = Promise.withResolvers<RemoteResult<void>>()
  465. vi.mocked(remote.close).mockReturnValueOnce(closing.promise)
  466. const pending = model.close()
  467. await model.dispose()
  468. const before = model.state.getSnapshot()
  469. closing.resolve(success(undefined))
  470. await pending
  471. model.mount()
  472. model.connect()
  473. expect(model.state.getSnapshot()).toBe(before)
  474. expect(remote.follow).toHaveBeenCalledOnce()
  475. })
  476. it.each(['write', 'resize'] as const)('keeps the output connection when %s loses control before its state frame arrives', async (operation) => {
  477. const { model, remote } = fixture()
  478. const transfer = Promise.withResolvers<undefined>()
  479. const response = Promise.withResolvers<RemoteResult<void>>()
  480. cleanups.push(() => { transfer.resolve(undefined); response.resolve(success(undefined)) })
  481. vi.mocked(remote.follow).mockImplementation(async function* (_session, id, controllerId, signal) {
  482. yield { type: 'snapshot', sequence: 0, screen: 'retained screen', info: { ...info, id, controllerId } }
  483. await transfer.promise
  484. yield { type: 'state', info: { ...info, id } }
  485. await untilAborted(signal)
  486. })
  487. await connected(model)
  488. vi.mocked(remote[operation]).mockReturnValueOnce(response.promise)
  489. if (operation === 'write') model.write('before transfer')
  490. else model.resize(100, 30)
  491. model.write('queued before transfer')
  492. await expect.poll(() => vi.mocked(remote[operation]).mock.calls.length).toBe(1)
  493. response.resolve({ ok: false, error: new RemoteError('terminal/control-unavailable', 'Another window owns input', { reason: 'read-only' }) })
  494. await expect.poll(() => model.state.getSnapshot().writable).toBe(false)
  495. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', error: undefined, issue: undefined, render: { frame: { screen: 'retained screen' } } })
  496. transfer.resolve(undefined)
  497. await expect.poll(() => model.state.getSnapshot().info?.controllerId).toBeUndefined()
  498. expect(remote.follow).toHaveBeenCalledOnce()
  499. expect(vi.mocked(remote.write).mock.calls.some(call => call[3] === 'queued before transfer')).toBe(false)
  500. })
  501. it('keeps an exited screen when a pending input is refused after the exit state arrives', async () => {
  502. const { model, remote } = fixture()
  503. const exit = Promise.withResolvers<undefined>()
  504. const response = Promise.withResolvers<RemoteResult<void>>()
  505. cleanups.push(() => { exit.resolve(undefined); response.resolve(success(undefined)) })
  506. vi.mocked(remote.follow).mockImplementation(async function* (_session, id, controllerId, signal) {
  507. yield { type: 'snapshot', sequence: 0, screen: 'final screen', info: { ...info, id, controllerId } }
  508. await exit.promise
  509. yield { type: 'state', info: { ...info, id, state: 'exited', exitCode: 0 } }
  510. await untilAborted(signal)
  511. })
  512. await connected(model)
  513. vi.mocked(remote.write).mockReturnValueOnce(response.promise)
  514. model.write('exit race')
  515. await expect.poll(() => vi.mocked(remote.write).mock.calls.length).toBe(1)
  516. exit.resolve(undefined)
  517. await expect.poll(() => model.state.getSnapshot().info?.state).toBe('exited')
  518. response.resolve({ ok: false, error: new RemoteError('terminal/control-unavailable', 'Terminal is not running', { reason: 'not-running' }) })
  519. await setImmediate()
  520. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', writable: false, error: undefined, issue: undefined, render: { frame: { screen: 'final screen' } } })
  521. })
  522. it('exposes a localized quota error and clears it after a successful retry', async () => {
  523. const { model, remote } = fixture()
  524. vi.mocked(remote.create).mockResolvedValueOnce({ ok: false, error: new RemoteError('terminal/limit-reached', 'Session terminal limit reached', { limit: 8 }) })
  525. await model.refresh()
  526. await model.start()
  527. expect(model.state.getSnapshot()).toMatchObject({ phase: 'failed', issue: 'terminalLimit' })
  528. await connected(model)
  529. expect(model.state.getSnapshot()).toMatchObject({ phase: 'connected', issue: undefined, error: undefined })
  530. })