Fix missing dispatcher test coverage

Author: kitten

src/__tests__/suspense.test.js

@@ -72,13 +72,21 @@ describe('renderPrepass', () => {
       const Outer = jest.fn(() => {
         const [state, setState] = useState('default')
 
-        const memoed = useMemo(() => state, [state])
+        const memoedA = useMemo(() => state, [state])
+        expect(memoedA).toBe(state)
+
+        // This is to test changing dependency arrays
+        const memoedB = useMemo(
+          () => state,
+          state === 'default' ? null : [state]
+        )
+        expect(memoedB).toBe(state)
+
         const ref = useRef('initial')
-        expect(memoed).toBe(state)
         expect(ref.current).toBe('initial')
 
         const value = getValue()
-        setState(value)
+        setState(() => value)
 
         return <Inner value={value} state={state} />
       })

src/__tests__/visitor.test.js

@@ -280,15 +280,20 @@ describe('visitElement', () => {
   it('renders function components', () => {
     const Test = () => {
       const [value, setValue] = useState('a')
-      if (value === 'a') setValue('b')
+      if (value === 'a') {
+        setValue('b')
+        setValue('c')
+        setValue('d')
+      }
+
       return <Noop>{value}</Noop>
     }
 
     const visitor = jest.fn()
     const children = visitElement(<Test />, [], visitor)
     expect(children.length).toBe(1)
     expect(children[0].type).toBe(Noop)
-    expect(children[0].props.children).toBe('b')
+    expect(children[0].props.children).toBe('d')
     expect(visitor).toHaveBeenCalledWith(<Test />)
   })
 

src/internals/__tests__/dispatcher.test.js

@@ -1,3 +1,4 @@
+import { setCurrentContextStore } from '../context'
 import { getCurrentIdentity, Dispatcher } from '../dispatcher'
 
 describe('getCurrentIdentity', () => {
@@ -6,6 +7,16 @@ describe('getCurrentIdentity', () => {
   })
 })
 
+describe('readContext', () => {
+  it('calls readContextValue', () => {
+    const map = new Map()
+    const ctx = {}
+    setCurrentContextStore(map)
+    map.set(ctx, 'value')
+    expect(Dispatcher.readContext(ctx)).toBe('value')
+  })
+})
+
 describe('useEffect', () => {
   it('is a noop', () => {
     expect(Dispatcher.useEffect).not.toThrow()