1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import { StrictMode } from 'react'import { fireEvent, render } from '@testing-library/react'import { useAtom } from 'jotai'import { atomWithReducer } from 'jotai/utils'import { getTestProvider } from '../testUtils'
const Provider = getTestProvider()
it('atomWithReducer with optional action argument', async () => {  const reducer = (state: number, action?: 'INCREASE' | 'DECREASE') => {    switch (action) {      case 'INCREASE':        return state + 1      case 'DECREASE':        return state - 1      case undefined:        return state    }  }  const countAtom = atomWithReducer(0, reducer)
  const Parent = () => {    const [count, dispatch] = useAtom(countAtom)    return (      <>        <div>count: {count}</div>        <button onClick={() => dispatch('INCREASE')}>dispatch INCREASE</button>        <button onClick={() => dispatch('DECREASE')}>dispatch DECREASE</button>        <button onClick={() => dispatch()}>dispatch empty</button>      </>    )  }
  const { findByText, getByText } = render(    <StrictMode>      <Provider>        <Parent />      </Provider>    </StrictMode>  )
  await findByText('count: 0')
  fireEvent.click(getByText('dispatch INCREASE'))  await findByText('count: 1')
  fireEvent.click(getByText('dispatch empty'))  await findByText('count: 1')
  fireEvent.click(getByText('dispatch DECREASE'))  await findByText('count: 0')})
it('atomWithReducer with non-optional action argument', async () => {  const reducer = (state: number, action: 'INCREASE' | 'DECREASE') => {    switch (action) {      case 'INCREASE':        return state + 1      case 'DECREASE':        return state - 1    }  }  const countAtom = atomWithReducer(0, reducer)
  const Parent = () => {    const [count, dispatch] = useAtom(countAtom)    return (      <>        <div>count: {count}</div>        <button onClick={() => dispatch('INCREASE')}>dispatch INCREASE</button>        <button onClick={() => dispatch('DECREASE')}>dispatch DECREASE</button>      </>    )  }
  const { findByText, getByText } = render(    <StrictMode>      <Provider>        <Parent />      </Provider>    </StrictMode>  )
  await findByText('count: 0')
  fireEvent.click(getByText('dispatch INCREASE'))  await findByText('count: 1')
  fireEvent.click(getByText('dispatch DECREASE'))  await findByText('count: 0')})