test(types): fix and improve dts tests for dom-to-react.test.tsx

Author: remarkablemark

test/types/lib/dom-to-react.test.tsx

@@ -1,39 +1,40 @@
+import { DomElement } from 'domhandler';
 import { HTMLReactParserOptions } from 'html-react-parser';
 import domToReact from 'html-react-parser/lib/dom-to-react';
 import * as React from 'react';
 import htmlToDOM from 'html-dom-parser';
 
-/* $ExpectType ReactElement | ReactElement[] */
-domToReact(htmlToDOM('<div>text</div>'));
+// $ExpectType DomElement[]
+htmlToDOM('<div>text</div>');
 
-// `options.replace`
+// $ExpectType Element | Element[]
+domToReact(htmlToDOM('<div>text</div>'));
 
-// Return React.createElement from `replace`
+// $ExpectType Element | Element[]
 domToReact(htmlToDOM('<p id="replace">text</p>'), {
   replace: domNode => {
     if (domNode.attribs && domNode.attribs.id === 'replace') {
-      return React.createElement('span', {}, 'replaced');
+      return <span>replaced</span>;
     }
   }
 });
 
-// Return ReactElement
-const options: HTMLReactParserOptions = {
+let options: HTMLReactParserOptions;
+
+options = {
   replace({ attribs }) {
-    return attribs && attribs.id === 'remove' && <React.Fragment />;
+    return attribs && attribs.id === 'remove' && <></>;
   }
 };
 
+// $ExpectType Element | Element[]
 domToReact(htmlToDOM('<p><br id="remove"></p>'), options);
 
-// Return domhandler node
+// $ExpectType Element | Element[]
 domToReact(htmlToDOM('<a id="header" href="#">Heading</a>'), {
   replace: node => {
     if (node.attribs && node.attribs.id === 'header') {
-      return {
-        type: 'h1',
-        props: { children: 'Heading' }
-      };
+      return;
     }
   }
 });