Fix/useStore

src/react/__tests__/useStore.test.tsx

@@ -90,6 +90,29 @@ describe('useStore', () => {
       ]
     `)
   })
+  it('should correct work, when store contains function', async () => {
+    const fn = jest.fn()
+    const changeStore = createEvent()
+    const $store = createStore(() => 0).on(changeStore, (_, p) => p)
+
+    const Display = () => {
+      const store = useStore($store)
+      fn()
+      return store()
+    }
+
+    await render(<Display />)
+
+    expect(container.firstChild).toMatchInlineSnapshot(`0`)
+    expect(fn).toHaveBeenCalledTimes(1)
+
+    await act(async () => {
+      changeStore(() => 1)
+    })
+
+    expect(container.firstChild).toMatchInlineSnapshot(`1`)
+    expect(fn).toHaveBeenCalledTimes(2)
+  })
   it('should subscribe before any react hook will change store', async () => {
     const fn = jest.fn()
     const fx = createEffect({

src/react/useStore.ts

@@ -6,7 +6,7 @@ import {throwError} from './throw'
 export function useStore<State>(store: Store<State>): State {
   if (!is.store(store)) throwError('expect useStore argument to be a store')
   const currentStore = React.useRef(store)
-  const setState = React.useState(store.getState())[1]
+  const setState = React.useReducer((_, action) => action, store.getState())[1];
   useIsomorphicLayoutEffect(() => {
     if (currentStore.current === store) {
       setState(store.getState())