Просмотр исходного кода

test: add more test cases for project.test.ts (#13355)

Aiden Cline 2 месяцев назад
Родитель
Сommit
d1ee4c8dca
1 измененных файлов с 123 добавлено и 14 удалено
  1. 123 14
      packages/opencode/test/project/project.test.ts

+ 123 - 14
packages/opencode/test/project/project.test.ts

@@ -1,5 +1,5 @@
-import { describe, expect, test } from "bun:test"
-import { Project } from "../../src/project/project"
+import { describe, expect, mock, test } from "bun:test"
+import type { Project as ProjectNS } from "../../src/project/project"
 import { Log } from "../../src/util/log"
 import { Storage } from "../../src/storage/storage"
 import { $ } from "bun"
@@ -8,12 +8,78 @@ import { tmpdir } from "../fixture/fixture"
 
 Log.init({ print: false })
 
+const bunModule = await import("bun")
+type Mode = "none" | "rev-list-fail" | "top-fail" | "common-dir-fail"
+let mode: Mode = "none"
+
+function render(parts: TemplateStringsArray, vals: unknown[]) {
+  return parts.reduce((acc, part, i) => `${acc}${part}${i < vals.length ? String(vals[i]) : ""}`, "")
+}
+
+function fakeShell(output: { exitCode: number; stdout: string; stderr: string }) {
+  const result = {
+    exitCode: output.exitCode,
+    stdout: Buffer.from(output.stdout),
+    stderr: Buffer.from(output.stderr),
+    text: async () => output.stdout,
+  }
+  const shell = {
+    quiet: () => shell,
+    nothrow: () => shell,
+    cwd: () => shell,
+    env: () => shell,
+    text: async () => output.stdout,
+    then: (onfulfilled: (value: typeof result) => unknown, onrejected?: (reason: unknown) => unknown) =>
+      Promise.resolve(result).then(onfulfilled, onrejected),
+    catch: (onrejected: (reason: unknown) => unknown) => Promise.resolve(result).catch(onrejected),
+    finally: (onfinally: (() => void) | undefined | null) => Promise.resolve(result).finally(onfinally),
+  }
+  return shell
+}
+
+mock.module("bun", () => ({
+  ...bunModule,
+  $: (parts: TemplateStringsArray, ...vals: unknown[]) => {
+    const cmd = render(parts, vals).replaceAll(",", " ").replace(/\s+/g, " ").trim()
+    if (
+      mode === "rev-list-fail" &&
+      cmd.includes("git rev-list") &&
+      cmd.includes("--max-parents=0") &&
+      cmd.includes("--all")
+    ) {
+      return fakeShell({ exitCode: 128, stdout: "", stderr: "fatal" })
+    }
+    if (mode === "top-fail" && cmd.includes("git rev-parse") && cmd.includes("--show-toplevel")) {
+      return fakeShell({ exitCode: 128, stdout: "", stderr: "fatal" })
+    }
+    if (mode === "common-dir-fail" && cmd.includes("git rev-parse") && cmd.includes("--git-common-dir")) {
+      return fakeShell({ exitCode: 128, stdout: "", stderr: "fatal" })
+    }
+    return (bunModule.$ as any)(parts, ...vals)
+  },
+}))
+
+async function withMode(next: Mode, run: () => Promise<void>) {
+  const prev = mode
+  mode = next
+  try {
+    await run()
+  } finally {
+    mode = prev
+  }
+}
+
+async function loadProject() {
+  return (await import("../../src/project/project")).Project
+}
+
 describe("Project.fromDirectory", () => {
   test("should handle git repository with no commits", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir()
     await $`git init`.cwd(tmp.path).quiet()
 
-    const { project } = await Project.fromDirectory(tmp.path)
+    const { project } = await p.fromDirectory(tmp.path)
 
     expect(project).toBeDefined()
     expect(project.id).toBe("global")
@@ -26,9 +92,10 @@ describe("Project.fromDirectory", () => {
   })
 
   test("should handle git repository with commits", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
 
-    const { project } = await Project.fromDirectory(tmp.path)
+    const { project } = await p.fromDirectory(tmp.path)
 
     expect(project).toBeDefined()
     expect(project.id).not.toBe("global")
@@ -39,13 +106,51 @@ describe("Project.fromDirectory", () => {
     const fileExists = await Bun.file(opencodeFile).exists()
     expect(fileExists).toBe(true)
   })
+
+  test("keeps git vcs when rev-list exits non-zero with empty output", async () => {
+    const p = await loadProject()
+    await using tmp = await tmpdir()
+    await $`git init`.cwd(tmp.path).quiet()
+
+    await withMode("rev-list-fail", async () => {
+      const { project } = await p.fromDirectory(tmp.path)
+      expect(project.vcs).toBe("git")
+      expect(project.id).toBe("global")
+      expect(project.worktree).toBe(tmp.path)
+    })
+  })
+
+  test("keeps git vcs when show-toplevel exits non-zero with empty output", async () => {
+    const p = await loadProject()
+    await using tmp = await tmpdir({ git: true })
+
+    await withMode("top-fail", async () => {
+      const { project, sandbox } = await p.fromDirectory(tmp.path)
+      expect(project.vcs).toBe("git")
+      expect(project.worktree).toBe(tmp.path)
+      expect(sandbox).toBe(tmp.path)
+    })
+  })
+
+  test("keeps git vcs when git-common-dir exits non-zero with empty output", async () => {
+    const p = await loadProject()
+    await using tmp = await tmpdir({ git: true })
+
+    await withMode("common-dir-fail", async () => {
+      const { project, sandbox } = await p.fromDirectory(tmp.path)
+      expect(project.vcs).toBe("git")
+      expect(project.worktree).toBe(tmp.path)
+      expect(sandbox).toBe(tmp.path)
+    })
+  })
 })
 
 describe("Project.fromDirectory with worktrees", () => {
   test("should set worktree to root when called from root", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
 
-    const { project, sandbox } = await Project.fromDirectory(tmp.path)
+    const { project, sandbox } = await p.fromDirectory(tmp.path)
 
     expect(project.worktree).toBe(tmp.path)
     expect(sandbox).toBe(tmp.path)
@@ -53,12 +158,13 @@ describe("Project.fromDirectory with worktrees", () => {
   })
 
   test("should set worktree to root when called from a worktree", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
 
     const worktreePath = path.join(tmp.path, "..", "worktree-test")
     await $`git worktree add ${worktreePath} -b test-branch`.cwd(tmp.path).quiet()
 
-    const { project, sandbox } = await Project.fromDirectory(worktreePath)
+    const { project, sandbox } = await p.fromDirectory(worktreePath)
 
     expect(project.worktree).toBe(tmp.path)
     expect(sandbox).toBe(worktreePath)
@@ -69,6 +175,7 @@ describe("Project.fromDirectory with worktrees", () => {
   })
 
   test("should accumulate multiple worktrees in sandboxes", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
 
     const worktree1 = path.join(tmp.path, "..", "worktree-1")
@@ -76,8 +183,8 @@ describe("Project.fromDirectory with worktrees", () => {
     await $`git worktree add ${worktree1} -b branch-1`.cwd(tmp.path).quiet()
     await $`git worktree add ${worktree2} -b branch-2`.cwd(tmp.path).quiet()
 
-    await Project.fromDirectory(worktree1)
-    const { project } = await Project.fromDirectory(worktree2)
+    await p.fromDirectory(worktree1)
+    const { project } = await p.fromDirectory(worktree2)
 
     expect(project.worktree).toBe(tmp.path)
     expect(project.sandboxes).toContain(worktree1)
@@ -91,15 +198,16 @@ describe("Project.fromDirectory with worktrees", () => {
 
 describe("Project.discover", () => {
   test("should discover favicon.png in root", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
-    const { project } = await Project.fromDirectory(tmp.path)
+    const { project } = await p.fromDirectory(tmp.path)
 
     const pngData = Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a])
     await Bun.write(path.join(tmp.path, "favicon.png"), pngData)
 
-    await Project.discover(project)
+    await p.discover(project)
 
-    const updated = await Storage.read<Project.Info>(["project", project.id])
+    const updated = await Storage.read<ProjectNS.Info>(["project", project.id])
     expect(updated.icon).toBeDefined()
     expect(updated.icon?.url).toStartWith("data:")
     expect(updated.icon?.url).toContain("base64")
@@ -107,14 +215,15 @@ describe("Project.discover", () => {
   })
 
   test("should not discover non-image files", async () => {
+    const p = await loadProject()
     await using tmp = await tmpdir({ git: true })
-    const { project } = await Project.fromDirectory(tmp.path)
+    const { project } = await p.fromDirectory(tmp.path)
 
     await Bun.write(path.join(tmp.path, "favicon.txt"), "not an image")
 
-    await Project.discover(project)
+    await p.discover(project)
 
-    const updated = await Storage.read<Project.Info>(["project", project.id])
+    const updated = await Storage.read<ProjectNS.Info>(["project", project.id])
     expect(updated.icon).toBeUndefined()
   })
 })