test: getContext fix

Tencent · Jun 23, 2022 · 19723bb · 19723bb
1 parent 749a730
commit 19723bb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/test/snap/__snapshots__/ssr.test.js.snap b/test/snap/__snapshots__/ssr.test.js.snap
diff --git a/test/snap/csr.test.js b/test/snap/csr.test.js
@@ -1,5 +1,6 @@
 import glob from 'glob';
 import MockDate from 'mockdate';
+import { vi } from 'vitest';
 import { mount } from '@vue/test-utils';
 
 // 固定时间，当使用 new Date() 时，返回固定时间，防止“当前时间”的副作用影响，导致 snapshot 变更，mockdate 插件见 https://github.com/boblauer/MockDate
@@ -9,6 +10,8 @@ function runTest() {
   const files = glob.sync('./examples/**/demos/*.vue');
 
   describe('csr snapshot test', () => {
+    HTMLCanvasElement.prototype.getContext = vi.fn();
+
     files.forEach((file) => {
       it(`csr test ${file}`, async () => {
         const demo = await import(`../.${file}`);

diff --git a/test/snap/ssr.test.js b/test/snap/ssr.test.js
@@ -1,5 +1,6 @@
 import glob from 'glob';
 import MockDate from 'mockdate';
+import { vi } from 'vitest';
 import { config } from '@vue/test-utils';
 
 // 固定时间，当使用 new Date() 时，返回固定时间，防止“当前时间”的副作用影响，导致 snapshot 变更，mockdate 插件见 https://github.com/boblauer/MockDate
@@ -10,6 +11,8 @@ function runTest() {
   const { createSSRApp } = config.global;
 
   describe('ssr snapshot test', () => {
+    HTMLCanvasElement.prototype.getContext = vi.fn();
+
     files.forEach((file) => {
       it(`ssr test ${file}`, async () => {
         const demo = await import(`../.${file}`);