From 5d00db2143d7ee2efffe3a34305a04878f32ffe2 Mon Sep 17 00:00:00 2001 From: Jin Yi Date: Wed, 21 Jan 2026 13:41:33 +0900 Subject: [PATCH] test: add missing mocks to useJobMenu.test.ts - Add isCloud mock for Cloud environment handling - Add ComfyWorkflow export to workflowStore mock - Add executionStore mock for clearInitializationByPromptId Co-Authored-By: Claude Opus 4.5 --- src/composables/queue/useJobMenu.test.ts | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/composables/queue/useJobMenu.test.ts b/src/composables/queue/useJobMenu.test.ts index 1b339d09c..9913b3b3e 100644 --- a/src/composables/queue/useJobMenu.test.ts +++ b/src/composables/queue/useJobMenu.test.ts @@ -5,6 +5,10 @@ import type { Ref } from 'vue' import type { JobListItem } from '@/composables/queue/useJobList' import type { MenuEntry } from '@/composables/queue/useJobMenu' +vi.mock('@/platform/distribution/types', () => ({ + isCloud: false +})) + const downloadFileMock = vi.fn() vi.mock('@/base/common/downloadUtil', () => ({ downloadFile: (...args: any[]) => downloadFileMock(...args) @@ -55,7 +59,8 @@ const workflowStoreMock = { createTemporary: vi.fn() } vi.mock('@/platform/workflow/management/stores/workflowStore', () => ({ - useWorkflowStore: () => workflowStoreMock + useWorkflowStore: () => workflowStoreMock, + ComfyWorkflow: class {} })) const interruptMock = vi.fn() @@ -104,6 +109,13 @@ vi.mock('@/stores/queueStore', () => ({ useQueueStore: () => queueStoreMock })) +const executionStoreMock = { + clearInitializationByPromptId: vi.fn() +} +vi.mock('@/stores/executionStore', () => ({ + useExecutionStore: () => executionStoreMock +})) + const getJobWorkflowMock = vi.fn() vi.mock('@/services/jobOutputCache', () => ({ getJobWorkflow: (...args: any[]) => getJobWorkflowMock(...args)