From 338e7286c08c717c98ee61a94eea01e863177196 Mon Sep 17 00:00:00 2001
From: gsxdsm <gsxdsm@users.noreply.github.com>
Date: Sat, 13 Jun 2026 14:54:23 -0700
Subject: [PATCH] FN-6395: align viewport hook mocks in dashboard tests

Update dashboard test doubles so mobile viewport helpers match the hook module API.

- Add getViewportMode and isMobileViewport exports to affected useViewportMode mocks.
- Keep mocked mobile detection tied to each test's configurable viewport mode.
- Normalize inline viewport mocks across dashboard component test suites.

Files changed:
 .../components/__tests__/AgentsView.orgchart.test.tsx   |  6 +++++-
 .../app/components/__tests__/AgentsView.test.tsx        |  2 ++
 .../app/components/__tests__/ChatView.rooms.test.tsx    |  1 +
 .../app/components/__tests__/GitManagerModal.test.tsx   |  2 ++
 .../app/components/__tests__/MailboxView.test.tsx       | 13 +++++++++----
 .../__tests__/MilestoneSliceInterviewModal.test.tsx     |  2 ++
 .../__tests__/MissionManager.swipe-back.test.tsx        |  2 ++
 .../__tests__/NewTaskModal.shared-cache.test.tsx        | 17 ++++++++++++++---
 .../app/components/__tests__/NewTaskModal.test.tsx      |  2 ++
 .../__tests__/PlanningModeModal.favorites.test.tsx      |  2 ++
 .../__tests__/PlanningModeModal.questions.test.tsx      |  2 ++
 .../__tests__/PlanningModeModal.swipe-back.test.tsx     |  2 ++
 .../PlanningModeModal.ui-interactions.test.tsx          |  4 ++++
 .../components/__tests__/QuickChatFAB.autosize.test.tsx | 13 +++++++++----
 .../__tests__/QuickChatFAB.shared-cache.test.tsx        | 11 +++++++++--
 .../app/components/__tests__/QuickChatFAB.test.tsx      | 11 +++++++++--
 .../__tests__/SettingsModal.testMode.test.tsx           |  6 +++++-
 .../components/__tests__/SubtaskBreakdownModal.test.tsx |  2 ++
 .../app/components/__tests__/TodoModal.test.tsx         |  2 ++
 .../components/__tests__/navigation-history.test.tsx    |  3 ++-
 20 files changed, 87 insertions(+), 18 deletions(-)

Fusion-Task-Id: FN-6395
Fusion-Task-Lineage: d830938b-4d60-4e17-852d-a2d311b6d71c
---
 .../__tests__/AgentsView.orgchart.test.tsx      |  6 +++++-
 .../components/__tests__/AgentsView.test.tsx    |  2 ++
 .../__tests__/ChatView.rooms.test.tsx           |  1 +
 .../__tests__/GitManagerModal.test.tsx          |  2 ++
 .../components/__tests__/MailboxView.test.tsx   | 13 +++++++++----
 .../MilestoneSliceInterviewModal.test.tsx       |  2 ++
 .../MissionManager.swipe-back.test.tsx          |  2 ++
 .../NewTaskModal.shared-cache.test.tsx          | 17 ++++++++++++++---
 .../components/__tests__/NewTaskModal.test.tsx  |  2 ++
 .../PlanningModeModal.favorites.test.tsx        |  2 ++
 .../PlanningModeModal.questions.test.tsx        |  2 ++
 .../PlanningModeModal.swipe-back.test.tsx       |  2 ++
 .../PlanningModeModal.ui-interactions.test.tsx  |  4 ++++
 .../__tests__/QuickChatFAB.autosize.test.tsx    | 13 +++++++++----
 .../QuickChatFAB.shared-cache.test.tsx          | 11 +++++++++--
 .../components/__tests__/QuickChatFAB.test.tsx  | 11 +++++++++--
 .../__tests__/SettingsModal.testMode.test.tsx   |  6 +++++-
 .../__tests__/SubtaskBreakdownModal.test.tsx    |  2 ++
 .../app/components/__tests__/TodoModal.test.tsx |  2 ++
 .../__tests__/navigation-history.test.tsx       |  3 ++-
 20 files changed, 87 insertions(+), 18 deletions(-)

