1
0

session.spec.ts 65 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657
  1. import { describe, expect, expectTypeOf, it, vi } from 'vitest'
  2. import { Context } from '@deepseek-ai/cordis'
  3. import { createUserMessage, ToolCallId, createMessage, createToolResultMessage, MessageId, ReasoningEffortId } from '@deepseek-ai/dsh-llm'
  4. import SessionStore, {
  5. adoptSessionEvent,
  6. SESSION_FORMAT_VERSION,
  7. Session,
  8. SessionEvent,
  9. SessionId,
  10. snapshotSessionEvent,
  11. } from '@deepseek-ai/dsh-session'
  12. import type { CreateSessionOptions, SessionEventType, SessionHeader, SessionSurface } from '@deepseek-ai/dsh-session'
  13. describe('Session', () => {
  14. it('exposes one stable readonly surface view', () => {
  15. const session = Session.create(SessionId('surface-view'))
  16. const surface = session.surface
  17. expectTypeOf(surface).toEqualTypeOf<SessionSurface>()
  18. expect(surface).toBe(session.surface)
  19. })
  20. it('derives message history from the event log', () => {
  21. const session = Session.create(SessionId('s1'))
  22. session.append('turn/start', { turn: 1 })
  23. session.append('user/message', createUserMessage({
  24. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  25. }), { surfaceOp: 'append' })
  26. session.append('assistant/chunk', { turn: 1, step: 1, chunk: { type: 'text-delta', index: 0, text: 'hi' } })
  27. session.append('assistant/message', {
  28. turn: 1, step: 1,
  29. message: createMessage({
  30. role: 'assistant',
  31. content: [
  32. { type: 'text', text: 'let me check' },
  33. { type: 'tool-call', id: ToolCallId('c1'), name: 'echo', arguments: '{}' },
  34. ],
  35. source: {
  36. kind: 'model',
  37. ...{ provider: 'mock', model: 'mock' },
  38. },
  39. }),
  40. }, { surfaceOp: 'append' })
  41. session.append('tool/result', {
  42. turn: 1, step: 1,
  43. message: createToolResultMessage({
  44. callId: ToolCallId('c1'),
  45. content: [{ type: 'text', text: 'ok' }],
  46. isError: false,
  47. }),
  48. }, { surfaceOp: 'append' })
  49. session.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
  50. const messages = session.deriveMessages()
  51. expect(messages.map(m => m.role)).toEqual(['user', 'assistant', 'user'])
  52. // raw chunks must NOT appear in derived history
  53. expect(messages[1]!.content).toHaveLength(2)
  54. expect(messages[2]!.content[0]).toMatchObject({ type: 'tool-result', toolCallId: ToolCallId('c1') })
  55. })
  56. it('accepts and round-trips a max-tokens turn/end reason', () => {
  57. // The max-tokens TurnEndReason variant carries no extra data, so it must
  58. // append and persist like any other reason (JSON-serializable, no fields).
  59. const session = Session.create(SessionId('s1'))
  60. session.append('turn/start', { turn: 1 })
  61. session.append('turn/end', { turn: 1, reason: { kind: 'max-tokens' } })
  62. const turnEnd = session.events.findLast(e => e.type === 'turn/end')!
  63. expect(turnEnd.data.reason).toEqual({ kind: 'max-tokens' })
  64. // survives a structuredClone (the persistence-serialization boundary)
  65. expect(structuredClone(turnEnd.data.reason)).toEqual({ kind: 'max-tokens' })
  66. })
  67. it('round-trips an aborted turn with its cancellation cause', () => {
  68. const session = Session.create(SessionId('aborted'))
  69. session.append('turn/start', { turn: 1 })
  70. session.append('turn/end', { turn: 1, reason: { kind: 'aborted', reason: { kind: 'user' } } })
  71. const replayed = Session.create(SessionId('aborted-replay'), structuredClone(session.events))
  72. expect(replayed.events.slice(0, -1)).toEqual(session.events)
  73. const turnEnd = replayed.events.findLast(event => event.type === 'turn/end')
  74. expect(turnEnd?.type === 'turn/end' && turnEnd.data.reason)
  75. .toEqual({ kind: 'aborted', reason: { kind: 'user' } })
  76. })
  77. it('renders injected-context and user messages as plain user content', () => {
  78. const session = Session.create(SessionId('s2'))
  79. session.append('user/message', createUserMessage({
  80. content: [{ type: 'text', text: 'file changed: a.ts' }],
  81. source: { kind: 'plugin', plugin: 'watcher' },
  82. }), { surfaceOp: 'append' })
  83. session.append('user/message', createUserMessage({
  84. content: [{ type: 'text', text: 'focus on tests' }],
  85. source: { kind: 'user' },
  86. }), { surfaceOp: 'append' })
  87. const [contextMessage, steeringMessage] = session.deriveMessages()
  88. expect(contextMessage!.role).toBe('user')
  89. expect(contextMessage!.content).toEqual([{ type: 'text', text: 'file changed: a.ts' }])
  90. expect(steeringMessage!.role).toBe('user')
  91. expect(steeringMessage!.content).toEqual([{ type: 'text', text: 'focus on tests' }])
  92. })
  93. it('keeps the exact identified context message in durable history and projection', () => {
  94. const session = Session.create(SessionId('s2-raw'))
  95. const message = createUserMessage({
  96. content: [{ type: 'text', text: '<system-reminder>Additional instructions from: pkg/AGENTS.md</system-reminder>' }],
  97. source: { kind: 'plugin', plugin: 'agent-instructions' },
  98. })
  99. session.append('user/message', message, { surfaceOp: 'append' })
  100. expect(session.deriveMessages()).toEqual([message])
  101. const event = session.events[0]
  102. expect(event?.type === 'user/message' && event.data.source).toEqual({ kind: 'plugin', plugin: 'agent-instructions' })
  103. })
  104. it('replays identically from a seeded event log', () => {
  105. const original = Session.create(SessionId('s3'))
  106. original.append('turn/start', { turn: 1 })
  107. original.append('user/message', createUserMessage({
  108. content: [{ type: 'text', text: 'q' }], source: { kind: 'user' },
  109. }), { surfaceOp: 'append' })
  110. original.append('assistant/message', {
  111. turn: 1, step: 1,
  112. message: createMessage({
  113. role: 'assistant',
  114. content: [{ type: 'text', text: 'a' }],
  115. source: {
  116. kind: 'model',
  117. ...{ provider: 'mock', model: 'mock' },
  118. },
  119. }),
  120. }, { surfaceOp: 'append' })
  121. original.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
  122. const replayed = Session.create(SessionId('s3-replay'), [...original.events])
  123. expect(replayed.deriveMessages()).toEqual(original.deriveMessages())
  124. // The seed verbatim, plus the end-seed event the constructor appends.
  125. expect(replayed.events.slice(0, original.seq)).toEqual(original.events)
  126. expect(replayed.seq).toBe(original.seq + 1)
  127. expect(replayed.firstLiveSeq).toBe(original.seq)
  128. })
  129. it('marks an explicitly empty seed without marking a fresh session', () => {
  130. const fresh = Session.create(SessionId('fresh-empty'))
  131. expect(fresh.events).toEqual([])
  132. const resumed = Session.create(SessionId('resumed-empty'), [])
  133. expect(resumed.firstLiveSeq).toBe(0)
  134. expect(resumed.events).toMatchObject([
  135. { type: 'session/end-seed', seq: 0, data: {} },
  136. ])
  137. const reopened = Session.create(SessionId('reopened-empty'), resumed.events)
  138. expect(reopened.firstLiveSeq).toBe(1)
  139. expect(reopened.events).toEqual(resumed.events)
  140. })
  141. it('rejects pre-provider request headers and assistant messages on seed/load', () => {
  142. const requestHeader = {
  143. type: 'request/header', seq: 0, time: 1,
  144. data: { header: { config: { model: 'old-model' } }, reason: 'initial' },
  145. } as unknown as SessionEvent
  146. expect(() => Session.create(SessionId('old-header'), [requestHeader]))
  147. .toThrow('seed request/header at index 0 lacks provider/model')
  148. const assistantMessage = {
  149. type: 'assistant/message', seq: 0, time: 1,
  150. data: { turn: 1, step: 1, content: [{ type: 'text', text: 'old' }] },
  151. surfaceOp: 'append',
  152. } as unknown as SessionEvent
  153. expect(() => Session.create(SessionId('old-assistant'), [assistantMessage]))
  154. .toThrow('seed assistant/message at index 0 lacks an identified message')
  155. const malformedHeader = {
  156. type: 'request/header', seq: 0, time: 1,
  157. data: { header: 'old-header' },
  158. } as unknown as SessionEvent
  159. expect(() => Session.create(SessionId('malformed-header'), [malformedHeader]))
  160. .toThrow('seed request/header at index 0 lacks provider/model')
  161. const unrelatedPrimitiveData = {
  162. type: 'plugin/event', seq: 0, time: 1, data: null,
  163. } as unknown as SessionEvent
  164. expect(Session.create(SessionId('primitive-plugin-data'), [unrelatedPrimitiveData]).events.slice(0, 1))
  165. .toEqual([unrelatedPrimitiveData])
  166. })
  167. it('rejects event-specific malformed message shapes on seed/load', () => {
  168. const user = {
  169. id: 'user',
  170. role: 'user',
  171. content: [{ type: 'text', text: 'content' }],
  172. source: { kind: 'user' },
  173. }
  174. const assistant = {
  175. id: 'assistant',
  176. role: 'assistant',
  177. content: [{ type: 'text', text: 'content' }],
  178. source: { kind: 'model', provider: 'mock', model: 'mock' },
  179. }
  180. const tool = {
  181. id: 'tool',
  182. role: 'user',
  183. content: [{
  184. type: 'tool-result',
  185. toolCallId: 'call',
  186. content: [{ type: 'text', text: 'result' }],
  187. }],
  188. source: { kind: 'tool', callId: 'call' },
  189. }
  190. const invalid = [
  191. {
  192. name: 'message record',
  193. event: {
  194. type: 'user/message', seq: 0, time: 1, surfaceOp: 'append',
  195. data: null,
  196. },
  197. message: 'lacks an identified message',
  198. },
  199. {
  200. name: 'user role',
  201. event: {
  202. type: 'user/message', seq: 0, time: 1, surfaceOp: 'append',
  203. data: { ...user, role: 'assistant' },
  204. },
  205. message: 'message must have role "user"',
  206. },
  207. {
  208. name: 'source',
  209. event: {
  210. type: 'user/message', seq: 0, time: 1, surfaceOp: 'append',
  211. data: { ...user, source: null },
  212. },
  213. message: 'message has invalid source',
  214. },
  215. {
  216. name: 'content shape',
  217. event: {
  218. type: 'user/message', seq: 0, time: 1, surfaceOp: 'append',
  219. data: { ...user, content: 'not-an-array' },
  220. },
  221. message: 'message has invalid content',
  222. },
  223. {
  224. name: 'assistant source',
  225. event: {
  226. type: 'assistant/message', seq: 0, time: 1, surfaceOp: 'append',
  227. data: {
  228. turn: 1,
  229. step: 1,
  230. message: { ...assistant, source: { kind: 'user' } },
  231. },
  232. },
  233. message: 'message must have model source',
  234. },
  235. {
  236. name: 'tool source',
  237. event: {
  238. type: 'tool/result', seq: 0, time: 1, surfaceOp: 'append',
  239. data: {
  240. turn: 1,
  241. step: 1,
  242. message: { ...tool, source: { kind: 'user' } },
  243. },
  244. },
  245. message: 'message must have tool source',
  246. },
  247. {
  248. name: 'tool tuple',
  249. event: {
  250. type: 'tool/result', seq: 0, time: 1, surfaceOp: 'append',
  251. data: {
  252. turn: 1,
  253. step: 1,
  254. message: { ...tool, content: [{ type: 'text', text: 'not a result' }] },
  255. },
  256. },
  257. message: 'message must contain one tool-result block',
  258. },
  259. {
  260. name: 'tool correlation',
  261. event: {
  262. type: 'tool/result', seq: 0, time: 1, surfaceOp: 'append',
  263. data: {
  264. turn: 1,
  265. step: 1,
  266. message: {
  267. ...tool,
  268. source: { kind: 'tool', callId: 'other-call' },
  269. },
  270. },
  271. },
  272. message: 'message has mismatched tool call ids',
  273. },
  274. ] as const
  275. for (const { name, event, message } of invalid) {
  276. expect(
  277. () => Session.create(SessionId(`invalid-${name}`), [event as unknown as SessionEvent]),
  278. name,
  279. ).toThrow(message)
  280. }
  281. })
  282. it('snapshots message events without validating plugin-owned block details', () => {
  283. const boundary = snapshotSessionEvent({
  284. type: 'turn/start',
  285. seq: 0,
  286. time: 1,
  287. data: { turn: 1 },
  288. })
  289. expect(boundary).toEqual({
  290. type: 'turn/start',
  291. seq: 0,
  292. time: 1,
  293. data: { turn: 1 },
  294. })
  295. const extended = snapshotSessionEvent({
  296. type: 'user/message',
  297. seq: 0,
  298. time: 1,
  299. surfaceOp: 'append',
  300. data: {
  301. id: 'extended-message',
  302. role: 'user',
  303. content: [{ type: 'plugin-block', value: 1 }],
  304. source: { kind: 'plugin-source', value: 1 },
  305. },
  306. } as unknown as SessionEvent)
  307. expect(extended.type === 'user/message' && extended.data.content)
  308. .toEqual([{ type: 'plugin-block', value: 1 }])
  309. })
  310. it('adopts exclusively owned messages in place and keeps snapshots detached', () => {
  311. const owned = {
  312. type: 'user/message',
  313. seq: 0,
  314. time: 1,
  315. surfaceOp: 'append',
  316. data: {
  317. id: 'owned-message',
  318. role: 'user',
  319. content: [{ type: 'text', text: 'owned' }],
  320. source: { kind: 'user' },
  321. },
  322. } as SessionEvent<'user/message'>
  323. expect(adoptSessionEvent(owned)).toBe(owned)
  324. expect(Object.isFrozen(owned.data)).toBe(true)
  325. expect(Object.isFrozen(owned.data.content)).toBe(true)
  326. const source = structuredClone(owned)
  327. const snapshot = snapshotSessionEvent(source)
  328. expect(snapshot).not.toBe(source)
  329. expect(snapshot.data).not.toBe(source.data)
  330. expect(snapshot.data.content).not.toBe(source.data.content)
  331. })
  332. it('validates message shape before adopting ownership', () => {
  333. const malformed = {
  334. type: 'user/message',
  335. seq: 0,
  336. time: 1,
  337. data: {
  338. id: 'wrong-role',
  339. role: 'assistant',
  340. content: [],
  341. source: { kind: 'user' },
  342. },
  343. } as unknown as SessionEvent
  344. expect(() => adoptSessionEvent(malformed)).toThrow('message must have role "user"')
  345. })
  346. it('round-trips a non-empty reasoning effort and rejects invalid durable values', () => {
  347. const valid = {
  348. type: 'request/header',
  349. seq: 0,
  350. time: 1,
  351. data: {
  352. header: {
  353. config: {
  354. provider: 'mock',
  355. model: 'model',
  356. reasoningEffort: ReasoningEffortId('adapter-owned'),
  357. },
  358. },
  359. reason: 'initial',
  360. },
  361. } as const
  362. expect(Session.create(SessionId('reasoning-effort'), [valid]).events[0])
  363. .toEqual(valid)
  364. for (const reasoningEffort of ['', 1]) {
  365. const invalid = structuredClone(valid) as unknown as SessionEvent
  366. if (invalid.type !== 'request/header') throw new Error('test fixture must be a request header')
  367. const config = invalid.data.header.config as unknown as Record<string, unknown>
  368. config.reasoningEffort = reasoningEffort
  369. expect(() => Session.create(SessionId('invalid-reasoning-effort'), [invalid]))
  370. .toThrow('seed request/header at index 0 has an invalid reasoningEffort')
  371. }
  372. })
  373. it('round-trips adapter-default markers and rejects invalid durable values', () => {
  374. const valid = {
  375. type: 'request/header',
  376. seq: 0,
  377. time: 1,
  378. data: {
  379. header: {
  380. config: {
  381. provider: 'mock',
  382. model: 'model',
  383. maxTokens: 256_000,
  384. },
  385. adapterDefaults: { maxTokens: true },
  386. },
  387. reason: 'initial',
  388. },
  389. } as const
  390. expect(Session.create(SessionId('adapter-defaults'), [valid]).events[0]).toEqual(valid)
  391. for (const adapterDefaults of [
  392. null,
  393. [],
  394. { unknown: true },
  395. { maxTokens: false },
  396. { reasoningEffort: true },
  397. ]) {
  398. const invalid = structuredClone(valid) as unknown as SessionEvent
  399. if (invalid.type !== 'request/header') throw new Error('test fixture must be a request header')
  400. invalid.data.header.adapterDefaults = adapterDefaults as never
  401. expect(() => Session.create(SessionId('invalid-adapter-defaults'), [invalid]))
  402. .toThrow('seed request/header at index 0 has invalid adapterDefaults')
  403. }
  404. })
  405. it('isolates the log from mutation through a derived message (append-only contract)', () => {
  406. const session = Session.create(SessionId('s4'))
  407. session.append('user/message', createUserMessage({
  408. content: [{ type: 'text', text: 'original' }], source: { kind: 'user' },
  409. }), { surfaceOp: 'append' })
  410. session.append('tool/result', {
  411. turn: 1, step: 1,
  412. message: createToolResultMessage({
  413. callId: ToolCallId('c1'),
  414. content: [{ type: 'text', text: 'tool out' }],
  415. isError: false,
  416. }),
  417. }, { surfaceOp: 'append' })
  418. const before = structuredClone(session.events)
  419. // A misbehaving consumer tries to mutate the messages it was handed.
  420. const messages = session.deriveMessages()
  421. const userBlock = messages[0]!.content[0]!
  422. expect(() => { if (userBlock.type === 'text') userBlock.text = 'HACKED' }).toThrow(TypeError)
  423. const toolBlock = messages[1]!.content[0]!
  424. expect(() => {
  425. if (toolBlock.type === 'tool-result') toolBlock.content.push({ type: 'text', text: 'injected' })
  426. }).toThrow(TypeError)
  427. expect(() => { messages[0]!.content.push({ type: 'text', text: 'extra' }) }).toThrow(TypeError)
  428. // The returned ARRAY is the caller's own snapshot, though — reordering it
  429. // is the caller's business and never reaches the cache or the log.
  430. messages.reverse()
  431. // The log is unchanged: deep-equal to the snapshot taken before mutation.
  432. expect(session.events).toEqual(before)
  433. // And a fresh derivation still reflects the original content and order.
  434. expect(session.deriveMessages()[0]!.content).toEqual([{ type: 'text', text: 'original' }])
  435. })
  436. it('rejects non-JSON-serializable event data at the source (incl. sparse arrays)', () => {
  437. const session = Session.create(SessionId('s5'))
  438. const bad = (extra: unknown) => () => session.append('user/message', { content: [{ type: 'text', text: 'x' }], source: { kind: 'user' }, extra } as never, { surfaceOp: 'append' })
  439. expect(bad(1n)).toThrow(/non-JSON-serializable/)
  440. expect(bad(() => 0)).toThrow(/non-JSON-serializable/)
  441. expect(bad(Symbol('s'))).toThrow(/non-JSON-serializable/)
  442. expect(bad(new Map())).toThrow(/non-JSON-serializable/)
  443. expect(bad(undefined)).toThrow(/non-JSON-serializable/)
  444. expect(bad(Infinity)).toThrow(/non-JSON-serializable/)
  445. // A sparse array: `every` skips the hole but JSON.stringify writes it null.
  446. // Build the hole without a sparse literal or `delete` (both linted).
  447. const sparse: unknown[] = Array(3)
  448. sparse[0] = 1
  449. sparse[2] = 3 // index 1 stays a hole
  450. expect(bad(sparse)).toThrow(/non-JSON-serializable/)
  451. // A DENSE array carrying a non-serializable element is rejected too.
  452. expect(bad([1, 2n, 3])).toThrow(/non-JSON-serializable/)
  453. // A nested non-serializable value (inside a plain object) is rejected.
  454. expect(bad({ nested: { deep: () => 0 } })).toThrow(/non-JSON-serializable/)
  455. // A circular reference is rejected (the seen-set guard, not a stack blow-up).
  456. const cyclic: Record<string, unknown> = { a: 1 }
  457. cyclic['self'] = cyclic
  458. expect(bad(cyclic)).toThrow(/non-JSON-serializable/)
  459. // The rejected appends never entered the log.
  460. expect(session.events).toHaveLength(0)
  461. })
  462. it('rejects a surface-eligible append with no surfaceOp marker (runtime guard for the union-widening loophole)', () => {
  463. const session = Session.create(SessionId('s5b'))
  464. session.append('turn/start', { turn: 1 })
  465. // A widened SessionEventType bypasses the overload's conditional requirement,
  466. // so the runtime guard must still reject the missing surface marker.
  467. const widenedType = 'user/message' as SessionEventType
  468. expect(() => session.append(widenedType, createUserMessage({
  469. content: [{ type: 'text', text: 'hi' }], source: { kind: 'user' },
  470. })))
  471. .toThrow(/surface-eligible and requires a surfaceOp marker/)
  472. // The rejected append never entered the log (only turn/start is present).
  473. expect(session.events).toHaveLength(1)
  474. })
  475. it('accepts dense arrays and nested plain objects', () => {
  476. const session = Session.create(SessionId('s6'))
  477. expect(() => session.append('user/message', { content: [{ type: 'text', text: 'x' }], source: { kind: 'user' }, extra: [1, 2, [3, { a: null, b: true }]] } as never, { surfaceOp: 'append' })).not.toThrow()
  478. expect(session.events).toHaveLength(1)
  479. })
  480. it('validates seed events: rejects a non-JSON-serializable seed', () => {
  481. // A replay/fork seed must satisfy the SAME invariant as Session.append, or
  482. // it builds a live log no backend can persist.
  483. const badSeed = [
  484. { type: 'user/message' as const, seq: 0, time: 1, data: { content: [{ type: 'text' as const, text: 'x' }], source: { kind: 'user' as const }, bad: 1n } },
  485. ] as unknown as SessionEvent[]
  486. expect(() => Session.create(SessionId('seed-bad'), badSeed)).toThrow(/losslessly JSON-serializable/)
  487. })
  488. it('validates seed events: rejects a non-contiguous seq', () => {
  489. const gapSeed = [
  490. { type: 'turn/start' as const, seq: 0, time: 1, data: { turn: 1 } },
  491. { type: 'turn/end' as const, seq: 5, time: 2, data: { turn: 1, reason: { kind: 'completed' as const } } }, // gap: expected seq 1
  492. ] as SessionEvent[]
  493. expect(() => Session.create(SessionId('seed-gap'), gapSeed)).toThrow(/contiguous|seq/)
  494. })
  495. it('validates seed events: rejects a surface-eligible event missing its surfaceOp marker', () => {
  496. // A surface-eligible event (user/message) with no surfaceOp would load fine
  497. // but vanish from deriveMessages() (the surface is the sole derivation path),
  498. // so a resume/fork would silently lose history. append() forbids this at
  499. // compile time; a raw seed must be rejected at runtime to match.
  500. const markerlessSeed = [
  501. { type: 'turn/start' as const, seq: 0, time: 1, data: { turn: 1 } },
  502. { type: 'user/message' as const, seq: 1, time: 2, data: createUserMessage({
  503. content: [{ type: 'text' as const, text: 'hi' }], source: { kind: 'user' as const },
  504. }) },
  505. { type: 'turn/end' as const, seq: 2, time: 3, data: { turn: 1, reason: { kind: 'completed' as const } } },
  506. ] as SessionEvent[]
  507. expect(() => Session.create(SessionId('seed-no-marker'), markerlessSeed)).toThrow(/requires a surfaceOp marker/)
  508. })
  509. it('accepts a well-formed contiguous serializable seed', () => {
  510. const goodSeed = [
  511. { type: 'turn/start' as const, seq: 0, time: 1, data: { turn: 1 } },
  512. { type: 'user/message' as const, seq: 1, time: 2, data: createUserMessage({
  513. content: [{ type: 'text' as const, text: 'hi' }], source: { kind: 'user' as const },
  514. }), surfaceOp: 'append' as const },
  515. { type: 'turn/end' as const, seq: 2, time: 3, data: { turn: 1, reason: { kind: 'completed' as const } } },
  516. ] as SessionEvent[]
  517. const session = Session.create(SessionId('seed-ok'), goodSeed)
  518. expect(session.events.slice(0, 3)).toEqual(goodSeed)
  519. expect(session.firstLiveSeq).toBe(3)
  520. })
  521. it('reads each seed array entry once so validation and storage use the same event', () => {
  522. const accepted = {
  523. type: 'turn/start' as const,
  524. seq: 0,
  525. time: 1,
  526. data: { turn: 1 },
  527. }
  528. const drifted = { ...accepted, seq: 99, data: { invalid: 1n } }
  529. let reads = 0
  530. const seed = new Array<SessionEvent>(1)
  531. Object.defineProperty(seed, 0, {
  532. enumerable: true,
  533. get: () => {
  534. reads += 1
  535. return reads === 1 ? accepted : drifted
  536. },
  537. })
  538. const session = Session.create(SessionId('seed-entry-snapshot'), seed)
  539. expect(reads).toBe(1)
  540. expect(session.events.slice(0, 1)).toEqual([accepted])
  541. })
  542. it('reads a nested seed-data getter once and stores its first JSON value', () => {
  543. let reads = 0
  544. const data = Object.defineProperty({}, 'value', {
  545. enumerable: true,
  546. get: () => {
  547. reads += 1
  548. return reads === 1 ? 'accepted' : 1n
  549. },
  550. })
  551. const seed = [{ type: 'test/unstable', seq: 0, time: 1, data }] as unknown as SessionEvent[]
  552. const session = Session.create(SessionId('seed-nested-drift'), seed)
  553. expect(reads).toBe(1)
  554. expect(session.events[0]!.data).toEqual({ value: 'accepted' })
  555. })
  556. it('rejects non-JSON surface metadata in a seed event', () => {
  557. const seed = [{
  558. type: 'user/message',
  559. seq: 0,
  560. time: 1,
  561. data: createUserMessage({
  562. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  563. }),
  564. surfaceOp: { op: 'replace', start: 1n, end: 2 },
  565. }] as unknown as SessionEvent[]
  566. expect(() => Session.create(SessionId('seed-bad-metadata'), seed))
  567. .toThrow(/losslessly JSON-serializable/)
  568. })
  569. it('rejects exotic seed metadata before cloning can erase its prototype', () => {
  570. class ReplaceOp {
  571. readonly op = 'replace' as const
  572. readonly start = 0
  573. readonly end = 0
  574. }
  575. const seed = [{
  576. type: 'user/message',
  577. seq: 0,
  578. time: 1,
  579. data: createUserMessage({
  580. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  581. }),
  582. surfaceOp: new ReplaceOp(),
  583. }] as unknown as SessionEvent[]
  584. expect(() => Session.create(SessionId('seed-exotic-metadata'), seed))
  585. .toThrow(/losslessly JSON-serializable/)
  586. })
  587. it('rejects an exotic seed event shell before spreading erases its prototype', () => {
  588. class SeedEvent {
  589. readonly type = 'turn/start' as const
  590. readonly seq = 0
  591. readonly time = 1
  592. readonly data = { turn: 1 }
  593. }
  594. const seed: SessionEvent[] = [new SeedEvent()]
  595. expect(() => Session.create(SessionId('seed-exotic-shell'), seed))
  596. .toThrow(/not losslessly JSON-serializable/)
  597. })
  598. it('accepts a null-prototype seed event shell as a plain JSON record', () => {
  599. const event = Object.assign(Object.create(null) as Record<string, unknown>, {
  600. type: 'turn/start' as const,
  601. seq: 0,
  602. time: 1,
  603. data: { turn: 1 },
  604. }) as unknown as SessionEvent
  605. const session = Session.create(SessionId('seed-null-prototype'), [event])
  606. expect(session.events.slice(0, 1)).toEqual([{ ...event }])
  607. })
  608. it('reads a nested seed-metadata getter once and stores its first JSON value', () => {
  609. let reads = 0
  610. const surfaceOp = Object.defineProperty({ op: 'replace', end: 0 }, 'start', {
  611. enumerable: true,
  612. get: () => {
  613. reads += 1
  614. return reads === 1 ? 0 : 1n
  615. },
  616. })
  617. const seed = [{
  618. type: 'user/message',
  619. seq: 0,
  620. time: 1,
  621. data: createUserMessage({
  622. content: [{ type: 'text', text: 'source' }], source: { kind: 'user' },
  623. }),
  624. surfaceOp: 'append',
  625. }, {
  626. type: 'user/message',
  627. seq: 1,
  628. time: 2,
  629. data: createUserMessage({
  630. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  631. }),
  632. surfaceOp,
  633. sourceEventSeqs: [0],
  634. }] as unknown as SessionEvent[]
  635. const session = Session.create(SessionId('seed-unstable-metadata'), seed)
  636. const event = session.events[1]!
  637. if (event.type !== 'user/message') throw new Error('test fixture must remain a user/message')
  638. expect(reads).toBe(1)
  639. expect(event.surfaceOp).toEqual({ op: 'replace', start: 0, end: 0 })
  640. })
  641. it.each([
  642. ['an Error', new Error('validator failed'), 'validator failed'],
  643. ['a non-Error value', 'validator failed', 'invalid surface metadata'],
  644. ] as const)('adds seed context when surface validation throws %s', (_name, failure, expected) => {
  645. const originalHasOwn = Object.hasOwn
  646. const hasOwn = vi.spyOn(Object, 'hasOwn').mockImplementation((object: object, property: PropertyKey): boolean => {
  647. if ((object as Record<string, unknown>)['op'] === 'replace') throw failure
  648. return originalHasOwn(object, property)
  649. })
  650. const seed = [{
  651. type: 'user/message',
  652. seq: 0,
  653. time: 1,
  654. data: createUserMessage({
  655. content: [{ type: 'text', text: 'source' }], source: { kind: 'user' },
  656. }),
  657. surfaceOp: 'append',
  658. }, {
  659. type: 'user/message',
  660. seq: 1,
  661. time: 2,
  662. data: createUserMessage({
  663. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  664. }),
  665. surfaceOp: { op: 'replace', start: 0, end: 0 },
  666. sourceEventSeqs: [0],
  667. }] as unknown as SessionEvent[]
  668. try {
  669. expect(() => Session.create(SessionId('seed-non-error-metadata-failure'), seed))
  670. .toThrow(`invalid seed event at index 1: ${expected}`)
  671. } finally {
  672. hasOwn.mockRestore()
  673. }
  674. })
  675. it('snapshots the seed: mutating the original after construction does not affect session.events', () => {
  676. const seed = [
  677. { type: 'turn/start' as const, seq: 0, time: 1, data: { turn: 1 } },
  678. { type: 'user/message' as const, seq: 1, time: 2, data: {
  679. id: MessageId('seed-input'),
  680. role: 'user' as const,
  681. content: [{ type: 'text' as const, text: 'original' }], source: { kind: 'user' as const },
  682. }, surfaceOp: 'append' as const },
  683. { type: 'turn/end' as const, seq: 2, time: 3, data: { turn: 1, reason: { kind: 'completed' as const } } },
  684. ] as SessionEvent[]
  685. const session = Session.create(SessionId('seed-snapshot'), seed)
  686. // Mutate the ORIGINAL seed objects after construction: a shared reference
  687. // would let this rewrite the forked log (or reintroduce non-serializable
  688. // data past validation). The snapshot must shield session.events.
  689. const um = seed[1]!
  690. ;(um.data as { content: { type: 'text'; text: string }[] }).content[0]!.text = 'HACKED'
  691. ;(um.data as Record<string, unknown>)['injected'] = 1n // would have failed validation
  692. const logged = session.events[1]!
  693. expect(logged.type === 'user/message' && (logged.data.content[0] as { text: string }).text).toBe('original')
  694. expect((logged.data as Record<string, unknown>)['injected']).toBeUndefined()
  695. })
  696. it('snapshots append data: mutating the passed object after append does not affect session.events', () => {
  697. const session = Session.create(SessionId('append-snapshot'))
  698. const data = {
  699. id: MessageId('append-input'),
  700. role: 'user' as const,
  701. content: [{ type: 'text' as const, text: 'original' }],
  702. source: { kind: 'user' as const },
  703. }
  704. const event = session.append('user/message', data, { surfaceOp: 'append' })
  705. // Mutate the caller's object after append returns. A shared reference would
  706. // make session.events diverge from the value that passed validation.
  707. data.content[0]!.text = 'HACKED'
  708. ;(data as Record<string, unknown>)['injected'] = 1n
  709. const logged = session.events[0]!
  710. expect(logged.type === 'user/message' && (logged.data.content[0] as { text: string }).text).toBe('original')
  711. expect((logged.data as Record<string, unknown>)['injected']).toBeUndefined()
  712. // The returned event carries the same snapshot, not the caller's input.
  713. expect((event.data.content[0] as { text: string }).text).toBe('original')
  714. })
  715. it('reads a nested append-data getter once and stores its first JSON value', () => {
  716. const session = Session.create(SessionId('append-nested-drift'))
  717. let reads = 0
  718. const data = Object.defineProperty({}, 'value', {
  719. enumerable: true,
  720. get: () => {
  721. reads += 1
  722. return reads === 1 ? 'accepted' : 1n
  723. },
  724. })
  725. const event = session.append('request/context', data as never)
  726. expect(reads).toBe(1)
  727. expect(event.data).toEqual({ value: 'accepted' })
  728. expect(session.events).toEqual([event])
  729. })
  730. it('rejects non-JSON surface metadata before appending the event', () => {
  731. const session = Session.create(SessionId('append-bad-metadata'))
  732. expect(() => session.append(
  733. 'user/message',
  734. createUserMessage({
  735. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  736. }),
  737. { surfaceOp: { op: 'replace', start: 1n, end: 2 } } as never,
  738. )).toThrow(/non-JSON-serializable surface metadata/)
  739. expect(session.events).toEqual([])
  740. })
  741. it('rejects exotic surface metadata before cloning can erase its prototype', () => {
  742. class ReplaceOp {
  743. readonly op = 'replace' as const
  744. readonly start = 0
  745. readonly end = 0
  746. }
  747. const session = Session.create(SessionId('append-exotic-metadata'))
  748. expect(() => session.append(
  749. 'user/message',
  750. createUserMessage({
  751. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  752. }),
  753. { surfaceOp: new ReplaceOp() },
  754. )).toThrow(/non-JSON-serializable surface metadata/)
  755. expect(session.events).toEqual([])
  756. })
  757. it('reads a nested append-metadata getter once and stores its first JSON value', () => {
  758. const session = Session.create(SessionId('append-unstable-metadata'))
  759. const source = session.append(
  760. 'user/message',
  761. createUserMessage({
  762. content: [{ type: 'text', text: 'source' }], source: { kind: 'user' },
  763. }),
  764. { surfaceOp: 'append' },
  765. )
  766. let reads = 0
  767. const surfaceOp = Object.defineProperty({ op: 'replace', end: 0 }, 'start', {
  768. enumerable: true,
  769. get: () => {
  770. reads += 1
  771. return reads === 1 ? 0 : 1n
  772. },
  773. })
  774. const event = session.append(
  775. 'user/message',
  776. createUserMessage({
  777. content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' },
  778. }),
  779. { surfaceOp, sourceEventSeqs: [0] } as never,
  780. )
  781. expect(reads).toBe(1)
  782. expect(event.surfaceOp).toEqual({ op: 'replace', start: 0, end: 0 })
  783. expect(session.events).toEqual([source, event])
  784. })
  785. it('rejects invalid plain surface metadata shapes at append', () => {
  786. const session = Session.create(SessionId('append-invalid-surface-shape'))
  787. const appendRaw = session.append.bind(session) as unknown as (
  788. type: SessionEventType,
  789. data: unknown,
  790. opts?: unknown,
  791. ) => SessionEvent
  792. const data = { content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' } }
  793. expect(() => appendRaw('user/message', data, { surfaceOp: 'invalid' }))
  794. .toThrow(/invalid surfaceOp/)
  795. expect(() => appendRaw('user/message', data, {
  796. surfaceOp: { op: 'replace', start: -1, end: 0 },
  797. })).toThrow(/invalid replace surfaceOp/)
  798. expect(() => appendRaw('user/message', data, {
  799. surfaceOp: 'append',
  800. sourceEventSeqs: [0, -1],
  801. })).toThrow(/non-negative safe integers/)
  802. expect(session.events).toEqual([])
  803. })
  804. it('rejects surface metadata on non-surface append and seed events', () => {
  805. const session = Session.create(SessionId('non-surface-metadata'))
  806. const appendRaw = session.append.bind(session) as unknown as (
  807. type: SessionEventType,
  808. data: unknown,
  809. opts?: unknown,
  810. ) => SessionEvent
  811. expect(() => appendRaw(
  812. 'turn/start',
  813. { turn: 1 },
  814. { surfaceOp: 'append' },
  815. )).toThrow(/not surface-eligible and cannot carry surfaceOp/)
  816. expect(() => Session.create(SessionId('non-surface-metadata-seed'), [{
  817. type: 'turn/start',
  818. seq: 0,
  819. time: 1,
  820. data: { turn: 1 },
  821. surfaceOp: 'append',
  822. } as unknown as SessionEvent])).toThrow(/invalid seed event.*not surface-eligible/)
  823. expect(session.events).toEqual([])
  824. })
  825. it('deep-freezes seeded and appended event snapshots', () => {
  826. const seeded = Session.create(SessionId('seed-frozen'), [{
  827. type: 'turn/start',
  828. seq: 0,
  829. time: 1,
  830. data: { turn: 1 },
  831. }])
  832. const seededEvent = seeded.events[0]!
  833. if (seededEvent.type !== 'turn/start') throw new Error('test fixture must remain a turn/start')
  834. expect(Object.isFrozen(seededEvent)).toBe(true)
  835. expect(Object.isFrozen(seededEvent.data)).toBe(true)
  836. expect(() => { seededEvent.data.turn = 99 }).toThrow(TypeError)
  837. const appended = Session.create(SessionId('append-frozen'))
  838. const appendedEvent = appended.append('user/message', createUserMessage({
  839. content: [{ type: 'text', text: 'first' }], source: { kind: 'user' },
  840. }), { surfaceOp: 'append' })
  841. expect(Object.isFrozen(appendedEvent)).toBe(true)
  842. expect(Object.isFrozen(appendedEvent.data)).toBe(true)
  843. expect(Object.isFrozen(appendedEvent.data.content)).toBe(true)
  844. expect(Object.isFrozen(appendedEvent.data.content[0])).toBe(true)
  845. expect(() => { (appendedEvent.data.content[0] as { text: string }).text = 'mutated' }).toThrow(TypeError)
  846. })
  847. it('iteratively freezes deeply nested restored event data', () => {
  848. const depth = 20_000
  849. const data: Record<string, unknown> = {}
  850. let tail = data
  851. for (let index = 0; index < depth; index += 1) {
  852. const child: Record<string, unknown> = {}
  853. tail['child'] = child
  854. tail = child
  855. }
  856. const event = {
  857. type: 'test/deep-restore', seq: 0, time: 1, data,
  858. } as unknown as SessionEvent
  859. expect(() => Session.fromRestore(SessionId('deep-restore'), [event], {
  860. version: SESSION_FORMAT_VERSION,
  861. id: SessionId('deep-restore'),
  862. createdAt: 1,
  863. })).not.toThrow()
  864. let current: unknown = event
  865. let frozenNodes = 0
  866. for (let index = 0; index <= depth + 1; index += 1) {
  867. if (!Object.isFrozen(current)) break
  868. frozenNodes += 1
  869. current = (current as Record<string, unknown>)['data']
  870. ?? (current as Record<string, unknown>)['child']
  871. }
  872. expect(frozenNodes).toBe(depth + 2)
  873. })
  874. it('returns cached frozen event-array snapshots that do not grow after append', () => {
  875. const session = Session.create(SessionId('events-snapshot'))
  876. session.append('turn/start', { turn: 1 })
  877. const before = session.events
  878. const beforeEvent = before[0]!
  879. if (beforeEvent.type !== 'turn/start') throw new Error('test fixture must remain a turn/start')
  880. expect(session.events).toBe(before)
  881. expect(Object.isFrozen(before)).toBe(true)
  882. expect(() => { (before as SessionEvent[]).push(beforeEvent) }).toThrow(TypeError)
  883. expect(() => { beforeEvent.data.turn = 99 }).toThrow(TypeError)
  884. session.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
  885. const after = session.events
  886. expect(before).toHaveLength(1)
  887. expect(after).toHaveLength(2)
  888. expect(after).not.toBe(before)
  889. expect(session.events).toBe(after)
  890. })
  891. it('detaches and freezes an explicitly supplied session header', () => {
  892. const input = {
  893. version: SESSION_FORMAT_VERSION,
  894. id: SessionId('header-owned'),
  895. createdAt: 123,
  896. cwd: '/accepted',
  897. parentSession: SessionId('parent'),
  898. seedLength: 2,
  899. }
  900. const session = Session.create(SessionId('header-owned'), undefined, input)
  901. input.cwd = '/caller-mutated'
  902. expect(session.header).toEqual({
  903. version: SESSION_FORMAT_VERSION,
  904. id: 'header-owned',
  905. createdAt: 123,
  906. cwd: '/accepted',
  907. parentSession: 'parent',
  908. seedLength: 2,
  909. })
  910. expect(session.header).not.toBe(input)
  911. expect(Object.isFrozen(session.header)).toBe(true)
  912. expect(Reflect.set(session.header, 'cwd', '/published-mutated')).toBe(false)
  913. expect(session.id).toBe('header-owned')
  914. expect(session.header.cwd).toBe('/accepted')
  915. })
  916. it('rejects an exotic, non-JSON, or mismatched supplied header', () => {
  917. class ExoticHeader implements SessionHeader {
  918. readonly version = SESSION_FORMAT_VERSION
  919. readonly id = SessionId('header-invalid')
  920. readonly createdAt = 123
  921. }
  922. expect(() => Session.create(SessionId('header-invalid'), undefined, new ExoticHeader()))
  923. .toThrow(/not losslessly JSON-serializable/)
  924. expect(() => Session.fromRestore(SessionId('header-invalid'), [], new ExoticHeader()))
  925. .toThrow(/not a plain JSON record/)
  926. for (const header of [null, 1, []]) {
  927. expect(() => Session.fromRestore(
  928. SessionId('header-invalid'),
  929. [],
  930. header as unknown as SessionHeader,
  931. )).toThrow(/not a plain JSON record/)
  932. }
  933. expect(() => Session.create(SessionId('header-invalid'), undefined, {
  934. version: SESSION_FORMAT_VERSION,
  935. id: SessionId('header-invalid'),
  936. createdAt: 123,
  937. parentSession: 1n,
  938. } as unknown as SessionHeader)).toThrow(/not losslessly JSON-serializable/)
  939. expect(() => Session.create(SessionId('header-invalid'), undefined, {
  940. version: SESSION_FORMAT_VERSION,
  941. id: SessionId('other'),
  942. createdAt: 123,
  943. })).toThrow(/does not match session id/)
  944. })
  945. it('rejects invalid scalar fields in an explicitly supplied header', () => {
  946. const base = {
  947. version: SESSION_FORMAT_VERSION,
  948. id: SessionId('header-shape'),
  949. createdAt: 123,
  950. }
  951. const cases: Array<{ header: unknown; error: RegExp }> = [
  952. { header: 1, error: /not a plain JSON record/ },
  953. { header: null, error: /not a plain JSON record/ },
  954. { header: { ...base, version: 1 }, error: /header version/ },
  955. { header: { ...base, createdAt: '123' }, error: /createdAt must be a non-negative safe integer/ },
  956. { header: { ...base, cwd: 1 }, error: /header cwd must be a string/ },
  957. { header: { ...base, cwd: 'relative' }, error: /header cwd must be an absolute path/ },
  958. { header: { ...base, parentSession: 1 }, error: /header parentSession must be a string/ },
  959. { header: { ...base, seedLength: '1' }, error: /seedLength must be a non-negative safe integer/ },
  960. { header: { ...base, seedLength: 0.5 }, error: /seedLength must be a non-negative safe integer/ },
  961. { header: { ...base, seedLength: -1 }, error: /seedLength must be a non-negative safe integer/ },
  962. ]
  963. for (const { header, error } of cases) {
  964. expect(() => Session.create(SessionId('header-shape'), undefined, header as SessionHeader)).toThrow(error)
  965. }
  966. })
  967. it('rejects seed records with invalid fixed-envelope fields', () => {
  968. const base = {
  969. type: 'turn/start',
  970. seq: 0,
  971. time: 1,
  972. data: { turn: 1 },
  973. }
  974. const cases: unknown[] = [
  975. { ...base, extra: true },
  976. { ...base, type: 1 },
  977. { ...base, seq: '0' },
  978. { ...base, seq: 0.5 },
  979. { ...base, seq: -1 },
  980. { ...base, time: '1' },
  981. { ...base, time: 0.5 },
  982. { type: base.type, seq: base.seq, time: base.time },
  983. ]
  984. for (const [index, event] of cases.entries()) {
  985. expect(() => Session.create(SessionId(`bad-envelope-${index}`), [event as SessionEvent]))
  986. .toThrow(/invalid event envelope/)
  987. }
  988. })
  989. })
  990. describe('SessionStore', () => {
  991. it('creates sessions, emits session/created and session/event', async () => {
  992. const ctx = new Context()
  993. await ctx.plugin(SessionStore)
  994. const created: Session[] = []
  995. const events: [Session, SessionEvent][] = []
  996. ctx.on('session/created', session => void created.push(session))
  997. ctx.on('session/event', (session, event) => void events.push([session, event]))
  998. const session = ctx.sessions.create()
  999. expect(created).toEqual([session])
  1000. // The store-owned append publication hooks are module-private. A JavaScript caller
  1001. // may create an unrelated property with the old implementation's name,
  1002. // but cannot suppress the durable event feed.
  1003. expect(Reflect.set(session, 'onAppend', undefined)).toBe(true)
  1004. session.append('turn/start', { turn: 1 })
  1005. session.append('user/message', createUserMessage({
  1006. content: [{ type: 'text', text: 'x' }], source: { kind: 'user' },
  1007. }), { surfaceOp: 'append' })
  1008. expect(events).toHaveLength(2)
  1009. expect(events[1]![0]).toBe(session)
  1010. expect(events[1]![1].type).toBe('user/message')
  1011. expect(ctx.sessions.get(session.id)).toBe(session)
  1012. expect(ctx.sessions.list()).toEqual([session])
  1013. })
  1014. it('rejects duplicate ids and supports seeding', async () => {
  1015. const ctx = new Context()
  1016. await ctx.plugin(SessionStore)
  1017. const a = ctx.sessions.create(SessionId('fixed'))
  1018. expect(() => ctx.sessions.create(SessionId('fixed'))).toThrow('already exists')
  1019. a.append('turn/start', { turn: 1 })
  1020. a.append('user/message', createUserMessage({
  1021. content: [{ type: 'text', text: 'q' }], source: { kind: 'user' },
  1022. }), { surfaceOp: 'append' })
  1023. const forked = ctx.sessions.create(SessionId('fork'), { seed: [...a.events] })
  1024. expect(forked.deriveMessages()).toEqual(a.deriveMessages())
  1025. })
  1026. it('enter() rejects a stale prepared session whose id is already live (no overwrite)', async () => {
  1027. // A stale prepared object must not replace the live same-id entry; its later
  1028. // detach would otherwise remove the wrong session.
  1029. const ctx = new Context()
  1030. await ctx.plugin(SessionStore)
  1031. const stale = ctx.sessions.prepare(SessionId('racy'))
  1032. const live = ctx.sessions.create(SessionId('racy'))
  1033. expect(() => ctx.sessions.enter(stale)).toThrow(/already exists/)
  1034. // The live session is intact and still the store entry.
  1035. expect(ctx.sessions.get(SessionId('racy'))).toBe(live)
  1036. })
  1037. it('prepare() + enter() + announce() register a session and emit session/created', async () => {
  1038. const ctx = new Context()
  1039. await ctx.plugin(SessionStore)
  1040. const created: Session[] = []
  1041. ctx.on('session/created', session => void created.push(session))
  1042. const session = ctx.sessions.prepare(SessionId('lifecycle'))
  1043. // prepare alone does NOT enter the store.
  1044. expect(ctx.sessions.get(SessionId('lifecycle'))).toBeUndefined()
  1045. const detach = ctx.sessions.enter(session)
  1046. expect(ctx.sessions.get(SessionId('lifecycle'))).toBe(session)
  1047. // enter does NOT announce.
  1048. expect(created).toEqual([])
  1049. ctx.sessions.announce(session)
  1050. expect(created).toEqual([session])
  1051. // The detach disposer removes the entry + stops notification.
  1052. detach()
  1053. detach() // idempotent: cannot disturb a later same-id lifecycle
  1054. expect(ctx.sessions.get(SessionId('lifecycle'))).toBeUndefined()
  1055. })
  1056. it('prevents simultaneous attachment of one session object to two stores', async () => {
  1057. const firstCtx = new Context()
  1058. const secondCtx = new Context()
  1059. await firstCtx.plugin(SessionStore)
  1060. await secondCtx.plugin(SessionStore)
  1061. const session = Session.create(SessionId('owned-key'))
  1062. const detachFirst = firstCtx.sessions.enter(session)
  1063. expect(() => secondCtx.sessions.enter(session)).toThrow(/already attached to a store/)
  1064. expect(firstCtx.sessions.get(SessionId('owned-key'))).toBe(session)
  1065. detachFirst()
  1066. expect(firstCtx.sessions.get(SessionId('owned-key'))).toBeUndefined()
  1067. const detachSecond = secondCtx.sessions.enter(session)
  1068. expect(secondCtx.sessions.get(SessionId('owned-key'))).toBe(session)
  1069. detachSecond()
  1070. })
  1071. it('rejects direct and reentrant repeat announcements to preserve one lifecycle pair', async () => {
  1072. const ctx = new Context()
  1073. await ctx.plugin(SessionStore)
  1074. let created = 0
  1075. let disposed = 0
  1076. let reentrantError = ''
  1077. ctx.on('session/created', (session) => {
  1078. created += 1
  1079. try {
  1080. ctx.sessions.announce(session)
  1081. } catch (error: unknown) {
  1082. reentrantError = String(error)
  1083. }
  1084. })
  1085. ctx.on('session/disposed', () => { disposed += 1 })
  1086. const session = ctx.sessions.prepare(SessionId('once'))
  1087. const detach = ctx.sessions.enter(session)
  1088. ctx.sessions.announce(session)
  1089. expect(reentrantError).toMatch(/already announced/)
  1090. expect(() => { ctx.sessions.announce(session) }).toThrow(/already announced/)
  1091. detach()
  1092. expect({ created, disposed }).toEqual({ created: 1, disposed: 1 })
  1093. })
  1094. it('defers a reentrant detach until the creation dispatch unwinds', async () => {
  1095. const ctx = new Context()
  1096. await ctx.plugin(SessionStore)
  1097. const order: string[] = []
  1098. const session = ctx.sessions.prepare(SessionId('reentrant-detach'))
  1099. const detach = ctx.sessions.enter(session)
  1100. ctx.on('session/created', (created) => {
  1101. order.push('created:first')
  1102. detach()
  1103. expect(ctx.sessions.get(created.id)).toBe(created)
  1104. })
  1105. ctx.on('session/created', (created) => {
  1106. order.push('created:second')
  1107. expect(ctx.sessions.get(created.id)).toBe(created)
  1108. })
  1109. ctx.on('session/disposed', (disposed) => {
  1110. order.push('disposed')
  1111. expect(ctx.sessions.get(disposed.id)).toBeUndefined()
  1112. })
  1113. ctx.sessions.announce(session)
  1114. expect(order).toEqual(['created:first', 'created:second', 'disposed'])
  1115. expect(ctx.sessions.get(session.id)).toBeUndefined()
  1116. detach()
  1117. })
  1118. it('rolls back create when its owner unloads from session/created', async () => {
  1119. const ctx = new Context()
  1120. await ctx.plugin(SessionStore)
  1121. let ownerCtx!: Context
  1122. const owner = await ctx.plugin(Object.assign((inner: Context) => { ownerCtx = inner }, { inject: ['sessions'] }))
  1123. const id = SessionId('create-unload-race')
  1124. ctx.on('session/created', (session) => {
  1125. if (session.id === id) void owner.dispose()
  1126. })
  1127. ownerCtx.sessions.create(id)
  1128. await owner.dispose()
  1129. expect(ctx.sessions.get(id)).toBeUndefined()
  1130. })
  1131. it('synthesizes a minimal current-version header for a bare-created session', async () => {
  1132. const ctx = new Context()
  1133. await ctx.plugin(SessionStore)
  1134. const session = ctx.sessions.create(SessionId('plain'))
  1135. expect(session.header).toMatchObject({ version: SESSION_FORMAT_VERSION, id: 'plain' })
  1136. expect(Number.isSafeInteger(session.header.createdAt)).toBe(true)
  1137. expect(session.header.cwd).toBeUndefined()
  1138. expect(session.header.parentSession).toBeUndefined()
  1139. })
  1140. it('attaches cwd and parentSession from meta to the header', async () => {
  1141. const ctx = new Context()
  1142. await ctx.plugin(SessionStore)
  1143. const session = ctx.sessions.create(SessionId('child'), {
  1144. meta: { cwd: '/work/project', parentSession: SessionId('parent') },
  1145. })
  1146. expect(session.header).toMatchObject({
  1147. version: SESSION_FORMAT_VERSION,
  1148. id: 'child',
  1149. cwd: '/work/project',
  1150. parentSession: 'parent',
  1151. })
  1152. })
  1153. it('attaches subagent origin and delegationDepth from meta to the header', async () => {
  1154. const ctx = new Context()
  1155. await ctx.plugin(SessionStore)
  1156. const session = ctx.sessions.create(SessionId('delegated-child'), {
  1157. meta: { parentSession: SessionId('parent'), origin: 'subagent', delegationDepth: 2 },
  1158. })
  1159. expect(session.header).toMatchObject({
  1160. id: 'delegated-child',
  1161. parentSession: 'parent',
  1162. origin: 'subagent',
  1163. delegationDepth: 2,
  1164. })
  1165. })
  1166. it('rejects non-JSON and invalid scalar session metadata', async () => {
  1167. const ctx = new Context()
  1168. await ctx.plugin(SessionStore)
  1169. const cases: Array<{ meta: unknown; error: RegExp }> = [
  1170. { meta: { parentSession: 1n }, error: /header is not losslessly JSON-serializable/ },
  1171. { meta: { cwd: 1 }, error: /header cwd must be a string/ },
  1172. { meta: { parentSession: 1 }, error: /header parentSession must be a string/ },
  1173. { meta: { createdAt: '123' }, error: /header createdAt must be a non-negative safe integer/ },
  1174. { meta: { createdAt: 1.5 }, error: /header createdAt must be a non-negative safe integer/ },
  1175. { meta: { createdAt: -1 }, error: /header createdAt must be a non-negative safe integer/ },
  1176. { meta: { createdAt: Number.MAX_SAFE_INTEGER + 1 }, error: /header createdAt must be a non-negative safe integer/ },
  1177. { meta: { seedLength: '1' }, error: /seedLength must be a non-negative safe integer/ },
  1178. { meta: { seedLength: 0.5 }, error: /seedLength must be a non-negative safe integer/ },
  1179. { meta: { seedLength: -1 }, error: /seedLength must be a non-negative safe integer/ },
  1180. { meta: { origin: 'fork' }, error: /origin must be "subagent"/ },
  1181. { meta: { delegationDepth: '1' }, error: /delegationDepth must be a non-negative safe integer/ },
  1182. { meta: { delegationDepth: 0.5 }, error: /delegationDepth must be a non-negative safe integer/ },
  1183. { meta: { delegationDepth: -1 }, error: /delegationDepth must be a non-negative safe integer/ },
  1184. { meta: { agentPreset: 1 }, error: /agentPreset must be a string/ },
  1185. ]
  1186. for (const [index, { meta, error }] of cases.entries()) {
  1187. expect(() => ctx.sessions.prepare(SessionId(`bad-meta-${index}`), {
  1188. meta: meta as NonNullable<CreateSessionOptions['meta']>,
  1189. })).toThrow(error)
  1190. }
  1191. })
  1192. it('rejects a non-absolute meta.cwd', async () => {
  1193. const ctx = new Context()
  1194. await ctx.plugin(SessionStore)
  1195. expect(() => ctx.sessions.create(SessionId('rel'), { meta: { cwd: 'relative/path' } }))
  1196. .toThrow(/cwd must be an absolute path/)
  1197. // the rejected session was not registered
  1198. expect(ctx.sessions.get(SessionId('rel'))).toBeUndefined()
  1199. })
  1200. it('a bare Session() constructed without the store still exposes a current-version header', () => {
  1201. const session = Session.create(SessionId('bare'))
  1202. expect(session.header).toMatchObject({ version: SESSION_FORMAT_VERSION, id: 'bare' })
  1203. expect(typeof session.header.createdAt).toBe('number')
  1204. })
  1205. it('detaches sessions when the creating fiber is disposed (HMR safety)', async () => {
  1206. const ctx = new Context()
  1207. await ctx.plugin(SessionStore)
  1208. let session!: Session
  1209. const fiber = await ctx.plugin(Object.assign((inner: Context) => {
  1210. session = inner.sessions.create(SessionId('scoped'))
  1211. }, { inject: ['sessions'] }))
  1212. expect(ctx.sessions.get(SessionId('scoped'))).toBe(session)
  1213. let observed = 0
  1214. ctx.on('session/event', () => void observed++)
  1215. await fiber.dispose()
  1216. expect(ctx.sessions.get(SessionId('scoped'))).toBeUndefined()
  1217. session.append('user/message', createUserMessage({
  1218. content: [{ type: 'text', text: 'late' }], source: { kind: 'user' },
  1219. }), { surfaceOp: 'append' })
  1220. expect(observed).toBe(0)
  1221. })
  1222. it('pairs a partial session/created announcement with disposal during rollback', async () => {
  1223. const ctx = new Context()
  1224. await ctx.plugin(SessionStore)
  1225. let threw = false
  1226. const disposed: Session[] = []
  1227. ctx.on('session/disposed', (session) => { disposed.push(session) })
  1228. ctx.on('session/created', () => {
  1229. if (!threw) { threw = true; throw new Error('boom created listener') }
  1230. })
  1231. // The throwing emit must roll the store entry back, not leak it.
  1232. expect(() => ctx.sessions.create(SessionId('fixed'))).toThrow('boom created listener')
  1233. expect(ctx.sessions.get(SessionId('fixed'))).toBeUndefined() // rolled back, not leaked
  1234. expect(disposed.map(session => session.id)).toEqual(['fixed'])
  1235. // A subsequent create of the SAME id succeeds (the already-exists check is
  1236. // not wedged) and its store-owned publication hooks are correctly wired.
  1237. const events: SessionEvent[] = []
  1238. ctx.on('session/event', (_session, event) => void events.push(event))
  1239. const session = ctx.sessions.create(SessionId('fixed'))
  1240. expect(ctx.sessions.get(SessionId('fixed'))).toBe(session)
  1241. session.append('turn/start', { turn: 1 })
  1242. session.append('user/message', createUserMessage({
  1243. content: [{ type: 'text', text: 'hi' }], source: { kind: 'user' },
  1244. }), { surfaceOp: 'append' })
  1245. expect(events.at(-1)?.type).toBe('user/message')
  1246. })
  1247. it('contains session/event observer failures after the append commit point', async () => {
  1248. const ctx = new Context()
  1249. await ctx.plugin(SessionStore)
  1250. const warnings: string[] = []
  1251. ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn
  1252. const session = ctx.sessions.create(SessionId('contained-event'))
  1253. const heard: SessionEvent[] = []
  1254. let committedBeforeNotify = false
  1255. ctx.on('session/event', (observedSession, event) => {
  1256. committedBeforeNotify = observedSession.events.at(-1) === event
  1257. throw new Error('sync event observer')
  1258. })
  1259. ctx.on('session/event', () => Promise.reject(new Error('async event observer')) as never)
  1260. ctx.on('session/event', (_observedSession, event) => { heard.push(event) })
  1261. let appended!: SessionEvent
  1262. expect(() => {
  1263. appended = session.append('turn/start', {
  1264. turn: 1,
  1265. })
  1266. }).not.toThrow()
  1267. expect(committedBeforeNotify).toBe(true)
  1268. expect(session.events).toEqual([appended])
  1269. expect(heard).toEqual([appended])
  1270. await Promise.resolve()
  1271. await Promise.resolve()
  1272. expect(warnings).toEqual([
  1273. 'session "contained-event": session/event listener threw: Error: sync event observer',
  1274. 'session "contained-event": session/event listener rejected: Error: async event observer',
  1275. ])
  1276. })
  1277. it('runs internal dispatch validation on one frozen candidate before commit and resets after a veto', async () => {
  1278. const ctx = new Context()
  1279. await ctx.plugin(SessionStore)
  1280. const session = ctx.sessions.create(SessionId('dispatch-veto'))
  1281. const validations: Array<{ event: SessionEvent; logLength: number; frozen: boolean }> = []
  1282. const observed: SessionEvent[] = []
  1283. let reject = true
  1284. ctx.on('internal/dispatch', (_mode, name, args) => {
  1285. if (name !== 'session/event') return
  1286. const [observedSession, event] = args as [Session, SessionEvent]
  1287. validations.push({
  1288. event,
  1289. logLength: observedSession.events.length,
  1290. frozen: Object.isFrozen(event) && Object.isFrozen(event.data),
  1291. })
  1292. if (reject) {
  1293. reject = false
  1294. throw new Error('reject first candidate')
  1295. }
  1296. })
  1297. ctx.on('session/event', (_observedSession, event) => { observed.push(event) })
  1298. expect(() => session.append('turn/start', {
  1299. turn: 1,
  1300. })).toThrow('reject first candidate')
  1301. expect(session.events).toEqual([])
  1302. expect(observed).toEqual([])
  1303. const appended = session.append('turn/start', {
  1304. turn: 1,
  1305. })
  1306. expect(validations.map(({ logLength, frozen }) => ({ logLength, frozen }))).toEqual([
  1307. { logLength: 0, frozen: true },
  1308. { logLength: 0, frozen: true },
  1309. ])
  1310. expect(validations.map(({ event }) => event.seq)).toEqual([0, 0])
  1311. expect(validations[1]!.event).toBe(appended)
  1312. expect(session.events).toEqual([appended])
  1313. expect(observed).toEqual([appended])
  1314. })
  1315. it('does not publish a surface transition rejected by internal dispatch', async () => {
  1316. const ctx = new Context()
  1317. await ctx.plugin(SessionStore)
  1318. const session = ctx.sessions.create(SessionId('surface-dispatch-veto'))
  1319. session.append('turn/start', { turn: 1 })
  1320. session.append('step/start', { turn: 1, step: 1 })
  1321. session.append('user/message', createUserMessage({
  1322. content: [{ type: 'text', text: 'source' }],
  1323. source: { kind: 'user' },
  1324. }), { surfaceOp: 'append' })
  1325. const surface = session.surface
  1326. let reject = true
  1327. ctx.on('internal/dispatch', (_mode, name) => {
  1328. if (name === 'session/event' && reject) {
  1329. reject = false
  1330. throw new Error('reject surface candidate')
  1331. }
  1332. })
  1333. expect(() => session.append('assistant/message', {
  1334. turn: 1,
  1335. step: 1,
  1336. message: createMessage({
  1337. role: 'assistant',
  1338. content: [{ type: 'text', text: 'replacement' }],
  1339. source: {
  1340. kind: 'model',
  1341. ...{ provider: 'mock', model: 'mock' },
  1342. },
  1343. }),
  1344. }, {
  1345. surfaceOp: { op: 'replace', start: 2, end: 2 },
  1346. sourceEventSeqs: [2],
  1347. })).toThrow('reject surface candidate')
  1348. expect(session.events).toHaveLength(3)
  1349. expect(surface.nodes).toEqual([2])
  1350. expect(surface.replaceGeneration).toBe(0)
  1351. session.append('user/message', createUserMessage({
  1352. content: [{ type: 'text', text: 'next' }],
  1353. source: { kind: 'user' },
  1354. }), { surfaceOp: 'append' })
  1355. expect(surface.nodes).toEqual([2, 3])
  1356. expect(surface.replaceGeneration).toBe(0)
  1357. })
  1358. it('resolves session/event dispatch before commit so instrumentation failure cannot hide a logged event', async () => {
  1359. const ctx = new Context()
  1360. await ctx.plugin(SessionStore)
  1361. const session = ctx.sessions.create(SessionId('dispatch-check'))
  1362. const observed: SessionEvent[] = []
  1363. ctx.on('internal/dispatch', (_mode, name) => {
  1364. if (name === 'session/event') throw new Error('dispatch instrumentation rejected the carrier')
  1365. })
  1366. ctx.on('session/event', (_observedSession, event) => { observed.push(event) })
  1367. expect(() => session.append('turn/start', {
  1368. turn: 1,
  1369. })).toThrow('dispatch instrumentation rejected the carrier')
  1370. expect(session.events).toEqual([])
  1371. expect(observed).toEqual([])
  1372. })
  1373. it('contains a reentrant observer append without reordering later observers', async () => {
  1374. const ctx = new Context()
  1375. await ctx.plugin(SessionStore)
  1376. const warnings: string[] = []
  1377. ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn
  1378. const session = ctx.sessions.create(SessionId('reentrant-observer'))
  1379. const heard: SessionEvent[] = []
  1380. ctx.on('session/event', (observedSession) => {
  1381. observedSession.append('request/context', { provider: 'mock', model: 'mock' })
  1382. })
  1383. ctx.on('session/event', (_observedSession, event) => { heard.push(event) })
  1384. const appended = session.append('turn/start', {
  1385. turn: 1,
  1386. })
  1387. expect(session.events).toEqual([appended])
  1388. expect(heard).toEqual([appended])
  1389. expect(warnings).toEqual([
  1390. 'session "reentrant-observer": session/event listener threw: Error: session append cannot reenter while another append is being published',
  1391. ])
  1392. })
  1393. it('defers detach through dispatch resolution, commit, and observer publication', async () => {
  1394. const ctx = new Context()
  1395. await ctx.plugin(SessionStore)
  1396. const order: string[] = []
  1397. const session = ctx.sessions.prepare(SessionId('detach-during-append'))
  1398. const detach = ctx.sessions.enter(session)
  1399. ctx.on('internal/dispatch', (_mode, name, args) => {
  1400. if (name !== 'session/event') return
  1401. const session = args[0] as Session
  1402. order.push(`resolve:${ctx.sessions.get(session.id) === session ? 'live' : 'detached'}`)
  1403. detach()
  1404. })
  1405. ctx.on('session/event', (session) => {
  1406. order.push(`observe:${ctx.sessions.get(session.id) === session ? 'live' : 'detached'}`)
  1407. })
  1408. ctx.on('session/disposed', (session) => {
  1409. order.push(`dispose:${ctx.sessions.get(session.id) === session ? 'live' : 'detached'}`)
  1410. })
  1411. ctx.sessions.announce(session)
  1412. const appended = session.append('turn/start', {
  1413. turn: 1,
  1414. })
  1415. expect(session.events).toEqual([appended])
  1416. expect(order).toEqual(['resolve:live', 'observe:live', 'dispose:detached'])
  1417. expect(ctx.sessions.get(session.id)).toBeUndefined()
  1418. })
  1419. it('observes async session/created rejection without rolling back or starving peers', async () => {
  1420. const ctx = new Context()
  1421. await ctx.plugin(SessionStore)
  1422. const warnings: string[] = []
  1423. ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn
  1424. const heard: string[] = []
  1425. ctx.on('session/created', () => Promise.reject(new Error('late creation failure')) as never)
  1426. ctx.on('session/created', (session) => { heard.push(session.id) })
  1427. const session = ctx.sessions.create(SessionId('async-created'))
  1428. await Promise.resolve()
  1429. await Promise.resolve()
  1430. expect(ctx.sessions.get(session.id)).toBe(session)
  1431. expect(heard).toEqual(['async-created'])
  1432. expect(warnings).toEqual([
  1433. 'session "async-created": session/created listener rejected: Error: late creation failure',
  1434. ])
  1435. })
  1436. it('contains synchronous and async session/disposed listener failures per observer', async () => {
  1437. const ctx = new Context()
  1438. await ctx.plugin(SessionStore)
  1439. const warnings: string[] = []
  1440. ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn
  1441. const heard: string[] = []
  1442. ctx.on('session/disposed', () => { throw new Error('sync disposed') })
  1443. ctx.on('session/disposed', () => Promise.reject(new Error('async disposed')) as never)
  1444. ctx.on('session/disposed', (session) => { heard.push(session.id) })
  1445. const unannounced = ctx.sessions.prepare(SessionId('never-announced'))
  1446. const detachUnannounced = ctx.sessions.enter(unannounced)
  1447. detachUnannounced()
  1448. expect(heard).toEqual([])
  1449. const announced = ctx.sessions.prepare(SessionId('contained-disposal'))
  1450. const detach = ctx.sessions.enter(announced)
  1451. ctx.sessions.announce(announced)
  1452. expect(() => { detach() }).not.toThrow()
  1453. await Promise.resolve()
  1454. await Promise.resolve()
  1455. expect(heard).toEqual(['contained-disposal'])
  1456. expect(warnings).toEqual([
  1457. 'session "contained-disposal": session/disposed listener threw: Error: sync disposed',
  1458. 'session "contained-disposal": session/disposed listener rejected: Error: async disposed',
  1459. ])
  1460. })
  1461. it('contains internal dispatch failure after session detachment', async () => {
  1462. const ctx = new Context()
  1463. await ctx.plugin(SessionStore)
  1464. const warnings: string[] = []
  1465. ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn
  1466. const heard: Session[] = []
  1467. ctx.on('internal/dispatch', (_mode, name) => {
  1468. if (name === 'session/disposed') throw new Error('disposed dispatch instrumentation')
  1469. })
  1470. ctx.on('session/disposed', (session) => { heard.push(session) })
  1471. const session = ctx.sessions.prepare(SessionId('disposed-dispatch'))
  1472. const detach = ctx.sessions.enter(session)
  1473. ctx.sessions.announce(session)
  1474. expect(() => { detach() }).not.toThrow()
  1475. expect(ctx.sessions.get(session.id)).toBeUndefined()
  1476. expect(heard).toEqual([])
  1477. expect(warnings).toEqual([
  1478. 'session "disposed-dispatch": session/disposed dispatch threw: Error: disposed dispatch instrumentation',
  1479. ])
  1480. })
  1481. it('does not let internal dispatch replace the disposed callback tuple', async () => {
  1482. const ctx = new Context()
  1483. await ctx.plugin(SessionStore)
  1484. const replacement = Session.create(SessionId('replacement-disposed'))
  1485. const heard: Session[] = []
  1486. ctx.on('internal/dispatch', (_mode, name, args) => {
  1487. if (name === 'session/disposed') args[0] = replacement
  1488. })
  1489. ctx.on('session/disposed', (session) => { heard.push(session) })
  1490. const session = ctx.sessions.prepare(SessionId('fixed-disposed-tuple'))
  1491. const detach = ctx.sessions.enter(session)
  1492. ctx.sessions.announce(session)
  1493. detach()
  1494. expect(heard).toEqual([session])
  1495. })
  1496. })