fix: mock not complete

zzxming · Sep 15, 2024 · 0973942 · 0973942
1 parent 35d6dd8
commit 0973942
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/packages/utils/__tests__/raf.test.ts b/packages/utils/__tests__/raf.test.ts
@@ -2,11 +2,15 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { caf, raf } from '..';
 
 let isServerMocked = true;
-vi.mock('../types', () => ({
-  get isServer() {
-    return isServerMocked;
-  },
-}));
+vi.mock('../types', async (importOriginal) => {
+  const mod = await importOriginal<typeof import('@cdx-component/hooks/use-support-touch')>();
+  return {
+    ...mod,
+    get isServer() {
+      return isServerMocked;
+    },
+  };
+});
 describe('RAF and CAF Functions', () => {
   beforeEach(() => {
     vi.useFakeTimers();