diff --git a/src/tasks/task-owner-access.test.ts b/src/tasks/task-owner-access.test.ts index 1105d39acbf..8db3c08f436 100644 --- a/src/tasks/task-owner-access.test.ts +++ b/src/tasks/task-owner-access.test.ts @@ -21,9 +21,19 @@ afterEach(() => { async function withTaskRegistryTempDir(run: () => Promise | T): Promise { return await withTempDir({ prefix: "openclaw-task-owner-access-" }, async (root) => { + const previousStateDir = process.env.OPENCLAW_STATE_DIR; process.env.OPENCLAW_STATE_DIR = root; resetTaskRegistryForTests({ persist: false }); - return await run(); + try { + return await run(); + } finally { + resetTaskRegistryForTests({ persist: false }); + if (previousStateDir == null) { + delete process.env.OPENCLAW_STATE_DIR; + } else { + process.env.OPENCLAW_STATE_DIR = previousStateDir; + } + } }); }