Add type to element in html pretty printer

jestjs · Apr 23, 2017 · 7723299 · 7723299
1 parent e7bc468
commit 7723299
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/pretty-format/src/plugins/HTMLElement.js b/packages/pretty-format/src/plugins/HTMLElement.js
@@ -57,17 +57,17 @@ function printAttributes(attributes, print, indent, colors, opts) {
 }
 
 const print = (
-  element: any,
+  element: HTMLElement | Text | Comment,
   print: Print,
   indent: Indent,
   opts: Options,
   colors: Colors,
 ) => {
-  if (element.nodeType === 3) {
+  if (element instanceof Text) {
     return element.data;
   }
 
-  if (element.nodeType === 8) {
+  if (element instanceof Comment) {
     return `<!--${element.data}-->`;
   }