Files
AFFiNE-Mirror/scripts/setup/vi-mock.ts
2024-12-13 10:49:35 +00:00

34 lines
694 B
TypeScript

import { vi } from 'vitest';
vi.mock('lottie-web', () => ({
default: {},
}));
vi.mock('@blocksuite/affine/presets', () => ({
AffineEditorContainer: vi.fn(),
BiDirectionalLinkPanel: vi.fn(),
DocMetaTags: vi.fn(),
DocTitle: vi.fn(),
EdgelessEditor: vi.fn(),
PageEditor: vi.fn(),
AIChatBlockSchema: {
version: 1,
model: {
version: 1,
flavour: 'affine:embed-ai-chat',
role: 'content',
children: [],
},
},
AIChatBlockSpec: {},
}));
if (typeof window !== 'undefined' && HTMLCanvasElement) {
// @ts-expect-error allow vitest global mock
HTMLCanvasElement.prototype.getContext = () => {
return {
fillRect: vi.fn(),
};
};
}