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({
+