test: provider & index

Author: chnliquan

test/index.spec.ts

@@ -0,0 +1,36 @@
+import { createStore, createModel } from '../src'
+import { Store } from '../src/core/Store'
+import { config, defaultStoreOptions } from './helper/shared'
+
+describe('entry test', () => {
+  it('createStore should be defined', () => {
+    expect(createStore).toBeDefined()
+  })
+
+  it('should return the instance of Store when call createStore', () => {
+    const store = createStore(
+      {
+        counter: config,
+      },
+      defaultStoreOptions
+    )
+
+    expect(store).toBeInstanceOf(Store)
+  })
+
+  it('createModel should be defined', () => {
+    expect(createModel).toBeDefined()
+  })
+
+  it('should return the default value when config is invalid', () => {
+    const store = createModel()({
+      state: {
+        count: 0,
+      },
+    })
+
+    expect(store.state).toEqual({ count: 0 })
+    expect(store.reducers).toEqual({})
+    expect(typeof store.effects).toBe('function')
+  })
+})

test/provider.spec.tsx

@@ -0,0 +1,156 @@
+import React from 'react'
+import { render } from '@testing-library/react'
+import { act } from '@testing-library/react-hooks'
+import { createStore } from '../src/index'
+import { counter } from './helper/model'
+import { createHook } from './helper/createHook'
+import { Counter } from './helper/CountClassComponent'
+
+describe('Provider test', () => {
+  it('should render correct when use provider in component', () => {
+    const store = createStore({
+      counter,
+    })
+    const { Provider, withProvider } = store
+
+    const Component: React.FC = () => <div></div>
+
+    expect(() => {
+      render(
+        <Provider>
+          <Component />
+        </Provider>
+      )
+    }).not.toThrow()
+
+    const WithProvider = withProvider(Component)
+
+    expect(() => {
+      render(<WithProvider />)
+    }).not.toThrow()
+  })
+
+  it('should throw error when call useModel without Provider wrapper', () => {
+    const originalError = console.error
+    console.error = jest.fn()
+
+    const store = createStore({
+      counter,
+    })
+    const { useModel } = store
+
+    const ErrorCounter: React.FC = () => {
+      const { state } = useModel('counter')
+      return <div>{state.count}</div>
+    }
+
+    expect(() => render(<ErrorCounter />)).toThrow()
+
+    console.error = originalError
+  })
+
+  it('should add the store to function component context', () => {
+    const store = createStore({
+      counter,
+    })
+    const { Provider, useModel } = store
+
+    const {
+      result: { current },
+    } = createHook(Provider, useModel, 'counter')
+
+    expect(current.state.count).toBe(0)
+    expect(current.reducers).toBeDefined()
+    expect(current.effects).toBeDefined()
+  })
+
+  it('should add the store to class component context', () => {
+    const store = createStore({
+      counter,
+    })
+    const { Provider, withModel } = store
+
+    const Component = withModel('counter')(Counter)
+
+    const wrapper = render(
+      <Provider>
+        <Component />
+      </Provider>
+    )
+
+    expect(wrapper.getByTestId('count').innerHTML).toBe('0')
+  })
+
+  it('should not reset store when component unmount', async () => {
+    const store = createStore({
+      counter,
+    })
+    const { Provider, useModel } = store
+    const { result, unmount } = createHook(Provider, useModel, 'counter')
+
+    act(() => {
+      expect(result.current.state.count).toBe(0)
+      result.current.reducers.increase()
+    })
+
+    expect(result.current.state.count).toBe(1)
+
+    unmount()
+
+    const {
+      result: { current },
+    } = createHook(Provider, useModel, 'counter')
+
+    expect(current.state.count).toBe(1)
+  })
+
+  it('setting autoReset to true, model should be reset when the component unmount', async () => {
+    const store = createStore({
+      counter,
+    }, {
+      autoReset: true
+    })
+    const { Provider, useModel } = store
+    const { result, unmount } = createHook(Provider, useModel, 'counter')
+
+    act(() => {
+      expect(result.current.state.count).toBe(0)
+      result.current.reducers.increase()
+    })
+
+    expect(result.current.state.count).toBe(1)
+
+    unmount()
+
+    const {
+      result: { current },
+    } = createHook(Provider, useModel, 'counter')
+
+    expect(current.state.count).toBe(0)
+  })
+
+  it('setting autoReset to specify model, should be reset when the component unmount', async () => {
+    const store = createStore({
+      counter,
+    }, {
+      autoReset: ['counter']
+    })
+    const { Provider, useModel } = store
+    const { result, unmount } = createHook(Provider, useModel, 'counter')
+
+    act(() => {
+      expect(result.current.state.count).toBe(0)
+      result.current.reducers.increase()
+    })
+
+    expect(result.current.state.count).toBe(1)
+
+    unmount()
+
+    const {
+      result: { current },
+    } = createHook(Provider, useModel, 'counter')
+
+    expect(current.state.count).toBe(0)
+  })
+})