issue-1147: Add option to pretty print html components

vuejs · May 9, 2019 · 76afc50 · 76afc50
1 parent 23e496b
commit 76afc50
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/packages/test-utils/package.json b/packages/test-utils/package.json
@@ -39,6 +39,7 @@
   },
   "dependencies": {
     "dom-event-types": "^1.0.0",
-    "lodash": "^4.17.4"
+    "lodash": "^4.17.4",
+    "pretty": "^2.0.0"
   }
 }
diff --git a/packages/test-utils/src/wrapper.js b/packages/test-utils/src/wrapper.js
@@ -1,6 +1,7 @@
 // @flow
 
 import Vue from 'vue'
+import pretty from 'pretty';
 import getSelector from './get-selector'
 import { REF_SELECTOR, FUNCTIONAL_OPTIONS, VUE_VERSION } from 'shared/consts'
 import config from './config'
@@ -221,7 +222,10 @@ export default class Wrapper implements BaseWrapper {
   /**
    * Returns HTML of element as a string
    */
-  html(): string {
+  html(options?: HtmlOptions): string {
+    if (options && options.prettyPrint) {
+      return pretty(this.element.outerHTML);
+    }
     return this.element.outerHTML
   }
 

diff --git a/packages/test-utils/types/index.d.ts b/packages/test-utils/types/index.d.ts
@@ -39,6 +39,13 @@ type NameSelector = {
   name: string
 }
 
+/**
+ * Options when printing html components as strings
+ */
+export class HtmlOptions {
+  prettyPrint: boolean
+}
+
 /**
  * Base class of Wrapper and WrapperArray
  * It has common methods on both Wrapper and WrapperArray