session.spec.ts 67 KB

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