@@ -319,13 +319,15 @@ describe("runEmbeddedPiAgent", () => {
|
319 | 319 | enqueue: immediateEnqueue, |
320 | 320 | }); |
321 | 321 | |
322 | | -expect(resolveModelAsyncMock).toHaveBeenCalledWith( |
323 | | -"openrouter", |
324 | | -"openrouter/auto", |
325 | | -agentDir, |
326 | | -cfg, |
327 | | -expect.objectContaining({ skipPiDiscovery: true }), |
328 | | -); |
| 322 | +const resolveModelCall = (resolveModelAsyncMock as unknown as { mock: { calls: unknown[][] } }) |
| 323 | +.mock.calls[0]; |
| 324 | +expect(resolveModelCall?.[0]).toBe("openrouter"); |
| 325 | +expect(resolveModelCall?.[1]).toBe("openrouter/auto"); |
| 326 | +expect(resolveModelCall?.[2]).toBe(agentDir); |
| 327 | +expect(resolveModelCall?.[3]).toBe(cfg); |
| 328 | +expect( |
| 329 | +(resolveModelCall?.[4] as { skipPiDiscovery?: boolean } | undefined)?.skipPiDiscovery, |
| 330 | +).toBe(true); |
329 | 331 | expect(ensureOpenClawModelsJsonMock).not.toHaveBeenCalled(); |
330 | 332 | }); |
331 | 333 | |
@@ -561,7 +563,7 @@ describe("runEmbeddedPiAgent", () => {
|
561 | 563 | }); |
562 | 564 | |
563 | 565 | expect(runEmbeddedAttemptMock).toHaveBeenCalledTimes(2); |
564 | | -expect(result.payloads?.[0]).toMatchObject({ text: "ok" }); |
| 566 | +expect(result.payloads?.[0]?.text).toBe("ok"); |
565 | 567 | expect(disposeSessionMcpRuntimeMock).toHaveBeenCalledTimes(1); |
566 | 568 | expect(disposeSessionMcpRuntimeMock).toHaveBeenCalledWith("session:test"); |
567 | 569 | }); |
@@ -616,7 +618,7 @@ describe("runEmbeddedPiAgent", () => {
|
616 | 618 | }); |
617 | 619 | |
618 | 620 | expect(runEmbeddedAttemptMock).toHaveBeenCalledTimes(2); |
619 | | -expect(result.payloads?.[0]).toMatchObject({ text: "done" }); |
| 621 | +expect(result.payloads?.[0]?.text).toBe("done"); |
620 | 622 | }); |
621 | 623 | |
622 | 624 | it("handles prompt error paths without dropping user state", async () => { |
|