diff --git a/packages/dashboard/app/components/__tests__/AgentsView.orgchart.test.tsx b/packages/dashboard/app/components/__tests__/AgentsView.orgchart.test.tsx
index bb22a958a9..4031a41bf7 100644
--- a/packages/dashboard/app/components/__tests__/AgentsView.orgchart.test.tsx
+++ b/packages/dashboard/app/components/__tests__/AgentsView.orgchart.test.tsx
@@ -5,7 +5,11 @@ import * as apiModule from "../../api";
 
 const mockViewportMode = vi.fn<() => "mobile" | "tablet" | "desktop">(() => "desktop");
 vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)", useViewportMode: () => mockViewportMode() }));
+  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockViewportMode(),
+  isMobileViewport: () => mockViewportMode() === "mobile",
+  useViewportMode: () => mockViewportMode(),
+}));
 vi.mock("../../hooks/useConfirm", () => ({ useConfirm: () => ({ confirm: vi.fn().mockResolvedValue(true) }) }));
 vi.mock("../AgentDetailView", () => ({ AgentDetailView: () => null, relativeTime: () => "now" }));
 
diff --git a/packages/dashboard/app/components/__tests__/AgentsView.test.tsx b/packages/dashboard/app/components/__tests__/AgentsView.test.tsx
index bed6c69404..1cbb942556 100644
--- a/packages/dashboard/app/components/__tests__/AgentsView.test.tsx
+++ b/packages/dashboard/app/components/__tests__/AgentsView.test.tsx
@@ -77,6 +77,8 @@ const mockViewportMode = vi.fn<() => "mobile" | "tablet" | "desktop">(() => "des
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockViewportMode(),
+  isMobileViewport: () => mockViewportMode() === "mobile",
   useViewportMode: () => mockViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/ChatView.rooms.test.tsx b/packages/dashboard/app/components/__tests__/ChatView.rooms.test.tsx
index 88dc9e37a7..958aeacae8 100644
--- a/packages/dashboard/app/components/__tests__/ChatView.rooms.test.tsx
+++ b/packages/dashboard/app/components/__tests__/ChatView.rooms.test.tsx
@@ -12,6 +12,7 @@ vi.mock("../../hooks/useChat");
 vi.mock("../../hooks/useMobileScrollLock", () => ({
   useMobileScrollLock: vi.fn(),
   useMobileKeyboardViewportLock: vi.fn(),
+  useMobileViewportRestoreReset: vi.fn(),
   isIOS: () => true,
   _resetLockState: vi.fn(),
 }));
diff --git a/packages/dashboard/app/components/__tests__/GitManagerModal.test.tsx b/packages/dashboard/app/components/__tests__/GitManagerModal.test.tsx
index 1a1a4de7c4..a17e4b4829 100644
--- a/packages/dashboard/app/components/__tests__/GitManagerModal.test.tsx
+++ b/packages/dashboard/app/components/__tests__/GitManagerModal.test.tsx
@@ -31,6 +31,8 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
 
 vi.mock("../../hooks/useMobileScrollLock", () => ({
   useMobileScrollLock: vi.fn(),
+  useMobileKeyboardViewportLock: vi.fn(),
+  useMobileViewportRestoreReset: vi.fn(),
 }));
 
 // Mock the API module with all functions
diff --git a/packages/dashboard/app/components/__tests__/MailboxView.test.tsx b/packages/dashboard/app/components/__tests__/MailboxView.test.tsx
index 2ece792422..fa466cae69 100644
--- a/packages/dashboard/app/components/__tests__/MailboxView.test.tsx
+++ b/packages/dashboard/app/components/__tests__/MailboxView.test.tsx
@@ -27,10 +27,15 @@ vi.mock("../../api", () => ({
   decideApproval: vi.fn(),
 }));
 
-vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
-  useViewportMode: vi.fn(),
-}));
+vi.mock("../../hooks/useViewportMode", () => {
+  const useViewportMode = vi.fn();
+  return {
+    MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+    getViewportMode: () => useViewportMode(),
+    isMobileViewport: () => useViewportMode() === "mobile",
+    useViewportMode,
+  };
+});
 
 vi.mock("../../hooks/useMobileKeyboard", () => ({
   useMobileKeyboard: vi.fn(),
diff --git a/packages/dashboard/app/components/__tests__/MilestoneSliceInterviewModal.test.tsx b/packages/dashboard/app/components/__tests__/MilestoneSliceInterviewModal.test.tsx
index 095766dd57..0678e30db3 100644
--- a/packages/dashboard/app/components/__tests__/MilestoneSliceInterviewModal.test.tsx
+++ b/packages/dashboard/app/components/__tests__/MilestoneSliceInterviewModal.test.tsx
@@ -66,6 +66,8 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => "mobile",
+  isMobileViewport: () => true,
   useViewportMode: () => "mobile",
 }));
 vi.mock("lucide-react", () => ({
diff --git a/packages/dashboard/app/components/__tests__/MissionManager.swipe-back.test.tsx b/packages/dashboard/app/components/__tests__/MissionManager.swipe-back.test.tsx
index 4d0105053a..3884e8f5f1 100644
--- a/packages/dashboard/app/components/__tests__/MissionManager.swipe-back.test.tsx
+++ b/packages/dashboard/app/components/__tests__/MissionManager.swipe-back.test.tsx
@@ -18,6 +18,8 @@ const mockSubscribeSse = vi.fn(() => vi.fn());
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockViewportMode(),
+  isMobileViewport: () => mockViewportMode() === "mobile",
   useViewportMode: () => mockViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/NewTaskModal.shared-cache.test.tsx b/packages/dashboard/app/components/__tests__/NewTaskModal.shared-cache.test.tsx
index a559e152e4..86eaa1ac5d 100644
--- a/packages/dashboard/app/components/__tests__/NewTaskModal.shared-cache.test.tsx
+++ b/packages/dashboard/app/components/__tests__/NewTaskModal.shared-cache.test.tsx
@@ -18,10 +18,21 @@ vi.mock("../../api", async (importOriginal) => {
 vi.mock("../../hooks/useSetupReadiness", () => ({ useSetupReadiness: vi.fn(() => ({ hasAiProvider: true, hasGithub: true, loading: false })) }));
 vi.mock("../../hooks/useConfirm", () => ({ useConfirm: vi.fn(() => ({ confirm: vi.fn().mockResolvedValue(true) })) }));
 vi.mock("../../hooks/useMobileKeyboard", () => ({ useMobileKeyboard: vi.fn(() => ({ keyboardOverlap: 0, viewportHeight: null, viewportOffsetTop: 0, keyboardOpen: false })) }));
-vi.mock("../../hooks/useMobileScrollLock", () => ({ useMobileScrollLock: vi.fn() }));
+vi.mock("../../hooks/useMobileScrollLock", () => ({
+  useMobileScrollLock: vi.fn(),
+  useMobileKeyboardViewportLock: vi.fn(),
+  useMobileViewportRestoreReset: vi.fn(),
+}));
 vi.mock("../../hooks/useNodes", () => ({ useNodes: vi.fn(() => ({ nodes: [] })) }));
-vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)", useViewportMode: vi.fn(() => "desktop") }));
+vi.mock("../../hooks/useViewportMode", () => {
+  const useViewportMode = vi.fn(() => "desktop");
+  return {
+    MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+    getViewportMode: () => useViewportMode(),
+    isMobileViewport: () => useViewportMode() === "mobile",
+    useViewportMode,
+  };
+});
 
 function deferred<T>() {
   let resolve!: (value: T) => void;
diff --git a/packages/dashboard/app/components/__tests__/NewTaskModal.test.tsx b/packages/dashboard/app/components/__tests__/NewTaskModal.test.tsx
index 76567e6fd6..47bf14cc4b 100644
--- a/packages/dashboard/app/components/__tests__/NewTaskModal.test.tsx
+++ b/packages/dashboard/app/components/__tests__/NewTaskModal.test.tsx
@@ -53,6 +53,8 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => "mobile",
+  isMobileViewport: () => true,
   useViewportMode: () => "mobile",
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/PlanningModeModal.favorites.test.tsx b/packages/dashboard/app/components/__tests__/PlanningModeModal.favorites.test.tsx
index d37a76aa47..25605fc8b8 100644
--- a/packages/dashboard/app/components/__tests__/PlanningModeModal.favorites.test.tsx
+++ b/packages/dashboard/app/components/__tests__/PlanningModeModal.favorites.test.tsx
@@ -101,6 +101,8 @@ vi.mock("../../hooks/useConfirm", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockUseViewportMode(),
+  isMobileViewport: () => mockUseViewportMode() === "mobile",
   useViewportMode: () => mockUseViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/PlanningModeModal.questions.test.tsx b/packages/dashboard/app/components/__tests__/PlanningModeModal.questions.test.tsx
index 7c74323d58..f732a922e2 100644
--- a/packages/dashboard/app/components/__tests__/PlanningModeModal.questions.test.tsx
+++ b/packages/dashboard/app/components/__tests__/PlanningModeModal.questions.test.tsx
@@ -102,6 +102,8 @@ vi.mock("../../hooks/useConfirm", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockUseViewportMode(),
+  isMobileViewport: () => mockUseViewportMode() === "mobile",
   useViewportMode: () => mockUseViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/PlanningModeModal.swipe-back.test.tsx b/packages/dashboard/app/components/__tests__/PlanningModeModal.swipe-back.test.tsx
index 89458b5a1d..d041063bad 100644
--- a/packages/dashboard/app/components/__tests__/PlanningModeModal.swipe-back.test.tsx
+++ b/packages/dashboard/app/components/__tests__/PlanningModeModal.swipe-back.test.tsx
@@ -12,6 +12,8 @@ const mockSubscribeSse = vi.fn(() => vi.fn());
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockViewportMode(),
+  isMobileViewport: () => mockViewportMode() === "mobile",
   useViewportMode: () => mockViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/PlanningModeModal.ui-interactions.test.tsx b/packages/dashboard/app/components/__tests__/PlanningModeModal.ui-interactions.test.tsx
index ec9b0b29f7..ecd904ed31 100644
--- a/packages/dashboard/app/components/__tests__/PlanningModeModal.ui-interactions.test.tsx
+++ b/packages/dashboard/app/components/__tests__/PlanningModeModal.ui-interactions.test.tsx
@@ -68,6 +68,7 @@ import {
 } from "./PlanningModeModal.test-helpers";
 
 vi.mock("../../api", () => ({
+  api: vi.fn().mockResolvedValue({ sessions: [] }),
   startPlanning: (...args: any[]) => mockStartPlanning(...args),
   startPlanningStreaming: (...args: any[]) => mockStartPlanningStreaming(...args),
   createPlanningDraft: (...args: any[]) => mockCreatePlanningDraft(...args),
@@ -99,6 +100,7 @@ vi.mock("../../api", () => ({
   fetchGlobalSettings: vi.fn().mockResolvedValue({}),
   fetchModels: (...args: any[]) => mockFetchModels(...args),
   fetchWorkflowSteps: vi.fn().mockResolvedValue([]),
+  fetchBoardWorkflows: vi.fn().mockResolvedValue({ flagEnabled: false, defaultWorkflowId: "", workflows: [], taskWorkflowIds: {} }),
   refineText: vi.fn(),
   getRefineErrorMessage: vi.fn((err: any) => err?.message || "Failed to refine"),
   updateGlobalSettings: vi.fn().mockResolvedValue({}),
@@ -112,6 +114,8 @@ vi.mock("../../hooks/useConfirm", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockUseViewportMode(),
+  isMobileViewport: () => mockUseViewportMode() === "mobile",
   useViewportMode: () => mockUseViewportMode(),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/QuickChatFAB.autosize.test.tsx b/packages/dashboard/app/components/__tests__/QuickChatFAB.autosize.test.tsx
index 4a261c1cee..4a5b95cfaf 100644
--- a/packages/dashboard/app/components/__tests__/QuickChatFAB.autosize.test.tsx
+++ b/packages/dashboard/app/components/__tests__/QuickChatFAB.autosize.test.tsx
@@ -84,10 +84,15 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
   })),
 }));
 
-vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
-  useViewportMode: vi.fn(() => "desktop"),
-}));
+vi.mock("../../hooks/useViewportMode", () => {
+  const useViewportMode = vi.fn(() => "desktop");
+  return {
+    MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+    getViewportMode: () => useViewportMode(),
+    isMobileViewport: () => useViewportMode() === "mobile",
+    useViewportMode,
+  };
+});
 
 vi.mock("react-markdown", () => ({
   default: ({ children }: { children: string }) => children,
diff --git a/packages/dashboard/app/components/__tests__/QuickChatFAB.shared-cache.test.tsx b/packages/dashboard/app/components/__tests__/QuickChatFAB.shared-cache.test.tsx
index 4d368b6923..fad8095815 100644
--- a/packages/dashboard/app/components/__tests__/QuickChatFAB.shared-cache.test.tsx
+++ b/packages/dashboard/app/components/__tests__/QuickChatFAB.shared-cache.test.tsx
@@ -46,8 +46,15 @@ vi.mock("../../hooks/useQuickChat", () => ({
 }));
 vi.mock("../../hooks/useFileMention", () => ({ useFileMention: vi.fn(() => ({ mentionActive: false, detectMention: vi.fn(), dismissMention: vi.fn(), handleKeyDown: vi.fn(), selectTask: vi.fn(), selectFile: vi.fn(), tasks: [], files: [], combinedItems: [], loading: false, mentionQuery: "", selectedIndex: 0, setSelectedIndex: vi.fn() })) }));
 vi.mock("../../hooks/useMobileKeyboard", () => ({ useMobileKeyboard: vi.fn(() => ({ keyboardOpen: false, keyboardOverlap: 0, viewportHeight: null, viewportOffsetTop: 0 })) }));
-vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)", useViewportMode: vi.fn(() => "desktop") }));
+vi.mock("../../hooks/useViewportMode", () => {
+  const useViewportMode = vi.fn(() => "desktop");
+  return {
+    MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+    getViewportMode: () => useViewportMode(),
+    isMobileViewport: () => useViewportMode() === "mobile",
+    useViewportMode,
+  };
+});
 vi.mock("react-markdown", () => ({ default: ({ children }: { children: string }) => children }));
 
 function deferred<T>() {
diff --git a/packages/dashboard/app/components/__tests__/QuickChatFAB.test.tsx b/packages/dashboard/app/components/__tests__/QuickChatFAB.test.tsx
index 1a0b91497a..0a519daa50 100644
--- a/packages/dashboard/app/components/__tests__/QuickChatFAB.test.tsx
+++ b/packages/dashboard/app/components/__tests__/QuickChatFAB.test.tsx
@@ -29,8 +29,15 @@ vi.mock("../../api", () => ({
 }));
 
 vi.mock("../../hooks/useAgents", () => ({ useAgents: vi.fn() }));
-vi.mock("../../hooks/useViewportMode", () => ({
-  MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)", useViewportMode: vi.fn() }));
+vi.mock("../../hooks/useViewportMode", () => {
+  const useViewportMode = vi.fn();
+  return {
+    MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+    getViewportMode: () => useViewportMode(),
+    isMobileViewport: () => useViewportMode() === "mobile",
+    useViewportMode,
+  };
+});
 vi.mock("../../hooks/useMobileKeyboard", () => ({ useMobileKeyboard: vi.fn() }));
 vi.mock("../../hooks/useAppSettings", () => ({ useAppSettings: vi.fn() }));
 vi.mock("../../hooks/useChatRooms", () => ({ useChatRooms: vi.fn() }));
diff --git a/packages/dashboard/app/components/__tests__/SettingsModal.testMode.test.tsx b/packages/dashboard/app/components/__tests__/SettingsModal.testMode.test.tsx
index 6d979a2104..28a97dd41c 100644
--- a/packages/dashboard/app/components/__tests__/SettingsModal.testMode.test.tsx
+++ b/packages/dashboard/app/components/__tests__/SettingsModal.testMode.test.tsx
@@ -26,7 +26,11 @@ vi.mock("../../hooks/useViewportMode", () => ({
 vi.mock("../../hooks/useMobileKeyboard", () => ({
   useMobileKeyboard: () => ({ keyboardOverlap: 0, viewportHeight: null, viewportOffsetTop: 0, keyboardOpen: false }),
 }));
-vi.mock("../../hooks/useMobileScrollLock", () => ({ useMobileScrollLock: vi.fn() }));
+vi.mock("../../hooks/useMobileScrollLock", () => ({
+  useMobileScrollLock: vi.fn(),
+  useMobileKeyboardViewportLock: vi.fn(),
+  useMobileViewportRestoreReset: vi.fn(),
+}));
 vi.mock("../../hooks/useConfirm", () => ({ useConfirm: () => ({ confirm: vi.fn() }) }));
 vi.mock("../../hooks/useWorkspaceFileBrowser", () => ({
   useWorkspaceFileBrowser: () => ({ entries: [], currentPath: ".", setPath: vi.fn(), loading: false, error: null, refresh: vi.fn() }),
diff --git a/packages/dashboard/app/components/__tests__/SubtaskBreakdownModal.test.tsx b/packages/dashboard/app/components/__tests__/SubtaskBreakdownModal.test.tsx
index 6bc29ef31a..232edc99a0 100644
--- a/packages/dashboard/app/components/__tests__/SubtaskBreakdownModal.test.tsx
+++ b/packages/dashboard/app/components/__tests__/SubtaskBreakdownModal.test.tsx
@@ -45,6 +45,8 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => "mobile",
+  isMobileViewport: () => true,
   useViewportMode: () => "mobile",
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/TodoModal.test.tsx b/packages/dashboard/app/components/__tests__/TodoModal.test.tsx
index e53eb3d02a..932d10b2c9 100644
--- a/packages/dashboard/app/components/__tests__/TodoModal.test.tsx
+++ b/packages/dashboard/app/components/__tests__/TodoModal.test.tsx
@@ -19,6 +19,8 @@ vi.mock("../../hooks/useMobileKeyboard", () => ({
 
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockUseViewportMode(),
+  isMobileViewport: () => mockUseViewportMode() === "mobile",
   useViewportMode: (...args: unknown[]) => mockUseViewportMode(...args),
 }));
 
diff --git a/packages/dashboard/app/components/__tests__/navigation-history.test.tsx b/packages/dashboard/app/components/__tests__/navigation-history.test.tsx
index 5369b8bbd3..67307ab4f8 100644
--- a/packages/dashboard/app/components/__tests__/navigation-history.test.tsx
+++ b/packages/dashboard/app/components/__tests__/navigation-history.test.tsx
@@ -301,8 +301,9 @@ vi.mock("../../hooks/useNodes", () => ({
 const mockUseViewportMode = vi.fn(() => "desktop");
 vi.mock("../../hooks/useViewportMode", () => ({
   MOBILE_MEDIA_QUERY: "(max-width: 768px), (max-height: 480px)",
+  getViewportMode: () => mockUseViewportMode(),
+  isMobileViewport: () => mockUseViewportMode() === "mobile",
   useViewportMode: (..._args: unknown[]) => mockUseViewportMode(..._args),
-  getViewportMode: () => "desktop",
 }));
 
 const mockUseMobileKeyboard = vi.fn(() => ({