From 59199731c381dfc6a5e329fa199eae8993ce01f4 Mon Sep 17 00:00:00 2001
From: William Valentin <william.valentin.info@gmail.com>
Date: Tue, 17 Feb 2026 16:06:03 -0800
Subject: [PATCH] fix(gateway): align node push token typing and router test
 mock

---
 src/gateway/server.test.ts | 6 ++++++
 src/gateway/server.ts      | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/gateway/server.test.ts b/src/gateway/server.test.ts
index 6eb6a46..3229d77 100644
--- a/src/gateway/server.test.ts
+++ b/src/gateway/server.test.ts
@@ -43,6 +43,12 @@ const mockModelClient = {
     stopReason: 'end_turn',
     usage: { inputTokens: 10, outputTokens: 5 },
   })),
+  setTier: vi.fn(() => true),
+  getTier: vi.fn(() => 'default' as const),
+  getLabel: vi.fn((tier: string) => tier),
+  setTierStrict: vi.fn(),
+  addTierChangeListener: vi.fn(),
+  removeTierChangeListener: vi.fn(),
 };
 
 const mockToolRegistry = {
diff --git a/src/gateway/server.ts b/src/gateway/server.ts
index eb5d6cf..c373eae 100644
--- a/src/gateway/server.ts
+++ b/src/gateway/server.ts
@@ -431,7 +431,7 @@ export class GatewayServer {
           return;
         }
         const provider = pushToken.provider === 'fcm' ? 'fcm' : 'apns';
-        const normalizedPushToken = {
+        const normalizedPushToken: NonNullable<NodeConnectionState['pushToken']> = {
           provider,
           token: pushToken.token,
           topic: pushToken.topic,