session.spec.ts 70 KB

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