@@ -152,13 +152,12 @@ describe("agent-events sequencing", () => {
152152 lastActiveAt : 12_345 ,
153153 } ) ;
154154
155- expect ( getAgentRunContext ( "run-ctx" ) ) . toMatchObject ( {
156- sessionKey : "session-main" ,
157- verboseLevel : "full" ,
158- isHeartbeat : true ,
159- isControlUiVisible : true ,
160- lastActiveAt : 12_345 ,
161- } ) ;
155+ const context = getAgentRunContext ( "run-ctx" ) ;
156+ expect ( context ?. sessionKey ) . toBe ( "session-main" ) ;
157+ expect ( context ?. verboseLevel ) . toBe ( "full" ) ;
158+ expect ( context ?. isHeartbeat ) . toBe ( true ) ;
159+ expect ( context ?. isControlUiVisible ) . toBe ( true ) ;
160+ expect ( context ?. lastActiveAt ) . toBe ( 12_345 ) ;
162161 } ) ;
163162
164163 test ( "falls back to registered sessionKey when event sessionKey is blank" , ( ) => {
@@ -232,7 +231,7 @@ describe("agent-events sequencing", () => {
232231
233232 stop ( ) ;
234233
235- expect ( second . getAgentRunContext ( "run-dup" ) ) . toMatchObject ( { sessionKey : "session-dup" } ) ;
234+ expect ( second . getAgentRunContext ( "run-dup" ) ?. sessionKey ) . toBe ( "session-dup" ) ;
236235 expect ( seen ) . toEqual ( [
237236 { seq : 1 , sessionKey : "session-dup" } ,
238237 { seq : 2 , sessionKey : "session-dup" } ,
@@ -256,7 +255,7 @@ describe("agent-events sequencing", () => {
256255 stop . mockReturnValue ( 1_000 ) ;
257256 expect ( sweepStaleRunContexts ( 500 ) ) . toBe ( 1 ) ;
258257 expect ( getAgentRunContext ( "run-stale" ) ) . toBeUndefined ( ) ;
259- expect ( getAgentRunContext ( "run-active" ) ) . toMatchObject ( { sessionKey : "session-active" } ) ;
258+ expect ( getAgentRunContext ( "run-active" ) ?. sessionKey ) . toBe ( "session-active" ) ;
260259
261260 const seen : Array < { runId : string ; seq : number } > = [ ] ;
262261 const unsubscribe = onAgentEvent ( ( evt ) => {
0 commit comments