diff --git a/src/discord/monitor/provider.lifecycle.test.ts b/src/discord/monitor/provider.lifecycle.test.ts index 3e9cc96e9..c3293c1a7 100644 --- a/src/discord/monitor/provider.lifecycle.test.ts +++ b/src/discord/monitor/provider.lifecycle.test.ts @@ -2,6 +2,9 @@ import { EventEmitter } from "node:events"; import type { Client } from "@buape/carbon"; import { beforeEach, describe, expect, it, vi } from "vitest"; import type { RuntimeEnv } from "../../runtime.js"; +import type { waitForDiscordGatewayStop } from "../monitor.gateway.js"; + +type WaitForDiscordGatewayStopParams = Parameters[0]; type WaitForDiscordGatewayStop = typeof import("../monitor.gateway.js").waitForDiscordGatewayStop; @@ -333,7 +336,7 @@ describe("runDiscordGatewayLifecycle", () => { }; getDiscordGatewayEmitterMock.mockReturnValueOnce(emitter); waitForDiscordGatewayStopMock.mockImplementationOnce( - (waitParams: { registerForceStop?: (stop: (err: unknown) => void) => void }) => + (waitParams: WaitForDiscordGatewayStopParams) => new Promise((_resolve, reject) => { waitParams.registerForceStop?.((err) => reject(err)); }), @@ -366,7 +369,7 @@ describe("runDiscordGatewayLifecycle", () => { getDiscordGatewayEmitterMock.mockReturnValueOnce(emitter); let resolveWait: (() => void) | undefined; waitForDiscordGatewayStopMock.mockImplementationOnce( - (waitParams: { registerForceStop?: (stop: (err: unknown) => void) => void }) => + (waitParams: WaitForDiscordGatewayStopParams) => new Promise((resolve, reject) => { resolveWait = resolve; waitParams.registerForceStop?.((err) => reject(err));