From 58cb5b6d6bfac0e06d63235b188d1df981e56ef1 Mon Sep 17 00:00:00 2001
From: neverland <chenjiahan@youzan.com>
Date: Thu, 11 Feb 2021 12:19:21 +0800
Subject: [PATCH] test: add vtu style plugin (#8132)

---
 test/index.ts  |  2 ++
 test/plugin.ts | 26 ++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 test/plugin.ts

diff --git a/test/index.ts b/test/index.ts
index 3db455b45..540f8ee38 100644
--- a/test/index.ts
+++ b/test/index.ts
@@ -1,3 +1,5 @@
+import './plugin';
+
 // promisify setTimeout
 export function later(delay = 0): Promise<void> {
   return new Promise((resolve) => {
diff --git a/test/plugin.ts b/test/plugin.ts
new file mode 100644
index 000000000..c3a1241a1
--- /dev/null
+++ b/test/plugin.ts
@@ -0,0 +1,26 @@
+/* eslint-disable max-classes-per-file */
+import { ComponentPublicInstance } from 'vue';
+import { config, VueWrapper, DOMWrapper } from '@vue/test-utils';
+
+declare module '@vue/test-utils' {
+  // eslint-disable-next-line
+  export class DOMWrapper<ElementType extends Element> {
+    style: CSSStyleDeclaration;
+  }
+
+  // eslint-disable-next-line
+  class VueWrapper<T extends ComponentPublicInstance> {
+    style: CSSStyleDeclaration;
+  }
+}
+
+const stylePlugin = (
+  wrapper: VueWrapper<ComponentPublicInstance> | DOMWrapper<Element>
+) => {
+  return {
+    style: (wrapper.element as HTMLElement).style,
+  };
+};
+
+config.plugins.DOMWrapper.install(stylePlugin);
+config.plugins.VueWrapper.install(stylePlugin);