test: migrate worktree tests to effect runner (#27177)

This commit is contained in:
Kit Langton
2026-05-12 19:55:07 -04:00
committed by GitHub
parent 1b6599f411
commit d8c8322ddf
+96 -106
View File
@@ -1,49 +1,60 @@
import { $ } from "bun"
import { afterEach, describe, expect } from "bun:test" import { afterEach, describe, expect } from "bun:test"
import * as fs from "fs/promises"
import path from "path" import path from "path"
import { Cause, Effect, Exit, Layer } from "effect" import { AppFileSystem } from "@opencode-ai/core/filesystem"
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner" import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
import { Instance } from "../../src/project/instance" import { Cause, Deferred, Effect, Exit, Layer } from "effect"
import { WithInstance } from "../../src/project/with-instance" import { GlobalBus, type GlobalEvent } from "../../src/bus/global"
import { InstanceRuntime } from "../../src/project/instance-runtime" import { Git } from "../../src/git"
import { Worktree } from "../../src/worktree" import { Worktree } from "../../src/worktree"
import { disposeAllInstances, provideInstance, provideTmpdirInstance } from "../fixture/fixture" import { disposeAllInstances, TestInstance } from "../fixture/fixture"
import { testEffect } from "../lib/effect" import { testEffect } from "../lib/effect"
const it = testEffect(Layer.mergeAll(Worktree.defaultLayer, CrossSpawnSpawner.defaultLayer)) const it = testEffect(
const wintest = process.platform !== "win32" ? it.live : it.live.skip Layer.mergeAll(Worktree.defaultLayer, AppFileSystem.defaultLayer, CrossSpawnSpawner.defaultLayer, Git.defaultLayer),
)
const wintest = process.platform !== "win32" ? it.instance : it.instance.skip
function normalize(input: string) { function normalize(input: string) {
return input.replace(/\\/g, "/").toLowerCase() return input.replace(/\\/g, "/").toLowerCase()
} }
async function waitReady() { const waitReady = Effect.fn("WorktreeTest.waitReady")(function* () {
const { GlobalBus } = await import("../../src/bus/global") const ready = yield* Deferred.make<{ name: string; branch?: string }>()
const on = (evt: GlobalEvent) => {
return await new Promise<{ name: string; branch?: string }>((resolve, reject) => {
const timer = setTimeout(() => {
GlobalBus.off("event", on)
reject(new Error("timed out waiting for worktree.ready"))
}, 10_000)
function on(evt: { directory?: string; payload: { type: string; properties: { name: string; branch?: string } } }) {
if (evt.payload.type !== Worktree.Event.Ready.type) return if (evt.payload.type !== Worktree.Event.Ready.type) return
clearTimeout(timer) Deferred.doneUnsafe(ready, Effect.succeed(evt.payload.properties))
GlobalBus.off("event", on)
resolve(evt.payload.properties)
} }
GlobalBus.on("event", on) GlobalBus.on("event", on)
yield* Effect.addFinalizer(() => Effect.sync(() => GlobalBus.off("event", on)))
return Deferred.await(ready).pipe(
Effect.race(
Effect.sleep("10 seconds").pipe(
Effect.flatMap(() => Effect.fail(new Error("timed out waiting for worktree.ready"))),
),
),
)
})
const git = Effect.fn("WorktreeTest.git")(function* (cwd: string, args: string[]) {
const service = yield* Git.Service
const result = yield* service.run(args, { cwd })
if (result.exitCode !== 0) throw new Error(`git ${args.join(" ")} failed: ${result.stderr.toString("utf8")}`)
return result.text()
})
const gitResult = Effect.fn("WorktreeTest.gitResult")(function* (cwd: string, args: string[]) {
const service = yield* Git.Service
return yield* service.run(args, { cwd })
}) })
}
describe("Worktree", () => { describe("Worktree", () => {
afterEach(() => disposeAllInstances()) afterEach(() => disposeAllInstances())
describe("makeWorktreeInfo", () => { describe("makeWorktreeInfo", () => {
it.live("returns info with name, branch, and directory", () => it.instance(
provideTmpdirInstance( "returns info with name, branch, and directory",
() => () =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
@@ -55,11 +66,10 @@ describe("Worktree", () => {
expect(info.directory).toContain(info.name) expect(info.directory).toContain(info.name)
}), }),
{ git: true }, { git: true },
),
) )
it.live("uses provided name as base", () => it.instance(
provideTmpdirInstance( "uses provided name as base",
() => () =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
@@ -69,11 +79,10 @@ describe("Worktree", () => {
expect(info.branch).toBe("opencode/my-feature") expect(info.branch).toBe("opencode/my-feature")
}), }),
{ git: true }, { git: true },
),
) )
it.live("slugifies the provided name", () => it.instance(
provideTmpdirInstance( "slugifies the provided name",
() => () =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
@@ -82,15 +91,15 @@ describe("Worktree", () => {
expect(info.name).toBe("my-feature-branch") expect(info.name).toBe("my-feature-branch")
}), }),
{ git: true }, { git: true },
),
) )
it.live("omits branch for detached info", () => it.instance(
provideTmpdirInstance( "omits branch for detached info",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
yield* Effect.promise(() => $`git branch opencode/my-feature`.cwd(dir).quiet()) yield* git(test.directory, ["branch", "opencode/my-feature"])
const info = yield* svc.makeWorktreeInfo({ name: "my-feature", detached: true }) const info = yield* svc.makeWorktreeInfo({ name: "my-feature", detached: true })
@@ -98,11 +107,9 @@ describe("Worktree", () => {
expect(info.branch).toBeUndefined() expect(info.branch).toBeUndefined()
}), }),
{ git: true }, { git: true },
),
) )
it.live("throws NotGitError for non-git directories", () => it.instance("throws NotGitError for non-git directories", () =>
provideTmpdirInstance(() =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const exit = yield* Effect.exit(svc.makeWorktreeInfo()) const exit = yield* Effect.exit(svc.makeWorktreeInfo())
@@ -110,156 +117,141 @@ describe("Worktree", () => {
expect(Exit.isFailure(exit)).toBe(true) expect(Exit.isFailure(exit)).toBe(true)
if (Exit.isFailure(exit)) expect(Cause.squash(exit.cause)).toBeInstanceOf(Worktree.NotGitError) if (Exit.isFailure(exit)) expect(Cause.squash(exit.cause)).toBeInstanceOf(Worktree.NotGitError)
}), }),
),
) )
wintest("creates detached git worktree when info has no branch", () => wintest(
provideTmpdirInstance( "creates detached git worktree when info has no branch",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const info = yield* svc.makeWorktreeInfo({ name: "detached-test", detached: true }) const info = yield* svc.makeWorktreeInfo({ name: "detached-test", detached: true })
const ready = waitReady() const ready = yield* waitReady()
yield* svc.createFromInfo(info) yield* svc.createFromInfo(info)
const list = yield* Effect.promise(() => $`git worktree list --porcelain`.cwd(dir).quiet().text()) const list = yield* git(test.directory, ["worktree", "list", "--porcelain"])
const normalizedList = normalize(list) const normalizedList = normalize(list)
const normalizedDir = normalize(info.directory) const normalizedDir = normalize(info.directory)
expect(normalizedList).toContain(normalizedDir) expect(normalizedList).toContain(normalizedDir)
const branch = yield* Effect.promise(() => const branch = yield* gitResult(info.directory, ["symbolic-ref", "-q", "--short", "HEAD"])
$`git symbolic-ref -q --short HEAD`.cwd(info.directory).quiet().nothrow(),
)
expect(branch.exitCode).not.toBe(0) expect(branch.exitCode).not.toBe(0)
const props = yield* Effect.promise(() => ready) const props = yield* ready
expect(props.name).toBe(info.name) expect(props.name).toBe(info.name)
expect(props.branch).toBeUndefined() expect(props.branch).toBeUndefined()
yield* svc.remove({ directory: info.directory }) yield* svc.remove({ directory: info.directory })
}), }),
{ git: true }, { git: true },
),
) )
}) })
describe("create + remove lifecycle", () => { describe("create + remove lifecycle", () => {
it.live("create returns worktree info and remove cleans up", () => it.instance(
provideTmpdirInstance( "create returns worktree info and remove cleans up",
() => () =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const ready = yield* waitReady()
const info = yield* svc.create() const info = yield* svc.create()
expect(info.name).toBeDefined() expect(info.name).toBeDefined()
expect(info.branch ?? "").toStartWith("opencode/") expect(info.branch ?? "").toStartWith("opencode/")
expect(info.directory).toBeDefined() expect(info.directory).toBeDefined()
yield* Effect.promise(() => Bun.sleep(1000)) yield* ready
const ok = yield* svc.remove({ directory: info.directory }) const ok = yield* svc.remove({ directory: info.directory })
expect(ok).toBe(true) expect(ok).toBe(true)
}), }),
{ git: true }, { git: true },
),
) )
it.live("create returns after setup and fires Event.Ready after bootstrap", () => it.instance(
provideTmpdirInstance( "create returns after setup and fires Event.Ready after bootstrap",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const fs = yield* AppFileSystem.Service
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const ready = waitReady() const ready = yield* waitReady()
const info = yield* svc.create() const info = yield* svc.create()
expect(info.name).toBeDefined() expect(info.name).toBeDefined()
expect(info.branch ?? "").toStartWith("opencode/") expect(info.branch ?? "").toStartWith("opencode/")
const text = yield* Effect.promise(() => $`git worktree list --porcelain`.cwd(dir).quiet().text()) const text = yield* git(test.directory, ["worktree", "list", "--porcelain"])
const next = yield* Effect.promise(() => fs.realpath(info.directory).catch(() => info.directory)) const next = yield* fs.realPath(info.directory).pipe(Effect.catch(() => Effect.succeed(info.directory)))
expect(normalize(text)).toContain(normalize(next)) expect(normalize(text)).toContain(normalize(next))
const props = yield* Effect.promise(() => ready) const props = yield* ready
expect(props.name).toBe(info.name) expect(props.name).toBe(info.name)
expect(props.branch).toBe(info.branch) expect(props.branch).toBe(info.branch)
yield* Effect.promise(() =>
WithInstance.provide({
directory: info.directory,
fn: () => InstanceRuntime.disposeInstance(Instance.current),
}),
)
yield* Effect.promise(() => Bun.sleep(100))
yield* svc.remove({ directory: info.directory }) yield* svc.remove({ directory: info.directory })
}), }),
{ git: true }, { git: true },
),
) )
it.live("create with custom name", () => it.instance(
provideTmpdirInstance( "create with custom name",
() => () =>
Effect.gen(function* () { Effect.gen(function* () {
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const ready = waitReady() const ready = yield* waitReady()
const info = yield* svc.create({ name: "test-workspace" }) const info = yield* svc.create({ name: "test-workspace" })
expect(info.name).toBe("test-workspace") expect(info.name).toBe("test-workspace")
expect(info.branch).toBe("opencode/test-workspace") expect(info.branch).toBe("opencode/test-workspace")
yield* Effect.promise(() => ready) yield* ready
yield* Effect.promise(() =>
WithInstance.provide({
directory: info.directory,
fn: () => InstanceRuntime.disposeInstance(Instance.current),
}),
)
yield* Effect.promise(() => Bun.sleep(100))
yield* svc.remove({ directory: info.directory }) yield* svc.remove({ directory: info.directory })
}), }),
{ git: true }, { git: true },
),
) )
}) })
describe("createFromInfo", () => { describe("createFromInfo", () => {
wintest("creates git worktree and boots asynchronously", () => wintest(
provideTmpdirInstance( "creates git worktree and boots asynchronously",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const info = yield* svc.makeWorktreeInfo({ name: "from-info-test" }) const info = yield* svc.makeWorktreeInfo({ name: "from-info-test" })
const ready = waitReady() const ready = yield* waitReady()
yield* svc.createFromInfo(info) yield* svc.createFromInfo(info)
const list = yield* Effect.promise(() => $`git worktree list --porcelain`.cwd(dir).quiet().text()) const list = yield* git(test.directory, ["worktree", "list", "--porcelain"])
const normalizedList = list.replace(/\\/g, "/") const normalizedList = list.replace(/\\/g, "/")
const normalizedDir = info.directory.replace(/\\/g, "/") const normalizedDir = info.directory.replace(/\\/g, "/")
expect(normalizedList).toContain(normalizedDir) expect(normalizedList).toContain(normalizedDir)
yield* Effect.promise(() => ready) yield* ready
yield* svc.remove({ directory: info.directory }) yield* svc.remove({ directory: info.directory })
}), }),
{ git: true }, { git: true },
),
) )
}) })
describe("list", () => { describe("list", () => {
it.live("uses parent folder name when worktree basename matches the primary worktree", () => it.instance(
provideTmpdirInstance( "uses parent folder name when worktree basename matches the primary worktree",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const fs = yield* AppFileSystem.Service
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const parent = path.join(path.dirname(dir), `${path.basename(dir)}-parent`) const parent = path.join(path.dirname(test.directory), `${path.basename(test.directory)}-parent`)
const target = path.join(parent, path.basename(dir)) const target = path.join(parent, path.basename(test.directory))
const branch = `same-basename-list-${Date.now()}` const branch = `same-basename-list-${Date.now()}`
yield* Effect.promise(() => fs.mkdir(parent, { recursive: true })) yield* fs.ensureDir(parent)
yield* Effect.promise(() => $`git worktree add -b ${branch} ${target}`.cwd(dir).quiet()) yield* git(test.directory, ["worktree", "add", "-b", branch, target])
const list = yield* svc.list() const list = yield* svc.list()
const directory = yield* Effect.promise(() => fs.realpath(target).catch(() => target)) const directory = yield* fs.realPath(target).pipe(Effect.catch(() => Effect.succeed(target)))
expect(list.map((item) => ({ ...item, directory: normalize(item.directory) }))).toContainEqual({ expect(list.map((item) => ({ ...item, directory: normalize(item.directory) }))).toContainEqual({
name: path.basename(parent), name: path.basename(parent),
@@ -270,33 +262,31 @@ describe("Worktree", () => {
yield* svc.remove({ directory: target }) yield* svc.remove({ directory: target })
}), }),
{ git: true }, { git: true },
),
) )
}) })
describe("remove edge cases", () => { describe("remove edge cases", () => {
it.live("remove non-existent directory succeeds silently", () => it.instance(
provideTmpdirInstance( "remove non-existent directory succeeds silently",
(dir) => () =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const ok = yield* svc.remove({ directory: path.join(dir, "does-not-exist") }) const ok = yield* svc.remove({ directory: path.join(test.directory, "does-not-exist") })
expect(ok).toBe(true) expect(ok).toBe(true)
}), }),
{ git: true }, { git: true },
),
) )
it.live("throws NotGitError for non-git directories", () => it.instance("throws NotGitError for non-git directories", () =>
provideTmpdirInstance(() =>
Effect.gen(function* () { Effect.gen(function* () {
const test = yield* TestInstance
const svc = yield* Worktree.Service const svc = yield* Worktree.Service
const exit = yield* Effect.exit(svc.remove({ directory: "/tmp/fake" })) const exit = yield* Effect.exit(svc.remove({ directory: path.join(test.directory, "fake") }))
expect(Exit.isFailure(exit)).toBe(true) expect(Exit.isFailure(exit)).toBe(true)
if (Exit.isFailure(exit)) expect(Cause.squash(exit.cause)).toBeInstanceOf(Worktree.NotGitError) if (Exit.isFailure(exit)) expect(Cause.squash(exit.cause)).toBeInstanceOf(Worktree.NotGitError)
}), }),
),
) )
}) })
}) })