import { vi } from "vitest";

import type * as SessionWriteLockModule from "../agents/session-write-lock.js";

type SessionWriteLockModuleShape = typeof SessionWriteLockModule;

acquireSessionWriteLock,

};

}

export function resetModulesWithSessionWriteLockDoMock(

modulePath: string,

acquireSessionWriteLock: SessionWriteLockModuleShape["acquireSessionWriteLock"],

): void {

vi.resetModules();

vi.doMock(modulePath, () =>

buildSessionWriteLockModuleMock(

() => vi.importActual<SessionWriteLockModuleShape>(modulePath),

acquireSessionWriteLock,

),

);

}