refactor(opencode): migrate session tests to layer nodes (#34494)
This commit is contained in:
@@ -1,9 +1,8 @@
|
|||||||
import { expect, test } from "bun:test"
|
import { expect, test } from "bun:test"
|
||||||
import { setTimeout as sleep } from "node:timers/promises"
|
import { setTimeout as sleep } from "node:timers/promises"
|
||||||
import { LayerNode } from "@opencode-ai/core/effect/layer-node"
|
import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
|
||||||
import { Effect, Layer } from "effect"
|
import { Effect, Layer } from "effect"
|
||||||
import { FSUtil } from "@opencode-ai/core/fs-util"
|
import { FSUtil } from "@opencode-ai/core/fs-util"
|
||||||
import { EffectFlock } from "@opencode-ai/core/util/effect-flock"
|
|
||||||
import { McpAuth } from "../../src/mcp/auth"
|
import { McpAuth } from "../../src/mcp/auth"
|
||||||
|
|
||||||
function authFile() {
|
function authFile() {
|
||||||
@@ -11,7 +10,7 @@ function authFile() {
|
|||||||
let activeWrites = 0
|
let activeWrites = 0
|
||||||
let sawOverlap = false
|
let sawOverlap = false
|
||||||
|
|
||||||
const layer = Layer.effect(
|
const fsLayer = Layer.effect(
|
||||||
FSUtil.Service,
|
FSUtil.Service,
|
||||||
Effect.gen(function* () {
|
Effect.gen(function* () {
|
||||||
const fs = yield* FSUtil.Service
|
const fs = yield* FSUtil.Service
|
||||||
@@ -42,14 +41,14 @@ function authFile() {
|
|||||||
: fs.writeJson(file, value, mode),
|
: fs.writeJson(file, value, mode),
|
||||||
})
|
})
|
||||||
}),
|
}),
|
||||||
).pipe(Layer.provide(LayerNode.compile(FSUtil.node)))
|
).pipe(Layer.provide(AppNodeBuilder.build(FSUtil.node)))
|
||||||
|
|
||||||
return { layer, raw: () => raw }
|
return { fsLayer, raw: () => raw }
|
||||||
}
|
}
|
||||||
|
|
||||||
function authService(layer: Layer.Layer<FSUtil.Service>) {
|
function authService(fsLayer: Layer.Layer<FSUtil.Service>) {
|
||||||
return McpAuth.Service.use((auth) => Effect.succeed(auth)).pipe(
|
return McpAuth.Service.use((auth) => Effect.succeed(auth)).pipe(
|
||||||
Effect.provide(LayerNode.compile(McpAuth.node, [[FSUtil.node, layer]])),
|
Effect.provide(AppNodeBuilder.build(McpAuth.node, [[FSUtil.node, fsLayer]])),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -58,8 +57,8 @@ test("serializes concurrent auth file updates across service instances", async (
|
|||||||
|
|
||||||
await Effect.runPromise(
|
await Effect.runPromise(
|
||||||
Effect.gen(function* () {
|
Effect.gen(function* () {
|
||||||
const first = yield* authService(file.layer)
|
const first = yield* authService(file.fsLayer)
|
||||||
const second = yield* authService(file.layer)
|
const second = yield* authService(file.fsLayer)
|
||||||
|
|
||||||
yield* Effect.all(
|
yield* Effect.all(
|
||||||
[
|
[
|
||||||
|
|||||||
@@ -4,21 +4,19 @@ import os from "os"
|
|||||||
import { Cause, Deferred, Effect, Exit, Fiber, Layer } from "effect"
|
import { Cause, Deferred, Effect, Exit, Fiber, Layer } from "effect"
|
||||||
import { EventV2Bridge } from "../../src/event-v2-bridge"
|
import { EventV2Bridge } from "../../src/event-v2-bridge"
|
||||||
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
||||||
import { Database } from "@opencode-ai/core/database/database"
|
|
||||||
import { Permission } from "../../src/permission"
|
import { Permission } from "../../src/permission"
|
||||||
import { InstanceBootstrap } from "../../src/project/bootstrap-service"
|
import { InstanceBootstrap } from "../../src/project/bootstrap"
|
||||||
import { InstanceStore } from "../../src/project/instance-store"
|
import { InstanceStore } from "../../src/project/instance-store"
|
||||||
import { TestInstance, tmpdirScoped } from "../fixture/fixture"
|
import { TestInstance, tmpdirScoped } from "../fixture/fixture"
|
||||||
import { testEffect } from "../lib/effect"
|
import { testEffect } from "../lib/effect"
|
||||||
import { MessageID, SessionID } from "../../src/session/schema"
|
import { MessageID, SessionID } from "../../src/session/schema"
|
||||||
|
import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
|
||||||
|
import { LayerNode } from "@opencode-ai/core/effect/layer-node"
|
||||||
|
|
||||||
const events = EventV2Bridge.defaultLayer
|
|
||||||
const noopBootstrap = Layer.succeed(InstanceBootstrap.Service, InstanceBootstrap.Service.of({ run: Effect.void }))
|
const noopBootstrap = Layer.succeed(InstanceBootstrap.Service, InstanceBootstrap.Service.of({ run: Effect.void }))
|
||||||
const env = Layer.mergeAll(
|
const env = AppNodeBuilder.build(
|
||||||
Permission.layer.pipe(Layer.provide(Database.defaultLayer), Layer.provide(events)),
|
LayerNode.group([Permission.node, EventV2Bridge.node, CrossSpawnSpawner.node, InstanceStore.node]),
|
||||||
events,
|
[[InstanceBootstrap.node, noopBootstrap]],
|
||||||
CrossSpawnSpawner.defaultLayer,
|
|
||||||
InstanceStore.defaultLayer.pipe(Layer.provide(noopBootstrap)),
|
|
||||||
)
|
)
|
||||||
const it = testEffect(env)
|
const it = testEffect(env)
|
||||||
|
|
||||||
|
|||||||
@@ -2,34 +2,39 @@ import { describe, expect, test } from "bun:test"
|
|||||||
import { SessionV1 } from "@opencode-ai/core/v1/session"
|
import { SessionV1 } from "@opencode-ai/core/v1/session"
|
||||||
import path from "path"
|
import path from "path"
|
||||||
import { Effect, FileSystem, Layer } from "effect"
|
import { Effect, FileSystem, Layer } from "effect"
|
||||||
import { FetchHttpClient } from "effect/unstable/http"
|
|
||||||
import { NodeFileSystem } from "@effect/platform-node"
|
|
||||||
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
||||||
import { FSUtil } from "@opencode-ai/core/fs-util"
|
|
||||||
|
|
||||||
import { Instruction } from "../../src/session/instruction"
|
import { Instruction } from "../../src/session/instruction"
|
||||||
import type { MessageV2 } from "../../src/session/message-v2"
|
import type { MessageV2 } from "../../src/session/message-v2"
|
||||||
import { MessageID, PartID, SessionID } from "../../src/session/schema"
|
import { MessageID, PartID, SessionID } from "../../src/session/schema"
|
||||||
import { Global } from "@opencode-ai/core/global"
|
import { Global } from "@opencode-ai/core/global"
|
||||||
import { RuntimeFlags } from "../../src/effect/runtime-flags"
|
import { RuntimeFlags } from "../../src/effect/runtime-flags"
|
||||||
import { provideInstance, provideTmpdirInstance, testInstanceStoreLayer, tmpdirScoped } from "../fixture/fixture"
|
import { provideInstance, provideTmpdirInstance, tmpdirScoped } from "../fixture/fixture"
|
||||||
import { testEffect } from "../lib/effect"
|
import { testEffect } from "../lib/effect"
|
||||||
import { TestConfig } from "../fixture/config"
|
import { TestConfig } from "../fixture/config"
|
||||||
import { ProviderV2 } from "@opencode-ai/core/provider"
|
import { ProviderV2 } from "@opencode-ai/core/provider"
|
||||||
import { ModelV2 } from "@opencode-ai/core/model"
|
import { ModelV2 } from "@opencode-ai/core/model"
|
||||||
|
import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
|
||||||
|
import { LayerNode } from "@opencode-ai/core/effect/layer-node"
|
||||||
|
import { LayerNodePlatform } from "@opencode-ai/core/effect/app-node-platform"
|
||||||
|
import { InstanceStore } from "@/project/instance-store"
|
||||||
|
import { InstanceBootstrap } from "@/project/bootstrap"
|
||||||
|
import { Config } from "@/config/config"
|
||||||
|
|
||||||
const it = testEffect(Layer.mergeAll(CrossSpawnSpawner.defaultLayer, NodeFileSystem.layer, testInstanceStoreLayer))
|
const it = testEffect(
|
||||||
|
AppNodeBuilder.build(LayerNode.group([CrossSpawnSpawner.node, LayerNodePlatform.filesystem, InstanceStore.node]), [
|
||||||
|
[InstanceBootstrap.node, Layer.succeed(InstanceBootstrap.Service, InstanceBootstrap.Service.of({ run: Effect.void }))],
|
||||||
|
]),
|
||||||
|
)
|
||||||
|
|
||||||
const configLayer = TestConfig.layer()
|
const configLayer = Layer.succeed(Config.Service, TestConfig.make())
|
||||||
|
|
||||||
const instructionLayer = (global: Partial<Global.Interface>, flags: Partial<RuntimeFlags.Info> = {}) =>
|
const instructionLayer = (global: Partial<Global.Interface>, flags: Partial<RuntimeFlags.Info> = {}) =>
|
||||||
Instruction.layer.pipe(
|
AppNodeBuilder.build(Instruction.node, [
|
||||||
Layer.provide(configLayer),
|
[Config.node, configLayer],
|
||||||
Layer.provide(FSUtil.defaultLayer),
|
[Global.node, Global.layerWith(global)],
|
||||||
Layer.provide(FetchHttpClient.layer),
|
[RuntimeFlags.node, RuntimeFlags.layer(flags)],
|
||||||
Layer.provide(Global.layerWith(global)),
|
])
|
||||||
Layer.provide(RuntimeFlags.layer(flags)),
|
|
||||||
)
|
|
||||||
|
|
||||||
const provideInstruction =
|
const provideInstruction =
|
||||||
(global: Partial<Global.Interface>, flags?: Partial<RuntimeFlags.Info>) =>
|
(global: Partial<Global.Interface>, flags?: Partial<RuntimeFlags.Info>) =>
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
import { describe, expect } from "bun:test"
|
import { describe, expect } from "bun:test"
|
||||||
import { SessionV1 } from "@opencode-ai/core/v1/session"
|
import { SessionV1 } from "@opencode-ai/core/v1/session"
|
||||||
import { Database } from "@opencode-ai/core/database/database"
|
|
||||||
import { EventV2 } from "@opencode-ai/core/event"
|
import { EventV2 } from "@opencode-ai/core/event"
|
||||||
import { SessionProjector } from "@opencode-ai/core/session/projector"
|
import { SessionProjector } from "@opencode-ai/core/session/projector"
|
||||||
import { Deferred, Effect, Exit, Layer } from "effect"
|
import { Deferred, Effect, Exit, Layer } from "effect"
|
||||||
@@ -8,26 +7,23 @@ import { Session as SessionNs } from "@/session/session"
|
|||||||
import { MessageV2 } from "../../src/session/message-v2"
|
import { MessageV2 } from "../../src/session/message-v2"
|
||||||
import { MessageID, PartID, type SessionID } from "../../src/session/schema"
|
import { MessageID, PartID, type SessionID } from "../../src/session/schema"
|
||||||
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
||||||
import { provideInstance, testInstanceStoreLayer, tmpdirScoped } from "../fixture/fixture"
|
import { provideInstance, tmpdirScoped } from "../fixture/fixture"
|
||||||
import { testEffect } from "../lib/effect"
|
import { testEffect } from "../lib/effect"
|
||||||
import { Storage } from "@/storage/storage"
|
|
||||||
import { RuntimeFlags } from "@/effect/runtime-flags"
|
import { RuntimeFlags } from "@/effect/runtime-flags"
|
||||||
import { BackgroundJob } from "@/background/job"
|
|
||||||
import { EventV2Bridge } from "@/event-v2-bridge"
|
import { EventV2Bridge } from "@/event-v2-bridge"
|
||||||
import { GlobalBus } from "@/bus/global"
|
import { GlobalBus } from "@/bus/global"
|
||||||
|
import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
|
||||||
|
import { LayerNode } from "@opencode-ai/core/effect/layer-node"
|
||||||
|
import { InstanceStore } from "@/project/instance-store"
|
||||||
|
import { InstanceBootstrap } from "@/project/bootstrap"
|
||||||
|
|
||||||
const it = testEffect(
|
const it = testEffect(
|
||||||
Layer.mergeAll(
|
AppNodeBuilder.build(
|
||||||
SessionNs.layer.pipe(
|
LayerNode.group([SessionNs.node, EventV2Bridge.node, SessionProjector.node, CrossSpawnSpawner.node, InstanceStore.node]),
|
||||||
Layer.provide(Storage.defaultLayer),
|
[
|
||||||
Layer.provide(Database.defaultLayer),
|
[RuntimeFlags.node, RuntimeFlags.layer({ experimentalWorkspaces: false })],
|
||||||
Layer.provideMerge(EventV2Bridge.defaultLayer),
|
[InstanceBootstrap.node, Layer.succeed(InstanceBootstrap.Service, InstanceBootstrap.Service.of({ run: Effect.void }))],
|
||||||
Layer.provide(SessionProjector.defaultLayer),
|
],
|
||||||
Layer.provide(RuntimeFlags.layer({ experimentalWorkspaces: false })),
|
|
||||||
Layer.provide(BackgroundJob.defaultLayer),
|
|
||||||
),
|
|
||||||
CrossSpawnSpawner.defaultLayer,
|
|
||||||
testInstanceStoreLayer,
|
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user