});

};

function firstMockCall(mock: { mock: { calls: unknown[][] } }, label: string): unknown[] {

const call = mock.mock.calls[0];

if (!call) {

throw new Error(`Expected ${label} to be called`);

}

return call;

}

function firstRunEmbeddedAttemptParams(): { sessionKey?: string } {

return firstMockCall(runEmbeddedAttemptMock, "embedded attempt")[0] as { sessionKey?: string };

}

describe("runEmbeddedPiAgent", () => {

it("skips models.json generation when dynamic model resolution succeeds", async () => {

const sessionFile = nextSessionFile();

enqueue: immediateEnqueue,

});

const resolveModelCall = (

resolveModelAsyncMock as unknown as { mock: { calls: unknown[][] } }

).mock.calls.at(0);

const resolveModelCall = firstMockCall(resolveModelAsyncMock, "model resolution");

expect(resolveModelCall?.[0]).toBe("openrouter");

expect(resolveModelCall?.[1]).toBe("openrouter/auto");

expect(resolveModelCall?.[2]).toBe(agentDir);

sessionId: "resume-123",

agentId: undefined,

});

const firstCall = runEmbeddedAttemptMock.mock.calls.at(0)?.[0] as { sessionKey?: string };

expect(firstCall.sessionKey).toBe("agent:test:resolved");

expect(firstRunEmbeddedAttemptParams().sessionKey).toBe("agent:test:resolved");

});

it("drops whitespace-only session keys when backfill cannot resolve a session key", async () => {

sessionId: "resume-124",

agentId: undefined,

});

const firstCall = runEmbeddedAttemptMock.mock.calls.at(0)?.[0] as { sessionKey?: string };

expect(firstCall.sessionKey).toBeUndefined();

expect(firstRunEmbeddedAttemptParams().sessionKey).toBeUndefined();

});

it("logs when embedded session-key backfill resolution fails", async () => {