diff --git a/src/gateway/gateway-connection.test-mocks.ts b/src/gateway/gateway-connection.test-mocks.ts index 20b0009f075..966ec8254c6 100644 --- a/src/gateway/gateway-connection.test-mocks.ts +++ b/src/gateway/gateway-connection.test-mocks.ts @@ -1,9 +1,11 @@ import { vi } from "vitest"; -export const loadConfigMock = vi.fn(); -export const resolveGatewayPortMock = vi.fn(); -export const pickPrimaryTailnetIPv4Mock = vi.fn(); -export const pickPrimaryLanIPv4Mock = vi.fn(); +type TestMock = ReturnType; + +export const loadConfigMock: TestMock = vi.fn(); +export const resolveGatewayPortMock: TestMock = vi.fn(); +export const pickPrimaryTailnetIPv4Mock: TestMock = vi.fn(); +export const pickPrimaryLanIPv4Mock: TestMock = vi.fn(); vi.mock("../config/config.js", async (importOriginal) => { const actual = await importOriginal();