diff --git a/apps/server/src/serverRuntimeStartup.autoPull.test.ts b/apps/server/src/serverRuntimeStartup.autoPull.test.ts index d5eab968b16c..8313d33dc220 100644 --- a/apps/server/src/serverRuntimeStartup.autoPull.test.ts +++ b/apps/server/src/serverRuntimeStartup.autoPull.test.ts @@ -7,8 +7,8 @@ import * as Deferred from "effect/Deferred"; import * as Effect from "effect/Effect"; import * as Layer from "effect/Layer"; import * as Path from "effect/Path"; -import { HttpServer } from "effect/unstable/http"; -import * as NetAddress from "effect/unstable/net/NetAddress"; +import { HttpServer } from "effect/http"; +import * as NetAddress from "effect/net/NetAddress"; import * as EnvironmentAuth from "./auth/EnvironmentAuth.ts"; import * as ServiceLauncherClient from "./cloud/serviceLauncherClient.ts"; @@ -17,6 +17,7 @@ import * as ServerEnvironment from "./environment/ServerEnvironment.ts"; import * as Keybindings from "./keybindings.ts"; import * as EffectWorker from "./orchestration-v2/EffectWorker.ts"; import * as LegacyV1ThreadImporter from "./orchestration-v2/legacy/LegacyV1ThreadImporter.ts"; +import * as Orchestrator from "./orchestration-v2/Orchestrator.ts"; import * as ProjectStore from "./orchestration-v2/ProjectStore.ts"; import * as ProviderRuntimeRecovery from "./orchestration-v2/ProviderRuntimeRecoveryService.ts"; import * as ProviderSessionManager from "./orchestration-v2/ProviderSessionManager.ts"; @@ -113,6 +114,7 @@ it.effect("parks automatic pull until activation without delaying command readin prepareForShutdown: Effect.void, reconcile: () => Effect.succeed(recovery), }), + Layer.mock(Orchestrator.OrchestratorV2)({ recoverDelegatedTasks: Effect.void }), Layer.mock(ProviderSessionManager.ProviderSessionManagerV2)({ shutdown: Effect.void }), Layer.mock(AgentAwarenessRelay.AgentAwarenessRelay)({ start: () => Effect.void }), Layer.mock(EffectWorker.OrchestrationEffectWorkerV2)({ runOnce: Effect.never }),