From 855bd08847f34fd5cc85405ae2fbd5cf7e8029bf Mon Sep 17 00:00:00 2001 From: elkaix Date: Wed, 12 Aug 2026 04:03:34 -0400 Subject: [PATCH] test: sequence the advisor mock with ordered handlers --- .../test/session/session-advisor.test.ts | 22 +++++++++++-------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/packages/agent-core/test/session/session-advisor.test.ts b/packages/agent-core/test/session/session-advisor.test.ts index f726a63a..34178f34 100644 --- a/packages/agent-core/test/session/session-advisor.test.ts +++ b/packages/agent-core/test/session/session-advisor.test.ts @@ -77,15 +77,19 @@ describe('SessionAdvisor', () => { const reviewGate = createDeferred(); const activeTurnGate = createDeferred(); const generate = fixture.main.rawGenerate; - let generateCall = 0; - vi.spyOn(fixture.main, 'rawGenerate').mockImplementation(async (...args) => { - generateCall += 1; - const currentCall = generateCall; - const result = await generate(...args); - if (currentCall === 2) await reviewGate.promise; - if (currentCall === 3) await activeTurnGate.promise; - return result; - }); + vi.spyOn(fixture.main, 'rawGenerate') + .mockImplementationOnce(generate) + .mockImplementationOnce(async (...args) => { + const result = await generate(...args); + await reviewGate.promise; + return result; + }) + .mockImplementationOnce(async (...args) => { + const result = await generate(...args); + await activeTurnGate.promise; + return result; + }) + .mockImplementation(generate); const steer = vi.spyOn(fixture.main.turn, 'steer').mockReturnValue(null); queueReview(fixture.scripted, 'Check the active turn.', 'concern');