vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 09:10:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: share media generation reset helpers

Browse Source
This commit is contained in:
Peter Steinberger
2026-04-19 04:48:52 +01:00
parent 5ca33f7cb4
commit 9686e518bc
4 changed files with 43 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/agents/tools/music-generate-tool.status.test.ts
View File

@@ -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();

22
src/agents/tools/music-generate-tool.test.ts
View File

@@ -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();

14
src/agents/tools/video-generate-tool.status.test.ts
View File

@@ -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();

28
src/agents/tools/video-generate-tool.test.ts
View File

@@ -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();