diff --git a/src/agents/tools/music-generate-tool.status.test.ts b/src/agents/tools/music-generate-tool.status.test.ts index b201af9e118..a4cd4ad6fdf 100644 --- a/src/agents/tools/music-generate-tool.status.test.ts +++ b/src/agents/tools/music-generate-tool.status.test.ts @@ -12,13 +12,15 @@ const taskRuntimeInternalMocks = vi.hoisted(() => ({ vi.mock("../../tasks/runtime-internal.js", () => taskRuntimeInternalMocks); +function resetMusicStatusMocks() { + vi.restoreAllMocks(); + vi.spyOn(musicGenerationRuntime, "listRuntimeMusicGenerationProviders").mockReturnValue([]); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); +} + describe("createMusicGenerateTool status actions", () => { - beforeEach(() => { - vi.restoreAllMocks(); - vi.spyOn(musicGenerationRuntime, "listRuntimeMusicGenerationProviders").mockReturnValue([]); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); - }); + beforeEach(resetMusicStatusMocks); afterEach(() => { vi.unstubAllEnvs(); diff --git a/src/agents/tools/music-generate-tool.test.ts b/src/agents/tools/music-generate-tool.test.ts index 377d00e21de..fff377d1ccd 100644 --- a/src/agents/tools/music-generate-tool.test.ts +++ b/src/agents/tools/music-generate-tool.test.ts @@ -104,17 +104,19 @@ function asConfig(value: unknown): OpenClawConfig { return value as OpenClawConfig; } +function resetMusicGenerateMocks() { + vi.restoreAllMocks(); + vi.spyOn(musicGenerationRuntime, "listRuntimeMusicGenerationProviders").mockReturnValue([]); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); + taskExecutorMocks.createRunningTaskRun.mockReset(); + taskExecutorMocks.completeTaskRunByRunId.mockReset(); + taskExecutorMocks.failTaskRunByRunId.mockReset(); + taskExecutorMocks.recordTaskRunProgressByRunId.mockReset(); +} + describe("createMusicGenerateTool", () => { - beforeEach(() => { - vi.restoreAllMocks(); - vi.spyOn(musicGenerationRuntime, "listRuntimeMusicGenerationProviders").mockReturnValue([]); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); - taskExecutorMocks.createRunningTaskRun.mockReset(); - taskExecutorMocks.completeTaskRunByRunId.mockReset(); - taskExecutorMocks.failTaskRunByRunId.mockReset(); - taskExecutorMocks.recordTaskRunProgressByRunId.mockReset(); - }); + beforeEach(resetMusicGenerateMocks); afterEach(() => { vi.unstubAllEnvs(); diff --git a/src/agents/tools/video-generate-tool.status.test.ts b/src/agents/tools/video-generate-tool.status.test.ts index 86e38ed0878..480d10054a7 100644 --- a/src/agents/tools/video-generate-tool.status.test.ts +++ b/src/agents/tools/video-generate-tool.status.test.ts @@ -12,13 +12,15 @@ const taskRuntimeInternalMocks = vi.hoisted(() => ({ vi.mock("../../tasks/runtime-internal.js", () => taskRuntimeInternalMocks); +function resetVideoStatusMocks() { + vi.restoreAllMocks(); + vi.spyOn(videoGenerationRuntime, "listRuntimeVideoGenerationProviders").mockReturnValue([]); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); +} + describe("createVideoGenerateTool status actions", () => { - beforeEach(() => { - vi.restoreAllMocks(); - vi.spyOn(videoGenerationRuntime, "listRuntimeVideoGenerationProviders").mockReturnValue([]); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); - }); + beforeEach(resetVideoStatusMocks); afterEach(() => { vi.unstubAllEnvs(); diff --git a/src/agents/tools/video-generate-tool.test.ts b/src/agents/tools/video-generate-tool.test.ts index 405c4d6f50b..79a733eb031 100644 --- a/src/agents/tools/video-generate-tool.test.ts +++ b/src/agents/tools/video-generate-tool.test.ts @@ -10,10 +10,10 @@ const taskRuntimeInternalMocks = vi.hoisted(() => ({ })); const taskExecutorMocks = vi.hoisted(() => ({ - createRunningTaskRun: vi.fn(), - completeTaskRunByRunId: vi.fn(), - failTaskRunByRunId: vi.fn(), recordTaskRunProgressByRunId: vi.fn(), + failTaskRunByRunId: vi.fn(), + completeTaskRunByRunId: vi.fn(), + createRunningTaskRun: vi.fn(), })); vi.mock("../../tasks/runtime-internal.js", () => taskRuntimeInternalMocks); @@ -70,17 +70,19 @@ function mockSavedVideoResult(fileName = "out.mp4") { return generateSpy; } +function resetVideoGenerateMocks() { + vi.restoreAllMocks(); + vi.spyOn(videoGenerationRuntime, "listRuntimeVideoGenerationProviders").mockReturnValue([]); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); + taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); + taskExecutorMocks.createRunningTaskRun.mockReset(); + taskExecutorMocks.completeTaskRunByRunId.mockReset(); + taskExecutorMocks.failTaskRunByRunId.mockReset(); + taskExecutorMocks.recordTaskRunProgressByRunId.mockReset(); +} + describe("createVideoGenerateTool", () => { - beforeEach(() => { - vi.restoreAllMocks(); - vi.spyOn(videoGenerationRuntime, "listRuntimeVideoGenerationProviders").mockReturnValue([]); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset(); - taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]); - taskExecutorMocks.createRunningTaskRun.mockReset(); - taskExecutorMocks.completeTaskRunByRunId.mockReset(); - taskExecutorMocks.failTaskRunByRunId.mockReset(); - taskExecutorMocks.recordTaskRunProgressByRunId.mockReset(); - }); + beforeEach(resetVideoGenerateMocks); afterEach(() => { vi.unstubAllEnvs();