Add type to element in html pretty printer

jestjs · Apr 23, 2017 · baff25e · baff25e
1 parent e7bc468
commit baff25e
Showing 1 changed file with 4 additions and 4 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) {
+): string => {
+  if (element instanceof Text) {
     return element.data;
   }
 
-  if (element.nodeType === 8) {
+  if (element instanceof Comment) {
     return `<!--${element.data}-->`;
   }