test(ui/jest): add mockStubs method to resolve transition stub out issue

affects: @varlet/ui
varletjs · May 18, 2021 · c47aee2 · c47aee2
1 parent 4f4e290
commit c47aee2
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/varlet-ui/src/utils/jest.ts b/packages/varlet-ui/src/utils/jest.ts
@@ -1,5 +1,5 @@
 import { ComponentPublicInstance, nextTick } from 'vue'
-import { VueWrapper, DOMWrapper } from '@vue/test-utils'
+import { VueWrapper, DOMWrapper, config } from '@vue/test-utils'
 
 export const delay = (time: number) => new Promise((resolve) => setTimeout(resolve, time))
 
@@ -194,3 +194,15 @@ export function mockFileReader(url: string) {
     },
   }
 }
+
+export function mockStubs() {
+  const originStubs = config.global.stubs
+
+  config.global.stubs = {}
+
+  return {
+    mockRestore() {
+      config.global.stubs = originStubs
+    },
+  }
+}