LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-13 15:47:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: enforce thread-first vitest configs

Browse Source
This commit is contained in:
Peter Steinberger
2026-04-04 05:49:56 +01:00
parent 33e10c4772
commit 757a20b656
21 changed files with 269 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
test/setup-openclaw-runtime.ts
View File

@@ -1,4 +1,4 @@
import { afterAll, afterEach, beforeAll } from "vitest";
import { afterAll, afterEach, beforeAll, vi } from "vitest";
import type {
ChannelId,
ChannelOutboundAdapter,
@@ -21,11 +21,14 @@ type WorkerRuntimeState = {
};
type WorkerCleanupDeps = {
resetContextWindowCacheForTest: typeof import("../src/agents/context.js").resetContextWindowCacheForTest;
resetModelsJsonReadyCacheForTest: typeof import("../src/agents/models-config.js").resetModelsJsonReadyCacheForTest;
clearSessionStoreCaches: typeof import("../src/config/sessions/store-cache.js").clearSessionStoreCaches;
drainFileLockStateForTest: typeof import("../src/infra/file-lock.js").drainFileLockStateForTest;
drainSessionStoreLockQueuesForTest: typeof import("../src/config/sessions/store-lock-state.js").drainSessionStoreLockQueuesForTest;
drainSessionWriteLockStateForTest: typeof import("../src/agents/session-write-lock.js").drainSessionWriteLockStateForTest;
resetContextWindowCacheForTest: typeof import("../src/agents/context-runtime-state.js").resetContextWindowCacheForTest;
resetFileLockStateForTest: typeof import("../src/infra/file-lock.js").resetFileLockStateForTest;
resetModelsJsonReadyCacheForTest: typeof import("../src/agents/models-config-state.js").resetModelsJsonReadyCacheForTest;
resetSessionWriteLockStateForTest: typeof import("../src/agents/session-write-lock.js").resetSessionWriteLockStateForTest;
cleanupSessionStateForTest: typeof import("../src/test-utils/session-state-cleanup.js").cleanupSessionStateForTest;
};
type ReplyToModeResolver = NonNullable<
@@ -49,21 +52,38 @@ async function loadWorkerCleanupDeps(): Promise<WorkerCleanupDeps> {
const [
{ resetContextWindowCacheForTest },
{ resetModelsJsonReadyCacheForTest },
{ clearSessionStoreCaches },
{ drainSessionStoreLockQueuesForTest },
{ drainFileLockStateForTest, resetFileLockStateForTest },
{ drainSessionWriteLockStateForTest, resetSessionWriteLockStateForTest },
{ cleanupSessionStateForTest },
] = await Promise.all([
import("../src/agents/context.js"),
import("../src/agents/models-config.js"),
import("../src/agents/session-write-lock.js"),
import("../src/test-utils/session-state-cleanup.js"),
vi.importActual<typeof import("../src/agents/context-runtime-state.js")>(
"../src/agents/context-runtime-state.js",
),
vi.importActual<typeof import("../src/agents/models-config-state.js")>(
"../src/agents/models-config-state.js",
),
vi.importActual<typeof import("../src/config/sessions/store-cache.js")>(
"../src/config/sessions/store-cache.js",
),
vi.importActual<typeof import("../src/config/sessions/store-lock-state.js")>(
"../src/config/sessions/store-lock-state.js",
),
vi.importActual<typeof import("../src/infra/file-lock.js")>("../src/infra/file-lock.js"),
vi.importActual<typeof import("../src/agents/session-write-lock.js")>(
"../src/agents/session-write-lock.js",
),
]);
return {
clearSessionStoreCaches,
drainFileLockStateForTest,
drainSessionStoreLockQueuesForTest,
resetContextWindowCacheForTest,
resetModelsJsonReadyCacheForTest,
drainSessionWriteLockStateForTest,
resetFileLockStateForTest,
resetSessionWriteLockStateForTest,
cleanupSessionStateForTest,
};
}
@@ -75,11 +95,6 @@ function getWorkerCleanupDeps(): Promise<WorkerCleanupDeps> {
return globalState[WORKER_CLEANUP_DEPS];
}
// Preload cleanup/runtime helpers before per-file vi.mock hoists run in
// non-isolated workers, otherwise test-scoped module mocks can leak into the
// shared cleanup dependency graph.
void getWorkerCleanupDeps();
const pickSendFn = (id: ChannelId, deps?: OutboundSendDeps) => {
return deps?.[id] as ((...args: unknown[]) => Promise<unknown>) | undefined;
};
@@ -355,12 +370,20 @@ beforeAll(() => {
afterEach(async () => {
const {
cleanupSessionStateForTest,
clearSessionStoreCaches,
drainFileLockStateForTest,
drainSessionStoreLockQueuesForTest,
drainSessionWriteLockStateForTest,
resetContextWindowCacheForTest,
resetFileLockStateForTest,
resetModelsJsonReadyCacheForTest,
resetSessionWriteLockStateForTest,
} = await getWorkerCleanupDeps();
await cleanupSessionStateForTest();
await drainSessionStoreLockQueuesForTest();
clearSessionStoreCaches();
await drainFileLockStateForTest();
await drainSessionWriteLockStateForTest();
resetFileLockStateForTest();
resetContextWindowCacheForTest();
resetModelsJsonReadyCacheForTest();
resetSessionWriteLockStateForTest();
@@ -368,9 +391,10 @@ afterEach(async () => {
});
afterAll(async () => {
const { cleanupSessionStateForTest, drainSessionWriteLockStateForTest } =
const { clearSessionStoreCaches, drainFileLockStateForTest, drainSessionWriteLockStateForTest } =
await getWorkerCleanupDeps();
await cleanupSessionStateForTest();
clearSessionStoreCaches();
await drainFileLockStateForTest();
await drainSessionWriteLockStateForTest();
testEnv.cleanup();
});