test: trim more helper partial mocks

This commit is contained in:
Peter Steinberger
2026-04-03 21:37:51 +01:00
parent 21e53aea9e
commit f336f0b83c
2 changed files with 6 additions and 4 deletions

View File

@@ -275,8 +275,8 @@ vi.mock("../system-prompt.js", () => ({
createSystemPromptOverride: (prompt: string) => () => prompt,
}));
vi.mock("../extra-params.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../extra-params.js")>();
vi.mock("../extra-params.js", async () => {
const actual = await vi.importActual<typeof import("../extra-params.js")>("../extra-params.js");
return {
...actual,
applyExtraParamsToAgent: () => ({ effectiveExtraParams: {} }),

View File

@@ -171,8 +171,10 @@ export function installFreshDirectiveBehaviorReplyMocks(params?: {
loadModelCatalog: loadModelCatalogMock,
}));
if (params?.runPreparedReply || params?.onActualRunPreparedReply) {
vi.doMock("./reply/get-reply-run.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("./reply/get-reply-run.js")>();
vi.doMock("./reply/get-reply-run.js", async () => {
const actual = await vi.importActual<typeof import("./reply/get-reply-run.js")>(
"./reply/get-reply-run.js",
);
params.onActualRunPreparedReply?.(actual.runPreparedReply);
return {
...actual,