AFFiNE-Mirror/scripts/setup/vi-mock.ts

import { vi } from 'vitest';

vi.mock('lottie-web', () => ({
  default: {},
}));

vi.mock('@blocksuite/presets', () => ({
  AffineEditorContainer: vi.fn(),
  BiDirectionalLinkPanel: vi.fn(),
  DocMetaTags: vi.fn(),
  DocTitle: vi.fn(),
  EdgelessEditor: vi.fn(),
  PageEditor: vi.fn(),
}));

vi.mock('@blocksuite/presets/ai', () => ({
  AIProvider: {
    slots: new Proxy(
      {},
      {
        get: () => ({
          on: vi.fn(),
        }),
      }
    ),
    provide: vi.fn(),
  },
  AIEdgelessRootBlockSpec: {},
  AICodeBlockSpec: {},
  AIImageBlockSpec: {},
  AIParagraphBlockSpec: {},
  AIPageRootBlockSpec: {},
}));

if (typeof window !== 'undefined' && HTMLCanvasElement) {
  // @ts-expect-error
  HTMLCanvasElement.prototype.getContext = () => {
    return {
      fillRect: vi.fn(),
    };
  };
}