diff --git a/src/components/image-editor/useImageCanvasGenerationWorkflow.test.tsx b/src/components/image-editor/useImageCanvasGenerationWorkflow.test.tsx index 1f5199fa8..bc07c6f09 100644 --- a/src/components/image-editor/useImageCanvasGenerationWorkflow.test.tsx +++ b/src/components/image-editor/useImageCanvasGenerationWorkflow.test.tsx @@ -22,6 +22,8 @@ const generateEditorSoundEffectMock = vi.hoisted(() => vi.fn()); const generateEditorBackgroundMusicMock = vi.hoisted(() => vi.fn()); const editEditorImageMock = vi.hoisted(() => vi.fn()); const renderCropExpandImageMock = vi.hoisted(() => vi.fn()); +const removeImageBackgroundMock = vi.hoisted(() => vi.fn()); +const resolveEditorImageReferenceDataUrlMock = vi.hoisted(() => vi.fn()); vi.mock('../../services/image-editor/editorImageReference', async () => { const actual = await vi.importActual< @@ -29,7 +31,7 @@ vi.mock('../../services/image-editor/editorImageReference', async () => { >('../../services/image-editor/editorImageReference'); return { ...actual, - resolveEditorImageReferenceDataUrl: vi.fn(async (src: string) => src), + resolveEditorImageReferenceDataUrl: resolveEditorImageReferenceDataUrlMock, resolveEditorImageReferenceDataUrlForGeneration: vi.fn( async (src: string) => src, ), @@ -57,6 +59,7 @@ vi.mock('./ImageCanvasRasterEditModel', async () => { >('./ImageCanvasRasterEditModel'); return { ...actual, + removeImageBackground: removeImageBackgroundMock, renderCropExpandImage: renderCropExpandImageMock, }; }); @@ -390,6 +393,12 @@ function GenerationWorkflowHarness({ +