packages/hooks/useStateRef.test.ts

   1 import { act, renderHook } from '@testing-library/react-hooks';
   2
   3 import { useStateRef } from '@proton/hooks';
   4
   5 describe('useStateRef', () => {
   6     it('should update ref when state is updated', () => {
   7         const hook = renderHook(() => useStateRef(0));
   8
   9         act(() => {
  10             const [, dispatch] = hook.result.current;
  11             dispatch(1);
  12         });
  13
  14         const [state, , ref] = hook.result.current;
  15
  16         expect(state).toEqual(1);
  17         expect(ref.current).toEqual(1);
  18     });
  19
  20     it('should update ref when state is updated with function', () => {
  21         const hook = renderHook(() => useStateRef(0));
  22
  23         act(() => {
  24             const [, dispatch] = hook.result.current;
  25             dispatch((prev) => prev + 1);
  26         });
  27
  28         const [state, , ref] = hook.result.current;
  29
  30         expect(state).toEqual(1);
  31         expect(ref.current).toEqual(1);
  32     });
  33 });