diff --git a/src/features/team-mode/team-worktree/manager.test.ts b/src/features/team-mode/team-worktree/manager.test.ts
index a133d179b..0f21fc2c1 100644
--- a/src/features/team-mode/team-worktree/manager.test.ts
+++ b/src/features/team-mode/team-worktree/manager.test.ts
@@ -1,12 +1,18 @@
///
-import { afterAll, beforeAll, describe, expect, mock, test } from "bun:test"
+import { afterAll, afterEach, beforeAll, describe, expect, mock, test } from "bun:test"
import { randomUUID } from "node:crypto"
import fs from "node:fs/promises"
import { tmpdir } from "node:os"
import path from "node:path"
-import { GitUnavailableError, createWorktree, setGitCommandRunnerForTests, validateWorktreeSpec } from "./manager"
+import {
+ GitUnavailableError,
+ createWorktree,
+ resetGitCommandRunnerForTests,
+ setGitCommandRunnerForTests,
+ validateWorktreeSpec,
+} from "./manager"
import { removeWorktree } from "./cleanup"
const temporaryDirectories: string[] = []
@@ -34,6 +40,10 @@ afterAll(async () => {
mock.restore()
})
+afterEach(() => {
+ resetGitCommandRunnerForTests()
+})
+
describe("team-worktree manager", () => {
test("given tmp git repo when createWorktree then registers detached worktree", async () => {
// given
@@ -81,13 +91,6 @@ describe("team-worktree manager", () => {
// then
await expect(create).rejects.toBeInstanceOf(GitUnavailableError)
- setGitCommandRunnerForTests(async (args) => {
- if (args[0] === "--version") {
- return { code: 0, stderr: "" }
- }
-
- return { code: 0, stderr: "" }
- })
})
test("given created worktree when removeWorktree then directory disappears", async () => {
diff --git a/src/features/team-mode/team-worktree/manager.ts b/src/features/team-mode/team-worktree/manager.ts
index 0f01bed71..fe07c3fc7 100644
--- a/src/features/team-mode/team-worktree/manager.ts
+++ b/src/features/team-mode/team-worktree/manager.ts
@@ -28,6 +28,10 @@ export function setGitCommandRunnerForTests(runner: typeof runGit): void {
gitCommandRunner = runner
}
+export function resetGitCommandRunnerForTests(): void {
+ gitCommandRunner = runGit
+}
+
export async function isGitAvailable(): Promise {
const result = await gitCommandRunner(["--version"])
return result.code === 0