diff --git a/src/utils.ts b/src/utils.ts index cc86f0c..d03253e 100644 --- a/src/utils.ts +++ b/src/utils.ts @@ -1,4 +1,5 @@ -import { FiberNode, FiberNodeisHTMLLike } from './mocked-types'; +import * as React from 'react'; +import { FiberNode, FiberNodeisHTMLLike, FiberNodeForFunctionComponent, FiberNodeForComponentClass } from './mocked-types'; function isNodeHtmlLike(node: FiberNode): node is FiberNodeisHTMLLike { return (typeof node.type === "string") || node.type === null; @@ -8,4 +9,12 @@ function isNodeNotHtmlLike(node: FiberNode): node is Exclude