type AgentIdentityGetHandlerArgs = Parameters<(typeof agentHandlers)["agent.identity.get"]>[0];

type AgentIdentityGetParams = AgentIdentityGetHandlerArgs["params"];

const realSetTimeout = globalThis.setTimeout.bind(globalThis);

let dateOnlyFakeClockActive = false;

function waitForRealTimer(ms: number) {

return new Promise<void>((resolve) => realSetTimeout(resolve, ms));

}

async function waitForAssertion(assertion: () => void, timeoutMs = 2_000, stepMs = 5) {

vi.useFakeTimers();

try {

let lastError: unknown;

for (let elapsed = 0; elapsed <= timeoutMs; elapsed += stepMs) {

try {

assertion();

return;

} catch (error) {

lastError = error;

}

await Promise.resolve();

let lastError: unknown;

for (let elapsed = 0; elapsed <= timeoutMs; elapsed += stepMs) {

try {

assertion();

return;

} catch (error) {

lastError = error;

}

await Promise.resolve();

if (vi.isFakeTimers() && !dateOnlyFakeClockActive) {

await vi.advanceTimersByTimeAsync(stepMs);

} else {

await waitForRealTimer(stepMs);

}

throw lastError ?? new Error("assertion did not pass in time");

} finally {

vi.useRealTimers();

}

throw lastError ?? new Error("assertion did not pass in time");

}

async function flushScheduledDispatchStep() {

await Promise.resolve();

if (vi.isFakeTimers()) {

if (vi.isFakeTimers() && !dateOnlyFakeClockActive) {

await vi.runOnlyPendingTimersAsync();

} else {

await new Promise<void>((resolve) => setTimeout(resolve, 15));

await waitForRealTimer(15);

}

await Promise.resolve();

}

function setupNewYorkTimeConfig(isoDate: string) {

vi.useFakeTimers({ toFake: ["Date"] });

dateOnlyFakeClockActive = true;

vi.setSystemTime(new Date(isoDate)); // Wed Jan 28, 8:30 PM EST

mocks.agentCommand.mockClear();

mocks.loadConfigReturn = {

function resetTimeConfig() {

mocks.loadConfigReturn = {};

dateOnlyFakeClockActive = false;

vi.useRealTimers();

}

mocks.resolveBareResetBootstrapFileAccess.mockReset().mockReturnValue(true);

mocks.listAgentIds.mockReset().mockReturnValue(["main"]);

mocks.resolveSendPolicy.mockReset().mockReturnValue("allow");

dateOnlyFakeClockActive = false;

vi.useRealTimers();

});

it("preserves ACP metadata from the current stored session entry", async () => {