Add additional test for hooks

Author: kitten

package.json

@@ -32,7 +32,7 @@
     ]
   },
   "lint-staged": {
-    "**/*.{js}": [
+    "**/*.js": [
       "flow focus-check",
       "prettier --write",
       "git add"

src/__tests__/suspense.test.js

@@ -1,4 +1,13 @@
-import React, { Fragment, Component, createElement, createContext, useContext, useState } from 'react'
+import React, {
+  Fragment,
+  Component,
+  createElement,
+  createContext,
+  useContext,
+  useState,
+  useMemo,
+  useRef
+} from 'react'
 
 import renderPrepass from '..'
 
@@ -44,6 +53,36 @@ describe('renderPrepass', () => {
       })
     })
 
+    it('preserves state correctly across suspensions', () => {
+      const getValue = jest.fn()
+        .mockImplementationOnce(() => { throw Promise.resolve() })
+        .mockImplementation(() => 'test')
+
+      const Inner = jest.fn(props => {
+        expect(props.value).toBe('test')
+        expect(props.state).toBe('test')
+      })
+
+      const Outer = jest.fn(() => {
+        const [state, setState] = useState('default')
+
+        const memoed = useMemo(() => state, [state])
+        const ref = useRef('initial')
+        expect(memoed).toBe(state)
+        expect(ref.current).toBe('initial')
+
+        const value = getValue()
+        setState(value)
+
+        return <Inner value={value} state={state} />
+      })
+
+      return renderPrepass(<Outer />).then(() => {
+        expect(Outer).toHaveBeenCalledTimes(3 * 3 * 3 /* welp */)
+        expect(Inner).toHaveBeenCalledTimes(2)
+      })
+    })
+
     it('ignores thrown non-promises', () => {
       const Outer = () => { throw new Error('test') }
       const render$ = renderPrepass(<Outer />)