File tree Expand file tree Collapse file tree
packages/agent-core/test/session Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -77,15 +77,19 @@ describe('SessionAdvisor', () => {
7777 const reviewGate = createDeferred < void > ( ) ;
7878 const activeTurnGate = createDeferred < void > ( ) ;
7979 const generate = fixture . main . rawGenerate ;
80- let generateCall = 0 ;
81- vi . spyOn ( fixture . main , 'rawGenerate' ) . mockImplementation ( async ( ...args ) => {
82- generateCall += 1 ;
83- const currentCall = generateCall ;
84- const result = await generate ( ...args ) ;
85- if ( currentCall === 2 ) await reviewGate . promise ;
86- if ( currentCall === 3 ) await activeTurnGate . promise ;
87- return result ;
88- } ) ;
80+ vi . spyOn ( fixture . main , 'rawGenerate' )
81+ . mockImplementationOnce ( generate )
82+ . mockImplementationOnce ( async ( ...args ) => {
83+ const result = await generate ( ...args ) ;
84+ await reviewGate . promise ;
85+ return result ;
86+ } )
87+ . mockImplementationOnce ( async ( ...args ) => {
88+ const result = await generate ( ...args ) ;
89+ await activeTurnGate . promise ;
90+ return result ;
91+ } )
92+ . mockImplementation ( generate ) ;
8993 const steer = vi . spyOn ( fixture . main . turn , 'steer' ) . mockReturnValue ( null ) ;
9094 queueReview ( fixture . scripted , 'Check the active turn.' , 'concern' ) ;
9195
You can’t perform that action at this time.
0 commit comments