Quick fix of #74

Author: danielkcz

src/index.js

@@ -1,34 +1,13 @@
 import React, { Suspense } from 'react'
-import { create, act } from 'react-test-renderer'
+import { act, create } from 'react-test-renderer'
 
-function TestHook({ callback, hookProps, children }) {
-  children(callback(hookProps))
-  return null
-}
-
-class ErrorBoundary extends React.Component {
-  constructor(props) {
-    super(props)
-    this.state = { hasError: false }
-  }
-
-  static getDerivedStateFromError() {
-    return { hasError: true }
-  }
-
-  componentDidCatch(error) {
-    this.props.onError(error)
-  }
-
-  componentDidUpdate(prevProps) {
-    if (this.props != prevProps && this.state.hasError) {
-      this.setState({ hasError: false })
-    }
-  }
-
-  render() {
-    return !this.state.hasError && this.props.children
+function TestHook({ callback, hookProps, onError, children }) {
+  try {
+    children(callback(hookProps))
+  } catch (err) {
+    onError(err)
   }
+  return null
 }
 
 function Fallback() {
@@ -77,13 +56,11 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
 
   const toRender = () =>
     wrapUiIfNeeded(
-      <ErrorBoundary onError={setError}>
-        <Suspense fallback={<Fallback />}>
-          <TestHook callback={callback} hookProps={hookProps.current}>
-            {setValue}
-          </TestHook>
-        </Suspense>
-      </ErrorBoundary>
+      <Suspense fallback={<Fallback />}>
+        <TestHook callback={callback} hookProps={hookProps.current} onError={setError}>
+          {setValue}
+        </TestHook>
+      </Suspense>
     )
 
   let testRenderer