ComfyUI_frontend/tests-ui/globalSetup.ts

module.exports = async function () {
  global.ResizeObserver = class ResizeObserver {
    observe() {}
    unobserve() {}
    disconnect() {}
  }

  require('reflect-metadata')
  const { nop } = require('./utils/nopProxy')
  global.enableWebGLCanvas = nop

  HTMLCanvasElement.prototype.getContext = nop

  localStorage['Comfy.Settings.Comfy.Logging.Enabled'] = 'false'

  jest.mock('@/services/dialogService', () => {
    return {
      showLoadWorkflowWarning: jest.fn(),
      showMissingModelsWarning: jest.fn(),
      showSettingsDialog: jest.fn(),
      showExecutionErrorDialog: jest.fn(),
      showTemplateWorkflowsDialog: jest.fn(),
      showPromptDialog: jest
        .fn()
        .mockImplementation((message, defaultValue) => {
          return Promise.resolve(defaultValue)
        })
    }
  })

  jest.mock('@/stores/toastStore', () => {
    return {
      useToastStore: () => ({
        addAlert: jest.fn()
      })
    }
  })

  jest.mock('@/stores/extensionStore', () => {
    return {
      useExtensionStore: () => ({
        registerExtension: jest.fn(),
        loadDisabledExtensionNames: jest.fn()
      })
    }
  })

  jest.mock('@/stores/workspaceStateStore', () => {
    return {
      useWorkspaceStore: () => ({
        shiftDown: false,
        spinner: false
      })
    }
  })

  jest.mock('vue-i18n', () => {
    return {
      useI18n: jest.fn()
    }
  })
}