diff --git a/src/tasks/task-flow-owner-access.test.ts b/src/tasks/task-flow-owner-access.test.ts index 2f8edbaa5d5..0c07cb760e1 100644 --- a/src/tasks/task-flow-owner-access.test.ts +++ b/src/tasks/task-flow-owner-access.test.ts @@ -1,4 +1,4 @@ -import { beforeEach, describe, expect, it } from "vitest"; +import { afterEach, beforeEach, describe, expect, it } from "vitest"; import { findLatestTaskFlowForOwner, getTaskFlowByIdForOwner, @@ -6,10 +6,24 @@ import { resolveTaskFlowForLookupTokenForOwner, } from "./task-flow-owner-access.js"; import { createManagedTaskFlow, resetTaskFlowRegistryForTests } from "./task-flow-registry.js"; +import { configureTaskFlowRegistryRuntime } from "./task-flow-registry.store.js"; beforeEach(() => { - resetTaskFlowRegistryForTests(); + configureTaskFlowRegistryRuntime({ + store: { + loadSnapshot: () => ({ flows: new Map() }), + saveSnapshot: () => {}, + upsertFlow: () => {}, + deleteFlow: () => {}, + }, + }); + resetTaskFlowRegistryForTests({ persist: false }); }); + +afterEach(() => { + resetTaskFlowRegistryForTests({ persist: false }); +}); + describe("task flow owner access", () => { it("returns owner-scoped flows for direct and owner-key lookups", () => { const older = createManagedTaskFlow({