test(project): migrate instance tests to effect runner (#27215)

This commit is contained in:
Kit Langton
2026-05-13 02:06:16 +00:00
committed by GitHub
parent a26a2a95bd
commit 6fbb08b724
+64 -41
View File
@@ -1,12 +1,12 @@
import { afterEach, describe, expect } from "bun:test" import { describe, expect } from "bun:test"
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner" import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
import { Deferred, Effect, Fiber, Layer } from "effect" import { Deferred, Effect, Fiber, Layer, Queue } from "effect"
import { InstanceRef } from "../../src/effect/instance-ref" import { InstanceRef } from "../../src/effect/instance-ref"
import { registerDisposer } from "../../src/effect/instance-registry" import { registerDisposer } from "../../src/effect/instance-registry"
import { InstanceBootstrap } from "../../src/project/bootstrap-service" import { InstanceBootstrap } from "../../src/project/bootstrap-service"
import { Instance } from "../../src/project/instance" import { Instance } from "../../src/project/instance"
import { InstanceStore } from "../../src/project/instance-store" import { InstanceStore } from "../../src/project/instance-store"
import { disposeAllInstances, TestInstance, tmpdirScoped } from "../fixture/fixture" import { TestInstance, tmpdirScoped } from "../fixture/fixture"
import { testEffect } from "../lib/effect" import { testEffect } from "../lib/effect"
let bootstrapRun: Effect.Effect<void> = Effect.void let bootstrapRun: Effect.Effect<void> = Effect.void
@@ -19,10 +19,19 @@ const it = testEffect(
Layer.mergeAll(InstanceStore.defaultLayer, CrossSpawnSpawner.defaultLayer).pipe(Layer.provide(noopBootstrap)), Layer.mergeAll(InstanceStore.defaultLayer, CrossSpawnSpawner.defaultLayer).pipe(Layer.provide(noopBootstrap)),
) )
afterEach(async () => { const setBootstrap = (run: Effect.Effect<void>) =>
bootstrapRun = Effect.void Effect.acquireRelease(
await disposeAllInstances() Effect.sync(() => {
}) bootstrapRun = run
}),
() =>
Effect.sync(() => {
bootstrapRun = Effect.void
}),
)
const registerDisposerScoped = (disposer: (directory: string) => Promise<void>) =>
Effect.acquireRelease(Effect.sync(() => registerDisposer(disposer)), (off) => Effect.sync(off))
describe("InstanceStore", () => { describe("InstanceStore", () => {
it.live("loads instance context without installing ALS for the caller", () => it.live("loads instance context without installing ALS for the caller", () =>
@@ -43,9 +52,11 @@ describe("InstanceStore", () => {
const store = yield* InstanceStore.Service const store = yield* InstanceStore.Service
let initializedDirectory: string | undefined let initializedDirectory: string | undefined
bootstrapRun = Effect.gen(function* () { yield* setBootstrap(
initializedDirectory = (yield* InstanceRef)?.directory Effect.gen(function* () {
}) initializedDirectory = (yield* InstanceRef)?.directory
}),
)
yield* store.load({ directory: dir }) yield* store.load({ directory: dir })
expect(initializedDirectory).toBe(dir) expect(initializedDirectory).toBe(dir)
@@ -59,9 +70,11 @@ describe("InstanceStore", () => {
const store = yield* InstanceStore.Service const store = yield* InstanceStore.Service
let initialized = 0 let initialized = 0
bootstrapRun = Effect.sync(() => { yield* setBootstrap(
initialized++ Effect.sync(() => {
}) initialized++
}),
)
const first = yield* store.load({ directory: dir }) const first = yield* store.load({ directory: dir })
const second = yield* store.load({ directory: dir }) const second = yield* store.load({ directory: dir })
@@ -78,18 +91,22 @@ describe("InstanceStore", () => {
const release = yield* Deferred.make<void>() const release = yield* Deferred.make<void>()
let initialized = 0 let initialized = 0
bootstrapRun = Effect.gen(function* () { yield* setBootstrap(
initialized++ Effect.gen(function* () {
yield* Deferred.succeed(started, undefined) initialized++
yield* Deferred.await(release) yield* Deferred.succeed(started, undefined)
}) yield* Deferred.await(release)
}),
)
const first = yield* store.load({ directory: dir }).pipe(Effect.forkScoped) const first = yield* store.load({ directory: dir }).pipe(Effect.forkScoped)
yield* Deferred.await(started) yield* Deferred.await(started)
bootstrapRun = Effect.sync(() => { yield* setBootstrap(
initialized++ Effect.sync(() => {
}) initialized++
}),
)
const second = yield* store.load({ directory: dir }).pipe(Effect.forkScoped) const second = yield* store.load({ directory: dir }).pipe(Effect.forkScoped)
expect(initialized).toBe(1) expect(initialized).toBe(1)
@@ -107,10 +124,12 @@ describe("InstanceStore", () => {
const store = yield* InstanceStore.Service const store = yield* InstanceStore.Service
let attempts = 0 let attempts = 0
bootstrapRun = Effect.sync(() => { yield* setBootstrap(
attempts++ Effect.sync(() => {
throw new Error("init failed") attempts++
}) throw new Error("init failed")
}),
)
const failed = yield* store.load({ directory: dir }).pipe( const failed = yield* store.load({ directory: dir }).pipe(
Effect.as(false), Effect.as(false),
Effect.catchCause(() => Effect.succeed(true)), Effect.catchCause(() => Effect.succeed(true)),
@@ -118,9 +137,11 @@ describe("InstanceStore", () => {
expect(failed).toBe(true) expect(failed).toBe(true)
bootstrapRun = Effect.sync(() => { yield* setBootstrap(
attempts++ Effect.sync(() => {
}) attempts++
}),
)
const ctx = yield* store.load({ directory: dir }) const ctx = yield* store.load({ directory: dir })
expect(ctx.directory).toBe(dir) expect(ctx.directory).toBe(dir)
@@ -149,16 +170,17 @@ describe("InstanceStore", () => {
const reloading = yield* Deferred.make<void>() const reloading = yield* Deferred.make<void>()
const releaseReload = yield* Deferred.make<void>() const releaseReload = yield* Deferred.make<void>()
const disposed: Array<string> = [] const disposed: Array<string> = []
const off = registerDisposer(async (directory) => { yield* registerDisposerScoped(async (directory) => {
disposed.push(directory) disposed.push(directory)
}) })
yield* Effect.addFinalizer(() => Effect.sync(off))
const first = yield* store.load({ directory: dir }) const first = yield* store.load({ directory: dir })
bootstrapRun = Effect.gen(function* () { yield* setBootstrap(
yield* Deferred.succeed(reloading, undefined) Effect.gen(function* () {
yield* Deferred.await(releaseReload) yield* Deferred.succeed(reloading, undefined)
}) yield* Deferred.await(releaseReload)
}),
)
const reload = yield* store.reload({ directory: dir }).pipe(Effect.forkScoped) const reload = yield* store.reload({ directory: dir }).pipe(Effect.forkScoped)
yield* Deferred.await(reloading) yield* Deferred.await(reloading)
@@ -178,22 +200,24 @@ describe("InstanceStore", () => {
const dir = yield* tmpdirScoped({ git: true }) const dir = yield* tmpdirScoped({ git: true })
const store = yield* InstanceStore.Service const store = yield* InstanceStore.Service
const disposing = yield* Deferred.make<void>() const disposing = yield* Deferred.make<void>()
const releaseDispose = yield* Deferred.make<void>() const releaseDispose = yield* Queue.unbounded<() => void>()
const disposed: Array<string> = [] const disposed: Array<string> = []
const off = registerDisposer(async (directory) => { yield* registerDisposerScoped((directory) => {
disposed.push(directory) disposed.push(directory)
Deferred.doneUnsafe(disposing, Effect.void) Deferred.doneUnsafe(disposing, Effect.void)
await Effect.runPromise(Deferred.await(releaseDispose)) return new Promise<void>((resolve) => {
Queue.offerUnsafe(releaseDispose, resolve)
})
}) })
yield* Effect.addFinalizer(() => Effect.sync(off))
yield* store.load({ directory: dir }) yield* store.load({ directory: dir })
const first = yield* store.disposeAll().pipe(Effect.forkScoped) const first = yield* store.disposeAll().pipe(Effect.forkScoped)
yield* Deferred.await(disposing) yield* Deferred.await(disposing)
const release = yield* Queue.take(releaseDispose)
const second = yield* store.disposeAll().pipe(Effect.forkScoped) const second = yield* store.disposeAll().pipe(Effect.forkScoped)
expect(disposed).toEqual([dir]) expect(disposed).toEqual([dir])
yield* Deferred.succeed(releaseDispose, undefined) yield* Effect.sync(release)
yield* Effect.all([Fiber.join(first), Fiber.join(second)]) yield* Effect.all([Fiber.join(first), Fiber.join(second)])
expect(disposed).toEqual([dir]) expect(disposed).toEqual([dir])
}), }),
@@ -205,10 +229,9 @@ describe("InstanceStore", () => {
const dir2 = yield* tmpdirScoped({ git: true }) const dir2 = yield* tmpdirScoped({ git: true })
const store = yield* InstanceStore.Service const store = yield* InstanceStore.Service
const disposed: Array<string> = [] const disposed: Array<string> = []
const off = registerDisposer(async (directory) => { yield* registerDisposerScoped(async (directory) => {
disposed.push(directory) disposed.push(directory)
}) })
yield* Effect.addFinalizer(() => Effect.sync(off))
yield* store.load({ directory: dir1 }) yield* store.load({ directory: dir1 })
yield* store.disposeAll() yield* store.disposeAll()