From a56459d23a2805a4b0bc8d003ba124cac0f4a48a Mon Sep 17 00:00:00 2001
From: Jinwoo Hong <73622457+Jinwoo-H@users.noreply.github.com>
Date: Sun, 31 May 2026 23:51:34 -0700
Subject: [PATCH] Fix git status polling test mock

---
 .../src/components/right-sidebar/useGitStatusPolling.test.ts   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/renderer/src/components/right-sidebar/useGitStatusPolling.test.ts b/src/renderer/src/components/right-sidebar/useGitStatusPolling.test.ts
index bbf800f9c60..0c72360bf8a 100644
--- a/src/renderer/src/components/right-sidebar/useGitStatusPolling.test.ts
+++ b/src/renderer/src/components/right-sidebar/useGitStatusPolling.test.ts
@@ -73,6 +73,8 @@ async function usePollingOnce(
   vi.doMock('@/store/selectors', () => ({
     useActiveWorktree: () =>
       options.pushTarget ? { ...worktree, pushTarget: options.pushTarget } : worktree,
+    useWorktreeById: () =>
+      options.pushTarget ? { ...worktree, pushTarget: options.pushTarget } : worktree,
     useAllWorktrees: () => [worktree],
     useRepoById: () => mockedRepo,
     useRepoMap: () => new Map([[mockedRepo.id, mockedRepo]])
@@ -233,6 +235,7 @@ describe('useGitStatusPolling', () => {
     }))
     vi.doMock('@/store/selectors', () => ({
       useActiveWorktree: () => worktree,
+      useWorktreeById: () => worktree,
       useAllWorktrees: () => [worktree],
       useRepoById: () => repo,
       useRepoMap: () => new Map([[repo.id, repo]])