test(plugins): simplify typing pulse mock helper

This commit is contained in:
Ayaan Zaidi
2026-03-28 10:32:02 +05:30
parent 3a341355bf
commit 16f8616d9d

View File

@@ -7,11 +7,14 @@ export function expectTypingPulseCount(pulse: { mock: { calls: unknown[] } }, ex
export function createPulseWithBackgroundFailure<
TPulse extends (...args: never[]) => Promise<unknown>,
>() {
const pulse = vi.fn(
async (..._args: never[]) => undefined as Awaited<ReturnType<TPulse>>,
) as ReturnType<typeof vi.fn<TPulse>>;
pulse.mockResolvedValueOnce(undefined as Awaited<ReturnType<TPulse>>);
pulse.mockRejectedValueOnce(new Error("boom"));
let callCount = 0;
const pulse = vi.fn((() => {
callCount += 1;
if (callCount === 2) {
return Promise.reject(new Error("boom"));
}
return Promise.resolve(undefined);
}) as TPulse);
return pulse;
}