session.spec.ts 66 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606
  1. import { afterEach, describe, expect, it, vi } from 'vitest'
  2. import { PassThrough } from 'node:stream'
  3. import { LocalPtySession } from '@deepseek-ai/dsh-terminal-bash/src/session.ts'
  4. import type { ResolvedConfig } from '@deepseek-ai/dsh-terminal-bash/src/config.ts'
  5. import type { TerminalSendOperation, TerminalSessionStatus, TerminalSignal } from '@deepseek-ai/dsh-terminal'
  6. import type {
  7. SubprocessOutcome,
  8. SubprocessTerminalHandle,
  9. SubprocessTerminalSignal,
  10. } from '@deepseek-ai/dsh-subprocess'
  11. import { TerminalError } from '@deepseek-ai/dsh-terminal'
  12. import type {
  13. ProcessIdentity,
  14. ProcessInspector,
  15. } from '@deepseek-ai/dsh-subprocess-local/src/process-inspector.ts'
  16. class FakeInspector implements ProcessInspector {
  17. pgid: number | undefined = 456
  18. waiting = false
  19. members: ProcessIdentity[] = []
  20. alive = new Set<number>()
  21. groups: Array<[number, TerminalSignal]> = []
  22. processes: Array<[number, 'SIGTERM' | 'SIGKILL']> = []
  23. throwGroup = false
  24. throwProcess = false
  25. removeOnSignal = true
  26. foregroundPgid() { return this.pgid }
  27. isStdinWaiting() { return this.waiting }
  28. snapshot() {
  29. return {
  30. tree: () => this.members,
  31. session: () => [],
  32. alive: (identity: ProcessIdentity) => this.alive.has(identity.pid),
  33. }
  34. }
  35. isAlive(identity: ProcessIdentity) { return this.alive.has(identity.pid) }
  36. signalGroup(pgid: number, signal: TerminalSignal) {
  37. if (this.throwGroup) throw new Error('group failed')
  38. this.groups.push([pgid, signal])
  39. }
  40. signalProcess(identity: ProcessIdentity, signal: 'SIGTERM' | 'SIGKILL') {
  41. if (this.throwProcess) throw new Error('process raced')
  42. this.processes.push([identity.pid, signal])
  43. if (this.removeOnSignal) this.alive.delete(identity.pid)
  44. }
  45. }
  46. class FakeTerminal implements SubprocessTerminalHandle {
  47. pid = 123
  48. readonly output = new PassThrough()
  49. readonly writes: string[] = []
  50. readonly kills: string[] = []
  51. readonly outcome = Promise.withResolvers<SubprocessOutcome>()
  52. readonly done = this.outcome.promise
  53. throwWrite = false
  54. throwKill = false
  55. autoExitOnKill = true
  56. terminateError: Error | undefined
  57. private cleanup: Promise<void> | undefined
  58. constructor(public inspector = new FakeInspector()) {}
  59. emitData(data: string): void {
  60. this.output.write(Buffer.from(data, 'utf8'))
  61. }
  62. emitBytes(data: Uint8Array): void {
  63. this.output.write(data)
  64. }
  65. emitError(error: Error): void {
  66. this.output.emit('error', error)
  67. }
  68. emitFailure(error: unknown): void {
  69. this.output.end()
  70. this.outcome.reject(error)
  71. }
  72. emitExit(exitCode = 0, signal?: number): void {
  73. this.output.end()
  74. this.outcome.resolve({
  75. exitCode: signal === undefined || signal === 0 ? exitCode : null,
  76. signal: signal === 9 ? 'SIGKILL' : signal === 15 ? 'SIGTERM' : null,
  77. })
  78. }
  79. async write(data: string): Promise<void> {
  80. if (this.throwWrite) throw new Error('write failed')
  81. this.writes.push(data)
  82. }
  83. async inspectForeground() {
  84. const processGroupId = this.inspector.foregroundPgid()
  85. return processGroupId === undefined
  86. ? undefined
  87. : { processGroupId, inputWaiting: this.inspector.isStdinWaiting() }
  88. }
  89. async signalForeground(signal: SubprocessTerminalSignal): Promise<number> {
  90. const foreground = await this.inspectForeground()
  91. if (foreground === undefined) throw new Error(`cannot resolve foreground process group for terminal ${this.pid}`)
  92. if (signal === 'SIGKILL' && foreground.processGroupId === this.pid) {
  93. throw new Error('refusing to SIGKILL the terminal shell; terminate the terminal session instead')
  94. }
  95. this.inspector.signalGroup(foreground.processGroupId, signal)
  96. return foreground.processGroupId
  97. }
  98. terminate(): Promise<void> {
  99. if (this.cleanup !== undefined) return this.cleanup
  100. const cleanup = this.terminateOnce()
  101. this.cleanup = cleanup
  102. void cleanup.catch(() => { this.cleanup = undefined })
  103. return cleanup
  104. }
  105. private async terminateOnce(): Promise<void> {
  106. if (this.terminateError !== undefined) throw this.terminateError
  107. if (this.throwKill) throw new Error('kill failed')
  108. this.kills.push('SIGTERM')
  109. if (this.autoExitOnKill) this.emitExit(0, 15)
  110. }
  111. }
  112. function makeSession(
  113. terminal: FakeTerminal,
  114. inspector: FakeInspector,
  115. resolved: ResolvedConfig,
  116. ): LocalPtySession {
  117. terminal.inspector = inspector
  118. return new LocalPtySession(terminal, resolved)
  119. }
  120. function config(overrides: Partial<ResolvedConfig> = {}): ResolvedConfig {
  121. return {
  122. backendType: 'shell', shellDialect: 'bash', shellPath: '/bin/bash', shellArgs: [], rows: 24, cols: 80,
  123. scrollbackLines: 10, scrollbackMaxBytes: 128, maxReadBytes: 64,
  124. pollIntervalMs: 10, exactProbeAfterMs: 20, idleSilenceMs: 50, handoffGraceMs: 10, timeoutMs: 100,
  125. disposeGraceMs: 20,
  126. ...overrides,
  127. }
  128. }
  129. afterEach(() => { vi.useRealTimers() })
  130. async function initialize(session: LocalPtySession, terminal: FakeTerminal): Promise<void> {
  131. const pending = session.initialize()
  132. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  133. await vi.advanceTimersByTimeAsync(10)
  134. await pending
  135. }
  136. describe('LocalPtySession readiness and output', () => {
  137. it('answers split cursor-position queries before publishing prompt readiness', async () => {
  138. vi.useFakeTimers()
  139. const terminal = new FakeTerminal()
  140. const inspector = new FakeInspector()
  141. const session = makeSession(terminal, inspector, config())
  142. const responseGate = Promise.withResolvers<undefined>()
  143. terminal.write = async (data) => {
  144. terminal.writes.push(data)
  145. await responseGate.promise
  146. }
  147. let initialized = false
  148. const pending = session.initialize().then(() => { initialized = true })
  149. terminal.emitData('\x1b]133;D;0\x07dsh> \x1b[')
  150. terminal.emitData('6n')
  151. await vi.advanceTimersByTimeAsync(20)
  152. expect(terminal.writes).toContain('\x1b[1;6R')
  153. expect(initialized).toBe(false)
  154. responseGate.resolve(undefined)
  155. await vi.advanceTimersByTimeAsync(10)
  156. await pending
  157. expect(session.motd).toBe('dsh> ')
  158. })
  159. it('drains terminal replies before caller input and re-inspects after concurrent output', async () => {
  160. vi.useFakeTimers()
  161. const terminal = new FakeTerminal()
  162. const inspector = new FakeInspector()
  163. const session = makeSession(terminal, inspector, config())
  164. await initialize(session, terminal)
  165. const firstInspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  166. const secondInspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  167. let inspections = 0
  168. terminal.inspectForeground = async () => {
  169. inspections += 1
  170. if (inspections === 1) return await firstInspection.promise
  171. if (inspections === 2) return await secondInspection.promise
  172. return { processGroupId: 456, inputWaiting: false }
  173. }
  174. const responseGate = Promise.withResolvers<undefined>()
  175. terminal.write = async (data) => {
  176. terminal.writes.push(data)
  177. if (data === '\x1b[1;6R') await responseGate.promise
  178. }
  179. const operation = session.startSend({ text: 'caller input', submit: true })
  180. await Promise.resolve()
  181. terminal.emitData('\x1b[6n')
  182. await vi.advanceTimersByTimeAsync(0)
  183. firstInspection.resolve({ processGroupId: 456, inputWaiting: true })
  184. await Promise.resolve()
  185. expect(terminal.writes).toEqual(['\x1b[1;6R'])
  186. responseGate.resolve(undefined)
  187. await vi.advanceTimersByTimeAsync(0)
  188. expect(inspections).toBe(2)
  189. terminal.emitData('\x1b[6n')
  190. await vi.advanceTimersByTimeAsync(0)
  191. secondInspection.resolve({ processGroupId: 456, inputWaiting: true })
  192. await vi.advanceTimersByTimeAsync(0)
  193. expect(inspections).toBe(3)
  194. expect(terminal.writes).toEqual(['\x1b[1;6R', '\x1b[1;6R', 'caller input\r'])
  195. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  196. await vi.advanceTimersByTimeAsync(10)
  197. expect((await operation.done).waitReason).toBe('stdin_read')
  198. })
  199. it('drains a terminal reply that is pending when caller input starts', async () => {
  200. vi.useFakeTimers()
  201. const terminal = new FakeTerminal()
  202. const inspector = new FakeInspector()
  203. const session = makeSession(terminal, inspector, config())
  204. await initialize(session, terminal)
  205. const responseGate = Promise.withResolvers<undefined>()
  206. terminal.write = async (data) => {
  207. terminal.writes.push(data)
  208. if (data === '\x1b[1;6R') await responseGate.promise
  209. }
  210. terminal.emitData('\x1b[6n')
  211. await vi.advanceTimersByTimeAsync(0)
  212. const operation = session.startSend({ text: 'caller input', submit: true })
  213. await Promise.resolve()
  214. expect(terminal.writes).toEqual(['\x1b[1;6R'])
  215. responseGate.resolve(undefined)
  216. await vi.advanceTimersByTimeAsync(0)
  217. expect(terminal.writes).toEqual(['\x1b[1;6R', 'caller input\r'])
  218. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  219. await vi.advanceTimersByTimeAsync(10)
  220. expect((await operation.done).waitReason).toBe('stdin_read')
  221. })
  222. it('resamples readiness foreground state after protocol activity during inspection', async () => {
  223. vi.useFakeTimers()
  224. const terminal = new FakeTerminal()
  225. const inspector = new FakeInspector()
  226. const session = makeSession(terminal, inspector, config())
  227. await initialize(session, terminal)
  228. const operation = session.startSend({ text: '', submit: false })
  229. await Promise.resolve()
  230. await Promise.resolve()
  231. const internal = session as unknown as {
  232. stopReadinessPolling(): void
  233. pollReadiness(operation: TerminalSendOperation): Promise<void>
  234. settleActive(reason: 'timeout'): void
  235. }
  236. internal.stopReadinessPolling()
  237. await vi.advanceTimersByTimeAsync(20)
  238. const firstInspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  239. let inspections = 0
  240. terminal.inspectForeground = async () => {
  241. inspections += 1
  242. return inspections === 1
  243. ? await firstInspection.promise
  244. : { processGroupId: 456, inputWaiting: false }
  245. }
  246. const polling = internal.pollReadiness(operation)
  247. await Promise.resolve()
  248. terminal.emitData('\x1b[6n')
  249. await vi.advanceTimersByTimeAsync(0)
  250. firstInspection.resolve({ processGroupId: 456, inputWaiting: true })
  251. await polling
  252. expect(inspections).toBe(2)
  253. expect((operation as unknown as { settled: boolean }).settled).toBe(false)
  254. internal.settleActive('timeout')
  255. expect((await operation.done).waitReason).toBe('timeout')
  256. })
  257. it('retains send ownership while a failed inspection drains a terminal reply', async () => {
  258. vi.useFakeTimers()
  259. const terminal = new FakeTerminal()
  260. const inspector = new FakeInspector()
  261. const session = makeSession(terminal, inspector, config())
  262. await initialize(session, terminal)
  263. const operation = session.startSend({ text: '', submit: false })
  264. await Promise.resolve()
  265. await Promise.resolve()
  266. const internal = session as unknown as {
  267. stopReadinessPolling(): void
  268. pollReadiness(operation: TerminalSendOperation): Promise<void>
  269. }
  270. internal.stopReadinessPolling()
  271. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  272. terminal.inspectForeground = async () => await inspection.promise
  273. const responseGate = Promise.withResolvers<undefined>()
  274. terminal.write = async (data) => {
  275. terminal.writes.push(data)
  276. await responseGate.promise
  277. }
  278. const polling = internal.pollReadiness(operation)
  279. await Promise.resolve()
  280. terminal.emitData('\x1b[6n')
  281. await vi.advanceTimersByTimeAsync(0)
  282. inspection.reject(new Error('inspection failed with reply pending'))
  283. await Promise.resolve()
  284. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow('active send')
  285. responseGate.resolve(undefined)
  286. await polling
  287. await expect(operation.done).rejects.toThrow('inspection failed with reply pending')
  288. })
  289. it('retains pre-write ownership when inspection fails with a terminal reply pending', async () => {
  290. vi.useFakeTimers()
  291. const terminal = new FakeTerminal()
  292. const inspector = new FakeInspector()
  293. const session = makeSession(terminal, inspector, config())
  294. await initialize(session, terminal)
  295. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  296. terminal.inspectForeground = async () => await inspection.promise
  297. const responseGate = Promise.withResolvers<undefined>()
  298. terminal.write = async (data) => {
  299. terminal.writes.push(data)
  300. await responseGate.promise
  301. }
  302. const operation = session.startSend({ text: 'must not execute', submit: true })
  303. await Promise.resolve()
  304. terminal.emitData('\x1b[6n')
  305. await vi.advanceTimersByTimeAsync(0)
  306. inspection.reject(new Error('pre-write inspection failed with reply pending'))
  307. await Promise.resolve()
  308. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow('active send')
  309. responseGate.resolve(undefined)
  310. await expect(operation.done).rejects.toThrow('pre-write inspection failed with reply pending')
  311. expect(terminal.writes).toEqual(['\x1b[1;6R'])
  312. })
  313. it('retains a timed-out send until its terminal-protocol response settles', async () => {
  314. vi.useFakeTimers()
  315. const terminal = new FakeTerminal()
  316. const inspector = new FakeInspector()
  317. const session = makeSession(terminal, inspector, config())
  318. await initialize(session, terminal)
  319. const responseGate = Promise.withResolvers<undefined>()
  320. terminal.write = async (data) => {
  321. terminal.writes.push(data)
  322. await responseGate.promise
  323. }
  324. const operation = session.startSend({ text: '', submit: false })
  325. terminal.emitData('\x1b[6n')
  326. await vi.advanceTimersByTimeAsync(100)
  327. expect((await operation.done).waitReason).toBe('timeout')
  328. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow('active send')
  329. responseGate.resolve(undefined)
  330. await vi.advanceTimersByTimeAsync(0)
  331. const successor = session.startSend({ text: '', submit: false })
  332. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  333. await vi.advanceTimersByTimeAsync(10)
  334. expect((await successor.done).waitReason).toBe('stdin_read')
  335. })
  336. it('contains terminal emulator and protocol-response failures', async () => {
  337. const responseTerminal = new FakeTerminal()
  338. responseTerminal.throwWrite = true
  339. const responseSession = new LocalPtySession(responseTerminal, config())
  340. const responseOperation = responseSession.startSend({ text: '', submit: false })
  341. responseTerminal.emitData('\x1b[6n')
  342. await expect(responseOperation.done).rejects.toThrow('write failed')
  343. expect(responseSession.status()).toEqual({ kind: 'exited', exitCode: null, signal: null })
  344. const emulatorTerminal = new FakeTerminal()
  345. const emulatorSession = new LocalPtySession(emulatorTerminal, config())
  346. const emulatorOperation = emulatorSession.startSend({ text: '', submit: false })
  347. const emulator = (emulatorSession as unknown as {
  348. emulator: { write(data: string, callback?: () => void): void }
  349. }).emulator
  350. emulator.write = () => { throw new Error('emulator failed') }
  351. emulatorTerminal.emitData('output')
  352. await expect(emulatorOperation.done).rejects.toThrow('emulator failed')
  353. expect(emulatorSession.status()).toEqual({ kind: 'exited', exitCode: null, signal: null })
  354. })
  355. it('ignores terminal-protocol failures after closing starts and drains changing queues', async () => {
  356. const terminal = new FakeTerminal()
  357. const session = new LocalPtySession(terminal, config())
  358. const internal = session as unknown as {
  359. closing: boolean
  360. emulator: { write(data: string, callback?: () => void): void }
  361. emulatorWrites: Promise<void>
  362. responseWrites: Promise<void>
  363. drainTerminalProtocol(): Promise<void>
  364. closeEmulator(): void
  365. }
  366. internal.closing = true
  367. terminal.throwWrite = true
  368. terminal.emitData('\x1b[6n')
  369. await internal.emulatorWrites
  370. await internal.responseWrites
  371. expect(session.status()).toEqual({ kind: 'running' })
  372. internal.emulator.write = () => { throw new Error('late emulator failure') }
  373. terminal.emitData('late output')
  374. await internal.emulatorWrites
  375. expect(session.status()).toEqual({ kind: 'running' })
  376. const first = Promise.withResolvers<undefined>()
  377. internal.emulatorWrites = first.promise
  378. const draining = internal.drainTerminalProtocol()
  379. internal.emulatorWrites = Promise.resolve()
  380. first.resolve(undefined)
  381. await draining
  382. internal.closeEmulator()
  383. internal.closeEmulator()
  384. terminal.emitData('after emulator close')
  385. expect(session.status()).toEqual({ kind: 'running' })
  386. })
  387. it('coalesces terminal output that arrives while an emulator parse is pending', async () => {
  388. const terminal = new FakeTerminal()
  389. const session = new LocalPtySession(terminal, config())
  390. const writes: Array<{ data: string; done: () => void }> = []
  391. const internal = session as unknown as {
  392. emulator: { write(data: string, callback?: () => void): void }
  393. emulatorWrites: Promise<void>
  394. closeEmulator(): void
  395. }
  396. internal.emulator.write = (data, callback) => {
  397. writes.push({ data, done: callback ?? (() => {}) })
  398. }
  399. terminal.emitData('first')
  400. terminal.emitData('second')
  401. terminal.emitData('third')
  402. await Promise.resolve()
  403. expect(writes.map(write => write.data)).toEqual(['first'])
  404. writes[0]!.done()
  405. await Promise.resolve()
  406. expect(writes.map(write => write.data)).toEqual(['first', 'secondthird'])
  407. writes[1]!.done()
  408. await internal.emulatorWrites
  409. internal.closeEmulator()
  410. })
  411. it('lets queued terminal output run before the first post-write readiness poll', async () => {
  412. vi.useFakeTimers()
  413. const terminal = new FakeTerminal()
  414. const inspector = new FakeInspector()
  415. const session = makeSession(terminal, inspector, config())
  416. await initialize(session, terminal)
  417. const inspect = terminal.inspectForeground.bind(terminal)
  418. let inspections = 0
  419. terminal.inspectForeground = async () => {
  420. inspections += 1
  421. return await inspect()
  422. }
  423. const operation = session.startSend({ text: 'true', submit: true })
  424. await Promise.resolve()
  425. await Promise.resolve()
  426. expect(inspections).toBe(1)
  427. await vi.advanceTimersByTimeAsync(0)
  428. expect(inspections).toBe(1)
  429. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  430. await vi.advanceTimersByTimeAsync(10)
  431. expect((await operation.done).waitReason).toBe('stdin_read')
  432. })
  433. it('discards prompt readiness observed during asynchronous pre-write inspection', async () => {
  434. vi.useFakeTimers()
  435. const terminal = new FakeTerminal()
  436. const session = new LocalPtySession(terminal, config())
  437. await initialize(session, terminal)
  438. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  439. terminal.inspectForeground = async () => await inspection.promise
  440. const operation = session.startSend({ text: 'long-running-command', submit: true })
  441. let settled = false
  442. void operation.done.then(() => { settled = true })
  443. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  444. inspection.resolve({ processGroupId: 456, inputWaiting: true })
  445. await vi.advanceTimersByTimeAsync(20)
  446. expect(terminal.writes).toEqual(['long-running-command\r'])
  447. expect(settled).toBe(false)
  448. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  449. await vi.advanceTimersByTimeAsync(10)
  450. expect((await operation.done).waitReason).toBe('stdin_read')
  451. })
  452. it('captures prompt MOTD, writes submit explicitly, and settles exact stdin waits', async () => {
  453. vi.useFakeTimers()
  454. const terminal = new FakeTerminal()
  455. const inspector = new FakeInspector()
  456. const session = makeSession(terminal, inspector, config())
  457. await initialize(session, terminal)
  458. expect(session.motd).toBe('dsh> ')
  459. inspector.waiting = true
  460. const operation = session.startSend({ text: 'python3', submit: true })
  461. await Promise.resolve()
  462. await Promise.resolve()
  463. expect(terminal.writes).toEqual(['python3\r'])
  464. inspector.pgid = 789
  465. terminal.emitData('Python\r\n>>> ')
  466. await vi.advanceTimersByTimeAsync(20)
  467. expect(await operation.done).toMatchObject({ waitReason: 'stdin_read', viewport: 'Python\n>>> ', sessionStatus: { kind: 'running' } })
  468. expect(operation.cancel()).toBe(false)
  469. })
  470. it('does not reuse a pre-write stdin wait as post-write readiness', async () => {
  471. vi.useFakeTimers()
  472. const terminal = new FakeTerminal()
  473. const inspector = new FakeInspector()
  474. const session = makeSession(terminal, inspector, config())
  475. await initialize(session, terminal)
  476. inspector.waiting = true
  477. const operation = session.startSend({ text: 'echo ready', submit: true })
  478. let settled = false
  479. void operation.done.then(() => { settled = true })
  480. await vi.advanceTimersByTimeAsync(20)
  481. expect(settled).toBe(false)
  482. inspector.waiting = false
  483. await vi.advanceTimersByTimeAsync(10)
  484. expect(settled).toBe(false)
  485. inspector.waiting = true
  486. await vi.advanceTimersByTimeAsync(10)
  487. expect((await operation.done).waitReason).toBe('stdin_read')
  488. })
  489. it('tracks a pre-write wait exit before exact probing begins', async () => {
  490. vi.useFakeTimers()
  491. const terminal = new FakeTerminal()
  492. const inspector = new FakeInspector()
  493. const session = makeSession(terminal, inspector, config({
  494. exactProbeAfterMs: 50,
  495. idleSilenceMs: 100,
  496. timeoutMs: 200,
  497. }))
  498. await initialize(session, terminal)
  499. inspector.waiting = true
  500. const operation = session.startSend({ text: 'fast command', submit: true })
  501. let settled = false
  502. void operation.done.then(() => { settled = true })
  503. inspector.waiting = false
  504. await vi.advanceTimersByTimeAsync(10)
  505. inspector.waiting = true
  506. await vi.advanceTimersByTimeAsync(30)
  507. expect(settled).toBe(false)
  508. await vi.advanceTimersByTimeAsync(10)
  509. expect(settled).toBe(true)
  510. expect((await operation.done).waitReason).toBe('stdin_read')
  511. })
  512. it('distinguishes inferred idle, timeout, exit signal, and operation reads', async () => {
  513. vi.useFakeTimers()
  514. const terminal = new FakeTerminal()
  515. const inspector = new FakeInspector()
  516. const session = makeSession(terminal, inspector, config())
  517. await initialize(session, terminal)
  518. inspector.pgid = undefined
  519. const inferred = session.startSend({ text: 'sleep', submit: false })
  520. terminal.emitData('working')
  521. expect(inferred.readOutput()).toEqual({ delta: 'working', truncated: false })
  522. await vi.advanceTimersByTimeAsync(60)
  523. expect((await inferred.done).waitReason).toBe('inferred_idle')
  524. const timeout = session.startSend({ text: 'blocked', submit: false })
  525. await vi.advanceTimersByTimeAsync(40)
  526. terminal.emitData('.')
  527. await vi.advanceTimersByTimeAsync(40)
  528. terminal.emitData('.')
  529. await vi.advanceTimersByTimeAsync(30)
  530. expect((await timeout.done).waitReason).toBe('timeout')
  531. const exiting = session.startSend({ text: 'exit', submit: true })
  532. terminal.emitExit(7, 9)
  533. expect(await exiting.done).toMatchObject({ waitReason: 'session_exit', sessionStatus: { kind: 'exited', exitCode: null, signal: 'SIGKILL' } })
  534. expect(() => session.startSend({ text: '', submit: false })).toThrow('has exited')
  535. })
  536. it('cancels with foreground-group SIGINT, observes AbortSignal, and contains write failures', async () => {
  537. vi.useFakeTimers()
  538. const terminal = new FakeTerminal()
  539. const inspector = new FakeInspector()
  540. const session = makeSession(terminal, inspector, config())
  541. await initialize(session, terminal)
  542. const controller = new AbortController()
  543. const operation = session.startSend({ text: 'sleep', submit: true, signal: controller.signal })
  544. expect(() => session.startSend({ text: 'again', submit: true })).toThrow('active send')
  545. controller.abort()
  546. await Promise.resolve()
  547. await Promise.resolve()
  548. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  549. expect(terminal.writes).not.toContain('\x03')
  550. terminal.emitData('\x1b]133;D;130\x07dsh> ')
  551. await vi.advanceTimersByTimeAsync(10)
  552. await operation.done
  553. const aborted = new AbortController()
  554. aborted.abort()
  555. expect(() => session.startSend({ text: '', submit: false, signal: aborted.signal })).toThrow('aborted before write')
  556. terminal.throwWrite = true
  557. const failed = session.startSend({ text: 'x', submit: false })
  558. await expect(failed.done).rejects.toThrow('write failed')
  559. const failedInternal = failed as unknown as { append(text: string): void; fail(error: unknown): void }
  560. failedInternal.append('ignored')
  561. failedInternal.fail(new Error('ignored'))
  562. })
  563. it('does not write a send canceled during asynchronous foreground inspection', async () => {
  564. vi.useFakeTimers()
  565. const terminal = new FakeTerminal()
  566. const inspector = new FakeInspector()
  567. const session = makeSession(terminal, inspector, config())
  568. await initialize(session, terminal)
  569. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  570. terminal.inspectForeground = async () => await inspection.promise
  571. const controller = new AbortController()
  572. const operation = session.startSend({ text: 'must not execute', submit: true, signal: controller.signal })
  573. controller.abort()
  574. inspection.resolve({ processGroupId: 456, inputWaiting: false })
  575. await Promise.resolve()
  576. await Promise.resolve()
  577. expect(terminal.writes).toEqual([])
  578. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  579. terminal.emitData('\x1b]133;D;130\x07dsh> ')
  580. await vi.advanceTimersByTimeAsync(10)
  581. await operation.done
  582. })
  583. it('retains a canceled send when the pre-write inspection rejects while its signal is in flight', async () => {
  584. vi.useFakeTimers()
  585. const terminal = new FakeTerminal()
  586. const inspector = new FakeInspector()
  587. const session = makeSession(terminal, inspector, config())
  588. await initialize(session, terminal)
  589. const failed = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  590. terminal.inspectForeground = async () => await failed.promise
  591. const uncanceled = session.startSend({ text: 'plain failure', submit: true })
  592. failed.reject(new Error('inspect failed before write'))
  593. await expect(uncanceled.done).rejects.toThrow('inspect failed before write')
  594. terminal.inspectForeground = FakeTerminal.prototype.inspectForeground.bind(terminal)
  595. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  596. terminal.inspectForeground = async () => await inspection.promise
  597. const signalGate = Promise.withResolvers<undefined>()
  598. terminal.signalForeground = async (signal) => {
  599. await signalGate.promise
  600. inspector.signalGroup(456, signal)
  601. return 456
  602. }
  603. const controller = new AbortController()
  604. const operation = session.startSend({ text: 'must stay owned', submit: true, signal: controller.signal })
  605. controller.abort()
  606. inspection.reject(new Error('transient inspection failure'))
  607. await Promise.resolve()
  608. await Promise.resolve()
  609. // The slot stays reserved while the cancellation's foreground signal is in flight.
  610. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow('active send')
  611. terminal.inspectForeground = async () => ({ processGroupId: 456, inputWaiting: false })
  612. signalGate.resolve(undefined)
  613. await vi.advanceTimersByTimeAsync(0)
  614. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  615. terminal.emitData('\x1b]133;D;130\x07dsh> ')
  616. await vi.advanceTimersByTimeAsync(10)
  617. await operation.done
  618. })
  619. it('retains a canceled send until asynchronous foreground signalling settles', async () => {
  620. vi.useFakeTimers()
  621. const terminal = new FakeTerminal()
  622. const inspector = new FakeInspector()
  623. const session = makeSession(terminal, inspector, config())
  624. await initialize(session, terminal)
  625. const signalGate = Promise.withResolvers<undefined>()
  626. terminal.signalForeground = async (signal) => {
  627. await signalGate.promise
  628. const foreground = await terminal.inspectForeground()
  629. if (foreground === undefined) throw new Error('cannot resolve foreground')
  630. inspector.signalGroup(foreground.processGroupId, signal)
  631. return foreground.processGroupId
  632. }
  633. const operation = session.startSend({ text: 'first', submit: true })
  634. await Promise.resolve()
  635. await Promise.resolve()
  636. expect(operation.cancel()).toBe(true)
  637. terminal.emitData('\x1b]133;D;130\x07dsh> ')
  638. await vi.advanceTimersByTimeAsync(100)
  639. expect((await operation.done).waitReason).toBe('timeout')
  640. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow('active send')
  641. signalGate.resolve(undefined)
  642. await vi.advanceTimersByTimeAsync(0)
  643. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  644. expect(inspector.groups).not.toContainEqual([789, 'SIGINT'])
  645. })
  646. it('does not let an in-flight readiness inspection release a canceled send before signalling settles', async () => {
  647. vi.useFakeTimers()
  648. const terminal = new FakeTerminal()
  649. const inspector = new FakeInspector()
  650. const session = makeSession(terminal, inspector, config())
  651. await initialize(session, terminal)
  652. const readiness = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  653. let inspections = 0
  654. terminal.inspectForeground = async () => {
  655. inspections += 1
  656. if (inspections === 1) return { processGroupId: 456, inputWaiting: false }
  657. if (inspections === 2) return await readiness.promise
  658. return { processGroupId: 456, inputWaiting: true }
  659. }
  660. const signalling = Promise.withResolvers<undefined>()
  661. const signalled = Promise.withResolvers<number>()
  662. terminal.signalForeground = async (signal) => {
  663. await signalling.promise
  664. const foreground = await terminal.inspectForeground()
  665. if (foreground === undefined) throw new Error('cannot resolve foreground')
  666. inspector.signalGroup(foreground.processGroupId, signal)
  667. signalled.resolve(foreground.processGroupId)
  668. return foreground.processGroupId
  669. }
  670. const operation = session.startSend({ text: 'first', submit: true })
  671. await Promise.resolve()
  672. await Promise.resolve()
  673. await vi.advanceTimersByTimeAsync(10)
  674. expect(inspections).toBe(2)
  675. await vi.advanceTimersByTimeAsync(50)
  676. expect(operation.cancel()).toBe(true)
  677. let settled = false
  678. void operation.done.then(() => { settled = true })
  679. readiness.resolve({ processGroupId: 456, inputWaiting: true })
  680. await Promise.resolve()
  681. await Promise.resolve()
  682. expect(settled).toBe(false)
  683. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow(TerminalError)
  684. signalling.resolve(undefined)
  685. expect(await signalled.promise).toBe(456)
  686. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  687. await session.close('test complete')
  688. expect((await operation.done).waitReason).toBe('session_exit')
  689. })
  690. it('does not let an in-flight readiness failure release a canceled send before signalling settles', async () => {
  691. vi.useFakeTimers()
  692. const terminal = new FakeTerminal()
  693. const inspector = new FakeInspector()
  694. const session = makeSession(terminal, inspector, config())
  695. await initialize(session, terminal)
  696. const readiness = Promise.withResolvers<never>()
  697. let inspections = 0
  698. terminal.inspectForeground = async () => {
  699. inspections += 1
  700. if (inspections === 1) return { processGroupId: 456, inputWaiting: false }
  701. if (inspections === 2) return await readiness.promise
  702. return { processGroupId: 456, inputWaiting: true }
  703. }
  704. const signalling = Promise.withResolvers<undefined>()
  705. const signalled = Promise.withResolvers<number>()
  706. terminal.signalForeground = async (signal) => {
  707. await signalling.promise
  708. const foreground = await terminal.inspectForeground()
  709. if (foreground === undefined) throw new Error('cannot resolve foreground')
  710. inspector.signalGroup(foreground.processGroupId, signal)
  711. signalled.resolve(foreground.processGroupId)
  712. return foreground.processGroupId
  713. }
  714. const operation = session.startSend({ text: 'first', submit: true })
  715. await Promise.resolve()
  716. await Promise.resolve()
  717. await vi.advanceTimersByTimeAsync(10)
  718. expect(inspections).toBe(2)
  719. expect(operation.cancel()).toBe(true)
  720. let settled = false
  721. void operation.done.then(() => { settled = true })
  722. readiness.reject(new Error('inspection failed during cancellation'))
  723. await Promise.resolve()
  724. await Promise.resolve()
  725. expect(settled).toBe(false)
  726. expect(() => session.startSend({ text: 'successor', submit: true })).toThrow(TerminalError)
  727. signalling.resolve(undefined)
  728. expect(await signalled.promise).toBe(456)
  729. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  730. await session.close('test complete')
  731. expect((await operation.done).waitReason).toBe('session_exit')
  732. })
  733. it('signals only after an in-flight provider write lands', async () => {
  734. vi.useFakeTimers()
  735. const terminal = new FakeTerminal()
  736. const inspector = new FakeInspector()
  737. const session = makeSession(terminal, inspector, config())
  738. await initialize(session, terminal)
  739. const writeGate = Promise.withResolvers<undefined>()
  740. terminal.write = async () => { await writeGate.promise }
  741. const operation = session.startSend({ text: 'must be interrupted', submit: true })
  742. await Promise.resolve()
  743. await Promise.resolve()
  744. expect(operation.cancel()).toBe(true)
  745. await vi.advanceTimersByTimeAsync(20)
  746. expect(inspector.groups).toEqual([])
  747. writeGate.resolve(undefined)
  748. await vi.advanceTimersByTimeAsync(0)
  749. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  750. terminal.emitData('\x1b]133;D;130\x07dsh> ')
  751. await vi.advanceTimersByTimeAsync(10)
  752. await operation.done
  753. })
  754. it('does not signal when a cancelled provider write rejects', async () => {
  755. vi.useFakeTimers()
  756. const terminal = new FakeTerminal()
  757. const inspector = new FakeInspector()
  758. const session = makeSession(terminal, inspector, config())
  759. await initialize(session, terminal)
  760. const writeGate = Promise.withResolvers<undefined>()
  761. terminal.write = async () => { await writeGate.promise }
  762. const operation = session.startSend({ text: 'rejected write', submit: true })
  763. await Promise.resolve()
  764. await Promise.resolve()
  765. expect(operation.cancel()).toBe(true)
  766. const rejected = expect(operation.done).rejects.toThrow('write failed after cancellation')
  767. writeGate.reject(new Error('write failed after cancellation'))
  768. await rejected
  769. expect(inspector.groups).toEqual([])
  770. const next = session.startSend({ text: '', submit: false })
  771. await vi.advanceTimersByTimeAsync(100)
  772. expect((await next.done).waitReason).toBe('inferred_idle')
  773. })
  774. it('releases a timed-out cancellation after the provider write and signal settle', async () => {
  775. vi.useFakeTimers()
  776. const terminal = new FakeTerminal()
  777. const inspector = new FakeInspector()
  778. const session = makeSession(terminal, inspector, config())
  779. await initialize(session, terminal)
  780. const writeGate = Promise.withResolvers<undefined>()
  781. terminal.write = async () => { await writeGate.promise }
  782. const operation = session.startSend({ text: 'slow cancelled write', submit: true })
  783. await Promise.resolve()
  784. await Promise.resolve()
  785. expect(operation.cancel()).toBe(true)
  786. await vi.advanceTimersByTimeAsync(100)
  787. expect((await operation.done).waitReason).toBe('timeout')
  788. expect(() => session.startSend({ text: 'must wait', submit: true })).toThrow(expect.objectContaining({
  789. code: 'SEND_ACTIVE',
  790. }))
  791. writeGate.resolve(undefined)
  792. await vi.advanceTimersByTimeAsync(0)
  793. expect(inspector.groups).toContainEqual([456, 'SIGINT'])
  794. const next = session.startSend({ text: '', submit: false })
  795. await vi.advanceTimersByTimeAsync(100)
  796. expect((await next.done).waitReason).toBe('inferred_idle')
  797. })
  798. it('retains the absolute timeout after cancellation while output stays active', async () => {
  799. vi.useFakeTimers()
  800. const terminal = new FakeTerminal()
  801. const inspector = new FakeInspector()
  802. const session = makeSession(terminal, inspector, config())
  803. await initialize(session, terminal)
  804. const operation = session.startSend({ text: 'ignore-sigint-and-write', submit: true })
  805. await Promise.resolve()
  806. await Promise.resolve()
  807. expect(operation.cancel()).toBe(true)
  808. for (let elapsed = 20; elapsed <= 100; elapsed += 20) {
  809. terminal.emitData('.')
  810. await vi.advanceTimersByTimeAsync(20)
  811. }
  812. expect(await operation.done).toMatchObject({ waitReason: 'timeout' })
  813. })
  814. it('does not resume cancellation polling after the terminal exits during signalling', async () => {
  815. vi.useFakeTimers()
  816. const terminal = new FakeTerminal()
  817. const inspector = new FakeInspector()
  818. const session = makeSession(terminal, inspector, config())
  819. await initialize(session, terminal)
  820. const signalGate = Promise.withResolvers<undefined>()
  821. terminal.signalForeground = async () => {
  822. await signalGate.promise
  823. return 456
  824. }
  825. const operation = session.startSend({ text: 'first', submit: true })
  826. await Promise.resolve()
  827. await Promise.resolve()
  828. expect(operation.cancel()).toBe(true)
  829. terminal.emitExit(0)
  830. await expect(operation.done).resolves.toMatchObject({ waitReason: 'session_exit' })
  831. signalGate.resolve(undefined)
  832. await vi.advanceTimersByTimeAsync(10)
  833. expect(session.status()).toEqual({ kind: 'exited', exitCode: 0, signal: null })
  834. })
  835. it('retains send ownership after timeout until an asynchronous provider write settles', async () => {
  836. vi.useFakeTimers()
  837. const terminal = new FakeTerminal()
  838. const inspector = new FakeInspector()
  839. const session = makeSession(terminal, inspector, config())
  840. await initialize(session, terminal)
  841. const writeGate = Promise.withResolvers<undefined>()
  842. terminal.write = async () => { await writeGate.promise }
  843. const operation = session.startSend({ text: 'slow write', submit: true })
  844. await Promise.resolve()
  845. await Promise.resolve()
  846. await vi.advanceTimersByTimeAsync(100)
  847. expect((await operation.done).waitReason).toBe('timeout')
  848. expect(() => session.startSend({ text: 'must wait', submit: true })).toThrow('active send')
  849. writeGate.resolve(undefined)
  850. await Promise.resolve()
  851. await Promise.resolve()
  852. const rejectedWrite = Promise.withResolvers<undefined>()
  853. terminal.write = async () => { await rejectedWrite.promise }
  854. const rejected = session.startSend({ text: 'late rejection', submit: true })
  855. await vi.advanceTimersByTimeAsync(100)
  856. expect((await rejected.done).waitReason).toBe('timeout')
  857. rejectedWrite.reject(new Error('write failed after timeout'))
  858. await Promise.resolve()
  859. await Promise.resolve()
  860. const next = session.startSend({ text: '', submit: false })
  861. await vi.advanceTimersByTimeAsync(100)
  862. expect((await next.done).waitReason).toBe('inferred_idle')
  863. })
  864. it('retains send ownership when cancellation signalling fails during an asynchronous write', async () => {
  865. vi.useFakeTimers()
  866. const terminal = new FakeTerminal()
  867. const inspector = new FakeInspector()
  868. const session = makeSession(terminal, inspector, config())
  869. await initialize(session, terminal)
  870. const writeGate = Promise.withResolvers<undefined>()
  871. terminal.write = async () => { await writeGate.promise }
  872. let signalCalls = 0
  873. terminal.signalForeground = async () => {
  874. signalCalls += 1
  875. throw new Error('interrupt failed')
  876. }
  877. const operation = session.startSend({ text: 'slow write', submit: true })
  878. await Promise.resolve()
  879. await Promise.resolve()
  880. expect(operation.cancel()).toBe(true)
  881. await vi.advanceTimersByTimeAsync(20)
  882. expect(signalCalls).toBe(0)
  883. expect(() => session.startSend({ text: 'must wait', submit: true })).toThrow('active send')
  884. writeGate.resolve(undefined)
  885. await expect(operation.done).rejects.toThrow('interrupt failed')
  886. expect(signalCalls).toBe(1)
  887. expect(session.status()).toEqual({ kind: 'exited', exitCode: null, signal: null })
  888. expect(() => session.startSend({ text: '', submit: false })).toThrow('has exited')
  889. })
  890. it('handles startup exit, unknown exit signals, cancel-write failure, and stale polls', async () => {
  891. vi.useFakeTimers()
  892. const startupTerminal = new FakeTerminal()
  893. const startup = new LocalPtySession(startupTerminal, config())
  894. const initializing = startup.initialize(new AbortController().signal)
  895. startupTerminal.emitExit(1)
  896. await expect(initializing).rejects.toThrow('exited during startup')
  897. expect(startup.status()).toEqual({ kind: 'exited', exitCode: 1, signal: null })
  898. const terminal = new FakeTerminal()
  899. const session = new LocalPtySession(terminal, config())
  900. await initialize(session, terminal)
  901. const operation = session.startSend({ text: '', submit: false })
  902. const operationInternal = operation as unknown as {
  903. append(text: string): void
  904. settle(reason: 'timeout', status: TerminalSessionStatus, inherited: boolean): void
  905. }
  906. operationInternal.append('')
  907. const sessionInternal = session as unknown as {
  908. pollReadiness(operation: TerminalSendOperation): void
  909. interrupt(operation: TerminalSendOperation): void
  910. schedulePoll(operation: TerminalSendOperation): void
  911. polling: boolean
  912. statusValue: TerminalSessionStatus
  913. appendOutput(text: string): void
  914. }
  915. sessionInternal.appendOutput('')
  916. sessionInternal.pollReadiness({} as TerminalSendOperation)
  917. sessionInternal.schedulePoll({} as TerminalSendOperation)
  918. sessionInternal.polling = true
  919. sessionInternal.schedulePoll(operation)
  920. sessionInternal.polling = false
  921. sessionInternal.interrupt({} as TerminalSendOperation)
  922. sessionInternal.statusValue = { kind: 'exited', exitCode: 2, signal: null }
  923. sessionInternal.pollReadiness(operation)
  924. await operation.done
  925. operationInternal.settle('timeout', { kind: 'running' }, false)
  926. const unknownTerminal = new FakeTerminal()
  927. const unknown = new LocalPtySession(unknownTerminal, config())
  928. unknownTerminal.emitExit(1, 999)
  929. await vi.waitFor(() => {
  930. expect(unknown.status()).toEqual({ kind: 'exited', exitCode: null, signal: null })
  931. })
  932. const cancelTerminal = new FakeTerminal()
  933. const cancelInspector = new FakeInspector()
  934. const cancel = makeSession(cancelTerminal, cancelInspector, config())
  935. await initialize(cancel, cancelTerminal)
  936. const cancellable = cancel.startSend({ text: '', submit: false })
  937. cancelInspector.throwGroup = true
  938. expect(cancellable.cancel()).toBe(true)
  939. await expect(cancellable.done).rejects.toThrow('group failed')
  940. expect(cancellable.cancel()).toBe(false)
  941. const missingGroupTerminal = new FakeTerminal()
  942. const missingGroupInspector = new FakeInspector()
  943. const missingGroup = makeSession(missingGroupTerminal, missingGroupInspector, config())
  944. await initialize(missingGroup, missingGroupTerminal)
  945. missingGroupInspector.pgid = undefined
  946. const unresolved = missingGroup.startSend({ text: '', submit: false })
  947. expect(unresolved.cancel()).toBe(true)
  948. await expect(unresolved.done).rejects.toThrow('cannot resolve foreground process group')
  949. })
  950. it('does not treat zero-output startup silence as readiness and fails on startup timeout', async () => {
  951. vi.useFakeTimers()
  952. const terminal = new FakeTerminal()
  953. const session = new LocalPtySession(terminal, config())
  954. let settled = false
  955. const initializing = session.initialize().then(() => { settled = true })
  956. await vi.advanceTimersByTimeAsync(60)
  957. expect(settled).toBe(false)
  958. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  959. await vi.advanceTimersByTimeAsync(10)
  960. await initializing
  961. const timeoutTerminal = new FakeTerminal()
  962. const timeout = new LocalPtySession(timeoutTerminal, config())
  963. const timedOut = expect(timeout.initialize()).rejects.toThrow('startup timeout')
  964. await vi.advanceTimersByTimeAsync(100)
  965. await timedOut
  966. })
  967. it('preserves the caller abort reason when startup cannot resolve a foreground group', async () => {
  968. const terminal = new FakeTerminal()
  969. const inspector = new FakeInspector()
  970. inspector.pgid = undefined
  971. const session = makeSession(terminal, inspector, config())
  972. const controller = new AbortController()
  973. const reason = new Error('startup cancelled')
  974. const initializing = session.initialize(controller.signal)
  975. const rejected = expect(initializing).rejects.toBe(reason)
  976. controller.abort(reason)
  977. await rejected
  978. })
  979. it('waits for printable prompt text when the startup marker is split from PS1', async () => {
  980. vi.useFakeTimers()
  981. const terminal = new FakeTerminal()
  982. const session = new LocalPtySession(terminal, config())
  983. let settled = false
  984. const initializing = session.initialize().then(() => { settled = true })
  985. terminal.emitData('\x1b]133;D;0\x07')
  986. await vi.advanceTimersByTimeAsync(20)
  987. expect(settled).toBe(false)
  988. terminal.emitData('dsh> ')
  989. await vi.advanceTimersByTimeAsync(10)
  990. await initializing
  991. expect(session.motd).toBe('dsh> ')
  992. })
  993. it('does not attribute a delayed prior prompt to the current send', async () => {
  994. vi.useFakeTimers()
  995. const terminal = new FakeTerminal()
  996. const session = new LocalPtySession(terminal, config({ idleSilenceMs: 100, timeoutMs: 200 }))
  997. await initialize(session, terminal)
  998. const operation = session.startSend({ text: "printf 'PID=%s\\n' \"$!\"", submit: true })
  999. let settled = false
  1000. void operation.done.then(() => { settled = true })
  1001. await Promise.resolve()
  1002. await Promise.resolve()
  1003. terminal.emitData('\x1b]133;D;0\x07dsh> printf \'PID=%s\\n\' "$!"\r\n')
  1004. await vi.advanceTimersByTimeAsync(20)
  1005. expect(settled).toBe(false)
  1006. terminal.emitData('PID=123\r\n\x1b]133;D;0\x07dsh> ')
  1007. await vi.advanceTimersByTimeAsync(10)
  1008. expect(await operation.done).toMatchObject({ waitReason: 'stdin_read' })
  1009. })
  1010. it('retains a prompt marker until the startup shell regains the foreground group', async () => {
  1011. vi.useFakeTimers()
  1012. const terminal = new FakeTerminal()
  1013. const inspector = new FakeInspector()
  1014. const session = makeSession(terminal, inspector, config())
  1015. await initialize(session, terminal)
  1016. const operation = session.startSend({ text: 'run', submit: true })
  1017. let settled = false
  1018. void operation.done.then(() => { settled = true })
  1019. await Promise.resolve()
  1020. await Promise.resolve()
  1021. inspector.pgid = 789
  1022. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1023. await vi.advanceTimersByTimeAsync(50)
  1024. expect(settled).toBe(false)
  1025. inspector.pgid = 456
  1026. await vi.advanceTimersByTimeAsync(10)
  1027. expect(settled).toBe(true)
  1028. expect((await operation.done).waitReason).toBe('stdin_read')
  1029. })
  1030. it('holds the idle fallback for the configured handoff grace, not one poll', async () => {
  1031. vi.useFakeTimers()
  1032. const terminal = new FakeTerminal()
  1033. const inspector = new FakeInspector()
  1034. const session = makeSession(terminal, inspector, config({ handoffGraceMs: 40 }))
  1035. await initialize(session, terminal)
  1036. const operation = session.startSend({ text: 'run', submit: true })
  1037. let settled = false
  1038. void operation.done.then(() => { settled = true })
  1039. await Promise.resolve()
  1040. await Promise.resolve()
  1041. inspector.pgid = 789
  1042. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1043. // One poll past the silence bound would already have settled inferred_idle.
  1044. await vi.advanceTimersByTimeAsync(70)
  1045. expect(settled).toBe(false)
  1046. inspector.pgid = 456
  1047. await vi.advanceTimersByTimeAsync(10)
  1048. expect((await operation.done).waitReason).toBe('stdin_read')
  1049. })
  1050. it('falls back to inferred idle when a foreground child emits an inherited prompt marker', async () => {
  1051. vi.useFakeTimers()
  1052. const terminal = new FakeTerminal()
  1053. const inspector = new FakeInspector()
  1054. const session = makeSession(terminal, inspector, config())
  1055. await initialize(session, terminal)
  1056. const operation = session.startSend({ text: 'bash -i', submit: true })
  1057. inspector.pgid = 789
  1058. terminal.emitData('\x1b]133;D;0\x07child> ')
  1059. await vi.advanceTimersByTimeAsync(100)
  1060. expect((await operation.done).waitReason).toBe('inferred_idle')
  1061. })
  1062. it('contains terminal transport failures and preserves the first failure', async () => {
  1063. const terminal = new FakeTerminal()
  1064. terminal.terminateError = new Error('cleanup after transport failure')
  1065. const session = new LocalPtySession(terminal, config())
  1066. const operation = session.startSend({ text: '', submit: false })
  1067. terminal.output.emit('data', 'plain text')
  1068. terminal.emitError(new Error('output transport failed'))
  1069. ;(session as unknown as { onTransportFailure(error: unknown): void })
  1070. .onTransportFailure(new Error('later failure'))
  1071. await expect(operation.done).rejects.toThrow('output transport failed')
  1072. expect(session.status()).toEqual({ kind: 'exited', exitCode: null, signal: null })
  1073. terminal.terminateError = undefined
  1074. await expect(session.close('transport')).rejects.toThrow('output transport failed')
  1075. const rejectedTerminal = new FakeTerminal()
  1076. const rejected = new LocalPtySession(rejectedTerminal, config())
  1077. const rejectedOperation = rejected.startSend({ text: '', submit: false })
  1078. rejectedTerminal.emitFailure('raw transport failure')
  1079. await expect(rejectedOperation.done).rejects.toThrow('raw transport failure')
  1080. })
  1081. it('replaces invalid UTF-8 terminal output', async () => {
  1082. const chunkTerminal = new FakeTerminal()
  1083. const chunkSession = new LocalPtySession(chunkTerminal, config())
  1084. const chunkOperation = chunkSession.startSend({ text: '', submit: false })
  1085. chunkTerminal.emitBytes(Uint8Array.from([0xff]))
  1086. expect(chunkOperation.readOutput()).toEqual({ delta: '�', truncated: false })
  1087. chunkTerminal.emitExit()
  1088. await chunkOperation.done
  1089. const endTerminal = new FakeTerminal()
  1090. const endSession = new LocalPtySession(endTerminal, config())
  1091. const endOperation = endSession.startSend({ text: '', submit: false })
  1092. endTerminal.emitBytes(Uint8Array.from([0xe2]))
  1093. endTerminal.emitExit()
  1094. expect((await endOperation.done).viewport).toBe('�')
  1095. })
  1096. it('contains readiness inspection failure and a stale inspection result', async () => {
  1097. vi.useFakeTimers()
  1098. const failedTerminal = new FakeTerminal()
  1099. const failedSession = new LocalPtySession(failedTerminal, config())
  1100. const failedOperation = failedSession.startSend({ text: '', submit: false })
  1101. await Promise.resolve()
  1102. await Promise.resolve()
  1103. failedTerminal.inspectForeground = async () => { throw new Error('inspect failed') }
  1104. const failedInternal = failedSession as unknown as {
  1105. pollReadiness(operation: TerminalSendOperation): Promise<void>
  1106. }
  1107. await failedInternal.pollReadiness(failedOperation)
  1108. await expect(failedOperation.done).rejects.toThrow('inspect failed')
  1109. const staleTerminal = new FakeTerminal()
  1110. const staleSession = new LocalPtySession(staleTerminal, config())
  1111. const staleOperation = staleSession.startSend({ text: '', submit: false })
  1112. await Promise.resolve()
  1113. await Promise.resolve()
  1114. const gate = Promise.withResolvers<ReturnType<FakeTerminal['inspectForeground']> extends Promise<infer T> ? T : never>()
  1115. staleTerminal.inspectForeground = async () => await gate.promise
  1116. const staleInternal = staleSession as unknown as {
  1117. active: TerminalSendOperation | undefined
  1118. pollReadiness(operation: TerminalSendOperation): Promise<void>
  1119. }
  1120. const polling = staleInternal.pollReadiness(staleOperation)
  1121. staleInternal.active = undefined
  1122. gate.resolve({ processGroupId: 456, inputWaiting: false })
  1123. await polling
  1124. ;(staleOperation as unknown as {
  1125. settle(reason: 'timeout', status: TerminalSessionStatus, inherited: boolean): void
  1126. }).settle('timeout', { kind: 'running' }, false)
  1127. })
  1128. it('reschedules readiness for a new send after a stale remote inspection releases the poll slot', async () => {
  1129. vi.useFakeTimers()
  1130. const terminal = new FakeTerminal()
  1131. const inspector = new FakeInspector()
  1132. const session = makeSession(terminal, inspector, config())
  1133. await initialize(session, terminal)
  1134. const old = session.startSend({ text: '', submit: false })
  1135. await Promise.resolve()
  1136. await Promise.resolve()
  1137. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  1138. let block = true
  1139. terminal.inspectForeground = async () => block
  1140. ? await inspection.promise
  1141. : { processGroupId: 456, inputWaiting: false }
  1142. const internals = session as unknown as {
  1143. pollReadiness(operation: TerminalSendOperation): Promise<void>
  1144. settleActive(reason: 'timeout'): void
  1145. }
  1146. const stalePoll = internals.pollReadiness(old)
  1147. internals.settleActive('timeout')
  1148. await old.done
  1149. block = false
  1150. const current = session.startSend({ text: '', submit: false })
  1151. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1152. await Promise.resolve()
  1153. await Promise.resolve()
  1154. inspection.resolve({ processGroupId: 456, inputWaiting: false })
  1155. await vi.advanceTimersByTimeAsync(0)
  1156. await stalePoll
  1157. await vi.advanceTimersByTimeAsync(10)
  1158. expect((await current.done).waitReason).toBe('stdin_read')
  1159. })
  1160. it('does not poll a successor before its own pre-write inspection and write complete', async () => {
  1161. vi.useFakeTimers()
  1162. const terminal = new FakeTerminal()
  1163. const inspector = new FakeInspector()
  1164. const session = makeSession(terminal, inspector, config())
  1165. await initialize(session, terminal)
  1166. const old = session.startSend({ text: '', submit: false })
  1167. await Promise.resolve()
  1168. await Promise.resolve()
  1169. const staleInspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  1170. const successorInspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  1171. let inspectCalls = 0
  1172. terminal.inspectForeground = async () => {
  1173. inspectCalls += 1
  1174. return inspectCalls === 1 ? await staleInspection.promise : await successorInspection.promise
  1175. }
  1176. const internals = session as unknown as {
  1177. pollReadiness(operation: TerminalSendOperation): Promise<void>
  1178. settleActive(reason: 'timeout'): void
  1179. }
  1180. const stalePoll = internals.pollReadiness(old)
  1181. internals.settleActive('timeout')
  1182. await old.done
  1183. const current = session.startSend({ text: 'successor', submit: true })
  1184. await Promise.resolve()
  1185. await Promise.resolve()
  1186. staleInspection.resolve({ processGroupId: 456, inputWaiting: false })
  1187. await stalePoll
  1188. await vi.advanceTimersByTimeAsync(10)
  1189. expect(inspectCalls).toBe(2)
  1190. expect(terminal.writes).toEqual([])
  1191. successorInspection.resolve({ processGroupId: 456, inputWaiting: false })
  1192. await Promise.resolve()
  1193. await Promise.resolve()
  1194. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1195. await vi.advanceTimersByTimeAsync(10)
  1196. expect(terminal.writes).toEqual(['successor\r'])
  1197. expect((await current.done).waitReason).toBe('stdin_read')
  1198. })
  1199. it('contains stale timer, write, inspection, and interrupt continuations', async () => {
  1200. vi.useFakeTimers()
  1201. const settle = (operation: TerminalSendOperation): void => {
  1202. ;(operation as unknown as {
  1203. settle(reason: 'timeout', status: TerminalSessionStatus, inherited: boolean): void
  1204. }).settle('timeout', { kind: 'running' }, false)
  1205. }
  1206. const deadlineTerminal = new FakeTerminal()
  1207. const deadlineSession = new LocalPtySession(deadlineTerminal, config())
  1208. const deadlineOperation = deadlineSession.startSend({ text: '', submit: false })
  1209. ;(deadlineSession as unknown as { active: TerminalSendOperation | undefined }).active = undefined
  1210. await vi.advanceTimersByTimeAsync(100)
  1211. settle(deadlineOperation)
  1212. const writeTerminal = new FakeTerminal()
  1213. const writeGate = Promise.withResolvers<undefined>()
  1214. writeTerminal.write = async () => { await writeGate.promise }
  1215. const writeSession = new LocalPtySession(writeTerminal, config())
  1216. const writeOperation = writeSession.startSend({ text: 'x', submit: false })
  1217. await Promise.resolve()
  1218. await Promise.resolve()
  1219. ;(writeSession as unknown as { closing: boolean }).closing = true
  1220. writeGate.resolve(undefined)
  1221. await Promise.resolve()
  1222. await Promise.resolve()
  1223. settle(writeOperation)
  1224. const beginTerminal = new FakeTerminal()
  1225. const beginGate = Promise.withResolvers<never>()
  1226. beginTerminal.inspectForeground = async () => await beginGate.promise
  1227. const beginSession = new LocalPtySession(beginTerminal, config())
  1228. const beginOperation = beginSession.startSend({ text: '', submit: false })
  1229. ;(beginSession as unknown as { active: TerminalSendOperation | undefined }).active = undefined
  1230. beginGate.reject(new Error('stale begin failure'))
  1231. await Promise.resolve()
  1232. await Promise.resolve()
  1233. settle(beginOperation)
  1234. const scheduledTerminal = new FakeTerminal()
  1235. const scheduledSession = new LocalPtySession(scheduledTerminal, config())
  1236. const scheduledOperation = scheduledSession.startSend({ text: '', submit: false })
  1237. await Promise.resolve()
  1238. await Promise.resolve()
  1239. const scheduledInternal = scheduledSession as unknown as {
  1240. schedulePoll(operation: TerminalSendOperation, delayMs?: number): void
  1241. settleActive(reason: 'timeout'): void
  1242. }
  1243. scheduledInternal.schedulePoll(scheduledOperation, 5)
  1244. scheduledInternal.settleActive('timeout')
  1245. await scheduledOperation.done
  1246. const pollTerminal = new FakeTerminal()
  1247. const pollSession = new LocalPtySession(pollTerminal, config())
  1248. const pollOperation = pollSession.startSend({ text: '', submit: false })
  1249. await Promise.resolve()
  1250. await Promise.resolve()
  1251. const pollGate = Promise.withResolvers<never>()
  1252. pollTerminal.inspectForeground = async () => await pollGate.promise
  1253. const pollInternal = pollSession as unknown as {
  1254. active: TerminalSendOperation | undefined
  1255. pollReadiness(operation: TerminalSendOperation): Promise<void>
  1256. }
  1257. const stalePoll = pollInternal.pollReadiness(pollOperation)
  1258. pollInternal.active = undefined
  1259. pollGate.reject(new Error('stale poll failure'))
  1260. await stalePoll
  1261. settle(pollOperation)
  1262. const interruptTerminal = new FakeTerminal()
  1263. const interruptGate = Promise.withResolvers<never>()
  1264. interruptTerminal.signalForeground = async () => await interruptGate.promise
  1265. const interruptSession = new LocalPtySession(interruptTerminal, config())
  1266. const interruptOperation = interruptSession.startSend({ text: '', submit: false })
  1267. expect(interruptOperation.cancel()).toBe(true)
  1268. ;(interruptSession as unknown as { active: TerminalSendOperation | undefined }).active = undefined
  1269. interruptGate.reject(new Error('stale interrupt failure'))
  1270. await Promise.resolve()
  1271. await Promise.resolve()
  1272. settle(interruptOperation)
  1273. })
  1274. })
  1275. describe('LocalPtySession bounds, signals, and teardown', () => {
  1276. it('validates pagination and enforces line/UTF-8 bounds', async () => {
  1277. vi.useFakeTimers()
  1278. const terminal = new FakeTerminal()
  1279. const session = new LocalPtySession(
  1280. terminal,
  1281. config({ scrollbackLines: 3, scrollbackMaxBytes: 12, maxReadBytes: 6 }),
  1282. )
  1283. expect(session.read({})).toMatchObject({ text: '' })
  1284. await initialize(session, terminal)
  1285. const operation = session.startSend({ text: '', submit: false })
  1286. terminal.emitData('一\n二\n三\n四')
  1287. await vi.advanceTimersByTimeAsync(60)
  1288. expect((await operation.done).truncated).toBe(true)
  1289. const page = session.read({ offset: 0, count: 3 })
  1290. expect(Buffer.byteLength(page.text)).toBeLessThanOrEqual(6)
  1291. expect(page.truncated).toBe(true)
  1292. expect(session.read({ offset: 999 })).toMatchObject({ text: '', lineBegin: 999, lineEnd: 999 })
  1293. expect(() => session.read({ offset: -1 })).toThrow('offset')
  1294. expect(() => session.read({ count: 0 })).toThrow('count')
  1295. const tinyTerminal = new FakeTerminal()
  1296. const tiny = new LocalPtySession(tinyTerminal, config({ maxReadBytes: 1 }))
  1297. await initialize(tiny, tinyTerminal)
  1298. const tinyOperation = tiny.startSend({ text: '', submit: false })
  1299. tinyTerminal.emitData('一')
  1300. await vi.advanceTimersByTimeAsync(60)
  1301. await tinyOperation.done
  1302. expect(tiny.read({ offset: 0, count: 1 }).text).toBe('')
  1303. })
  1304. it('signals verified groups and refuses unresolved or shell-targeted hard kills', async () => {
  1305. const terminal = new FakeTerminal()
  1306. const inspector = new FakeInspector()
  1307. const session = makeSession(terminal, inspector, config())
  1308. expect(await session.signal('SIGINT')).toEqual({ delivered: true, targetPgid: 456 })
  1309. inspector.pgid = terminal.pid
  1310. await expect(session.signal('SIGKILL')).rejects.toThrow('terminate the terminal session')
  1311. inspector.pgid = undefined
  1312. await expect(session.signal('SIGTERM')).rejects.toThrow('cannot resolve')
  1313. })
  1314. it('closes idempotently and rejects new signals', async () => {
  1315. const terminal = new FakeTerminal()
  1316. terminal.throwKill = true
  1317. const session = new LocalPtySession(terminal, config({ disposeGraceMs: 1 }))
  1318. const closing = session.close('test')
  1319. expect(session.close('other')).toBe(closing)
  1320. await expect(closing).rejects.toThrow('PTY cleanup failed (test)')
  1321. expect(() => session.startSend({ text: '', submit: false })).toThrow('closing')
  1322. await expect(session.signal('SIGTERM')).rejects.toThrow('closing')
  1323. })
  1324. it('reports cleanup failure without waiting for top-level exit and permits retry', async () => {
  1325. const terminal = new FakeTerminal()
  1326. terminal.autoExitOnKill = false
  1327. terminal.terminateError = new Error('terminal cleanup failed; surviving pids: 456')
  1328. const session = new LocalPtySession(terminal, config())
  1329. await expect(session.close('survivor')).rejects.toMatchObject({
  1330. message: 'PTY cleanup failed (survivor)',
  1331. cause: terminal.terminateError,
  1332. })
  1333. expect((session as unknown as { emulatorClosed: boolean }).emulatorClosed).toBe(true)
  1334. expect(terminal.kills).toEqual([])
  1335. terminal.terminateError = undefined
  1336. terminal.autoExitOnKill = true
  1337. await expect(session.close('retry')).resolves.toBeUndefined()
  1338. expect(terminal.kills).toEqual(['SIGTERM'])
  1339. })
  1340. it('settles an active send as session_exit when closed mid-operation', async () => {
  1341. vi.useFakeTimers()
  1342. const terminal = new FakeTerminal()
  1343. const session = new LocalPtySession(terminal, config({ disposeGraceMs: 50 }))
  1344. await initialize(session, terminal)
  1345. const operation = session.startSend({ text: 'run', submit: true })
  1346. // The shell returns to its prompt while the send is active; a running
  1347. // readiness poll would otherwise mis-settle this as stdin_read once close
  1348. // begins, so teardown must stop polling before its grace period.
  1349. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1350. terminal.autoExitOnKill = false
  1351. const closing = session.close('mid-send')
  1352. await vi.advanceTimersByTimeAsync(20)
  1353. terminal.emitExit(0, 15)
  1354. expect((await operation.done).waitReason).toBe('session_exit')
  1355. await closing
  1356. })
  1357. it('settles a closing send when provider termination cancels inspection', async () => {
  1358. vi.useFakeTimers()
  1359. const terminal = new FakeTerminal()
  1360. const session = new LocalPtySession(terminal, config())
  1361. await initialize(session, terminal)
  1362. const write = Promise.withResolvers<undefined>()
  1363. terminal.write = async () => { await write.promise }
  1364. const writeOperation = session.startSend({ text: 'pending write', submit: true })
  1365. await Promise.resolve()
  1366. await Promise.resolve()
  1367. await session.close('pending write')
  1368. expect((await writeOperation.done).waitReason).toBe('session_exit')
  1369. // The rejection lands after close released the send; it must stay contained.
  1370. write.reject(new Error('write rejected during close'))
  1371. await Promise.resolve()
  1372. await Promise.resolve()
  1373. })
  1374. it('settles a closing send when provider termination cancels a pending inspection', async () => {
  1375. vi.useFakeTimers()
  1376. const terminal = new FakeTerminal()
  1377. const session = new LocalPtySession(terminal, config())
  1378. await initialize(session, terminal)
  1379. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  1380. terminal.inspectForeground = async () => await inspection.promise
  1381. const terminate = terminal.terminate.bind(terminal)
  1382. terminal.terminate = async () => {
  1383. inspection.reject(new Error('terminal terminated'))
  1384. await terminate()
  1385. }
  1386. const operation = session.startSend({ text: 'pending inspection', submit: true })
  1387. await Promise.resolve()
  1388. await session.close('pending inspection')
  1389. expect((await operation.done).waitReason).toBe('session_exit')
  1390. expect(terminal.writes).toEqual([])
  1391. })
  1392. it('does not let an in-flight readiness inspection outrun close', async () => {
  1393. vi.useFakeTimers()
  1394. const terminal = new FakeTerminal()
  1395. const session = new LocalPtySession(terminal, config())
  1396. await initialize(session, terminal)
  1397. const inspection = Promise.withResolvers<{ processGroupId: number; inputWaiting: boolean }>()
  1398. const originalInspect = terminal.inspectForeground.bind(terminal)
  1399. let inspections = 0
  1400. terminal.inspectForeground = async () => {
  1401. inspections += 1
  1402. return inspections === 1 ? await originalInspect() : await inspection.promise
  1403. }
  1404. const operation = session.startSend({ text: 'pending readiness', submit: true })
  1405. await Promise.resolve()
  1406. await Promise.resolve()
  1407. terminal.emitData('\x1b]133;D;0\x07dsh> ')
  1408. await vi.advanceTimersByTimeAsync(10)
  1409. expect(inspections).toBe(2)
  1410. const termination = Promise.withResolvers<undefined>()
  1411. terminal.terminate = async () => {
  1412. await termination.promise
  1413. terminal.emitExit(0, 15)
  1414. }
  1415. const closing = session.close('in-flight readiness')
  1416. let settled = false
  1417. void operation.done.then(() => { settled = true })
  1418. inspection.resolve({ processGroupId: 456, inputWaiting: true })
  1419. await Promise.resolve()
  1420. await Promise.resolve()
  1421. expect(settled).toBe(false)
  1422. termination.resolve(undefined)
  1423. await closing
  1424. expect((await operation.done).waitReason).toBe('session_exit')
  1425. })
  1426. })