From 0edf8f92b04262ca7c0daf3dc9664553773d7c3d Mon Sep 17 00:00:00 2001
From: Brennan Benson <79079362+brennanb2025@users.noreply.github.com>
Date: Sun, 16 Aug 2026 20:33:06 -0700
Subject: [PATCH] test(github): expose SSH probe generation in mocks

---
 src/main/github/client-pr-local-runtime.test.ts | 1 +
 src/main/github/client-test-mocks.ts            | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/main/github/client-pr-local-runtime.test.ts b/src/main/github/client-pr-local-runtime.test.ts
index 543d6f239f2..0633b137b1c 100644
--- a/src/main/github/client-pr-local-runtime.test.ts
+++ b/src/main/github/client-pr-local-runtime.test.ts
@@ -85,6 +85,7 @@ vi.mock('./github-enterprise-repository', () => ({
 }))
 
 vi.mock('../providers/ssh-git-dispatch', () => ({
+  getSshGitProviderGeneration: vi.fn(() => 0),
   getSshGitProvider: vi.fn()
 }))
 
diff --git a/src/main/github/client-test-mocks.ts b/src/main/github/client-test-mocks.ts
index c27af20defe..15fe0f36214 100644
--- a/src/main/github/client-test-mocks.ts
+++ b/src/main/github/client-test-mocks.ts
@@ -132,12 +132,14 @@ export function gitRunnerModuleMock(mocks: GitHubClientMocks): GitRunnerModuleMo
 }
 
 export type SshGitDispatchModuleMock = {
+  getSshGitProviderGeneration: Mock
   getSshGitProvider: Mock
   SSH_GIT_PROVIDER_UNAVAILABLE_MESSAGE: string
 }
 
 export function sshGitDispatchModuleMock(mocks: GitHubClientMocks): SshGitDispatchModuleMock {
   return {
+    getSshGitProviderGeneration: vi.fn(() => 0),
     getSshGitProvider: mocks.getSshGitProviderMock,
     SSH_GIT_PROVIDER_UNAVAILABLE_MESSAGE: 'Remote connection dropped.'
   }