Moved utilities into seperate files

Author: mpeyper

src/createContext.js

@@ -0,0 +1,15 @@
+import uuid from 'uuid-v4'
+
+export default () => {
+  const context = {
+    id: uuid(),
+    resolveComponent: (Component) => Component
+  }
+
+  context.update = (newValues) => {
+    Object.entries(newValues).forEach(([key, value]) => (context[key] = value))
+    return context
+  }
+
+  return context
+}

src/flushEffects.js

@@ -0,0 +1,9 @@
+import invariant from 'invariant'
+
+export default (context) => () => {
+  invariant(context.rerender, 'You must use the hook before effects can be flushed')
+
+  context.flushEffects()
+
+  return context.result
+}

src/index.js

@@ -1 +1,6 @@
-export * from './useHook'
+import createContext from './createContext'
+import useHookAdvanced from './useHookAdvanced'
+
+export { cleanup } from 'react-testing-library'
+
+export const useHook = (hook) => useHookAdvanced(hook, createContext())

src/update.js

@@ -0,0 +1,9 @@
+import invariant from 'invariant'
+
+export default (context) => (props) => {
+  invariant(context.rerender, 'You must use the hook before it can be updated')
+
+  context.rerender(props)
+
+  return context.result
+}

src/use.js

@@ -0,0 +1,23 @@
+import React from 'react'
+import { render } from 'react-testing-library'
+import invariant from 'invariant'
+
+export default (hook, context) => (props) => {
+  const HookHarness = (props) => {
+    context.update({ result: hook(props) })
+    return <div data-testid={context.id} />
+  }
+
+  const { queryByTestId, rerender } = render(context.resolveComponent(<HookHarness {...props} />))
+
+  const container = queryByTestId(context.id)
+
+  invariant(container !== null, 'Failed to render wrapper component')
+
+  context.update({
+    rerender: (newProps) => rerender(context.resolveComponent(<HookHarness {...newProps} />)),
+    flushEffects: () => context.rerender(props)
+  })
+
+  return context.result
+}

src/useHook.js

@@ -0,0 +1,13 @@
+import use from './use'
+import withContextProvider from './withContextProvider'
+import update from './update'
+import flushEffects from './flushEffects'
+
+export default (hook, context) => {
+  return {
+    use: use(hook, context),
+    withContextProvider: withContextProvider(hook, context),
+    update: update(context),
+    flushEffects: flushEffects(context)
+  }
+}

src/useHookAdvanced.js

@@ -0,0 +1,14 @@
+import React from 'react'
+import useHookAdvanced from './useHookAdvanced'
+
+export default (hook, context) => (ContextProvider, props) => {
+  const { resolveComponent } = context
+  return useHookAdvanced(
+    hook,
+    context.update({
+      resolveComponent: (Component) => (
+        <ContextProvider {...props}>{resolveComponent(Component)}</ContextProvider>
+      )
+    })
+  )
+}