From 9686e518bcf72ac3c5a05a12a373a89c4372b2b7 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Sun, 19 Apr 2026 04:48:52 +0100 Subject: [PATCH] test: share media generation reset helpers --- .../tools/music-generate-tool.status.test.ts | 14 ++++++---- src/agents/tools/music-generate-tool.test.ts | 22 ++++++++------- .../tools/video-generate-tool.status.test.ts | 14 ++++++---- src/agents/tools/video-generate-tool.test.ts | 28 ++++++++++--------- 4 files changed, 43 insertions(+), 35 deletions(-) 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();