@@ -96,20 +96,19 @@ async function runFastModeCase(params: {
|
96 | 96 | |
97 | 97 | expect(result.status).toBe("ok"); |
98 | 98 | expect(runEmbeddedPiAgentMock).toHaveBeenCalledOnce(); |
99 | | -expect(runEmbeddedPiAgentMock.mock.calls[0][0]).toMatchObject({ |
100 | | -provider: "openai", |
101 | | -model: EXPECTED_OPENAI_MODEL, |
102 | | -fastMode: params.expectedFastMode, |
103 | | -cleanupBundleMcpOnRunEnd: params.expectedCleanupBundleMcpOnRunEnd ?? true, |
104 | | -allowGatewaySubagentBinding: true, |
105 | | -}); |
| 99 | +const [embeddedRunParams] = runEmbeddedPiAgentMock.mock.calls[0]; |
| 100 | +expect(embeddedRunParams.provider).toBe("openai"); |
| 101 | +expect(embeddedRunParams.model).toBe(EXPECTED_OPENAI_MODEL); |
| 102 | +expect(embeddedRunParams.fastMode).toBe(params.expectedFastMode); |
| 103 | +expect(embeddedRunParams.cleanupBundleMcpOnRunEnd).toBe( |
| 104 | +params.expectedCleanupBundleMcpOnRunEnd ?? true, |
| 105 | +); |
| 106 | +expect(embeddedRunParams.allowGatewaySubagentBinding).toBe(true); |
106 | 107 | if (params.expectedRetiredSessionId) { |
107 | | -expect(retireSessionMcpRuntimeMock).toHaveBeenCalledWith( |
108 | | -expect.objectContaining({ |
109 | | -sessionId: params.expectedRetiredSessionId, |
110 | | -reason: "cron-session-rollover", |
111 | | -}), |
112 | | -); |
| 108 | +expect(retireSessionMcpRuntimeMock).toHaveBeenCalledOnce(); |
| 109 | +const [retireParams] = retireSessionMcpRuntimeMock.mock.calls[0]; |
| 110 | +expect(retireParams.sessionId).toBe(params.expectedRetiredSessionId); |
| 111 | +expect(retireParams.reason).toBe("cron-session-rollover"); |
113 | 112 | return; |
114 | 113 | } |
115 | 114 | expect(retireSessionMcpRuntimeMock).not.toHaveBeenCalled(); |
|
