From 86fcca2352c0df38715be3e41d388a039ddb40e8 Mon Sep 17 00:00:00 2001
From: Ayaan Zaidi <zaidi@uplause.io>
Date: Mon, 23 Feb 2026 11:47:27 +0530
Subject: [PATCH] fix(gateway): annotate connection test mocks

---
 src/gateway/gateway-connection.test-mocks.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

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<typeof vi.fn>;
+
+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<typeof import("../config/config.js")>();