issue-1147: Add test for pretty print html

addisonssense · addisonssense · commit 23dc630ef1a6 · 2019-05-09T14:24:40.000-04:00
diff --git a/packages/test-utils/src/wrapper.js b/packages/test-utils/src/wrapper.js
@@ -1,7 +1,7 @@
 // @flow
 
 import Vue from 'vue'
-import pretty from 'pretty';
+import pretty from 'pretty'
 import getSelector from './get-selector'
 import { REF_SELECTOR, FUNCTIONAL_OPTIONS, VUE_VERSION } from 'shared/consts'
 import config from './config'
@@ -224,7 +224,7 @@ export default class Wrapper implements BaseWrapper {
    */
   html(options?: HtmlOptions): string {
     if (options && options.prettyPrint) {
-      return pretty(this.element.outerHTML);
+      return pretty(this.element.outerHTML)
     }
     return this.element.outerHTML
   }
diff --git a/test/specs/wrapper/html.spec.js b/test/specs/wrapper/html.spec.js
@@ -39,4 +39,21 @@ describeWithShallowAndMount('html', mountingMethod => {
     const wrapper = mountingMethod(compiled)
     expect(wrapper.html()).to.equal(expectedHtml)
   })
+
+  it('returns a Wrappers HTML as a pretty printed string', () => {
+    const expectedHtml =
+      '<body>\n' +
+      '  <div>\n' +
+      '    <ul>\n' +
+      '      <li></li>\n' +
+      '      <li></li>\n' +
+      '    </ul>\n' +
+      '  </div>\n' +
+      '</body>'
+
+    const compiled = compileToFunctions(expectedHtml)
+    const wrapper = mountingMethod(compiled)
+    const options = { prettyPrint: true }
+    expect(wrapper.html(options)).to.equal(expectedHtml)
+  })
 })
