feat: useUpdateEffect hook

src/useUpdateEffect.ts

@@ -0,0 +1,14 @@
+import { DependencyList, EffectCallback, useEffect } from 'react';
+import { useFirstMountState } from './useFirstMountState';
+
+export function useUpdateEffect(effect: EffectCallback, deps?: DependencyList): void {
+  const isFirstMount = useFirstMountState();
+
+  // eslint-disable-next-line consistent-return
+  useEffect(() => {
+    if (!isFirstMount) {
+      return effect();
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, deps);
+}

tests/dom/useUpdateEffect.test.ts

@@ -0,0 +1,43 @@
+import { renderHook } from '@testing-library/react-hooks/dom';
+import { useUpdateEffect } from '../../src/useUpdateEffect';
+
+describe('useUpdateEffect', () => {
+  it('should call effector only on updates (after first render)', () => {
+    const spy = jest.fn();
+
+    const { rerender, unmount } = renderHook(() => useUpdateEffect(spy));
+
+    expect(spy).toHaveBeenCalledTimes(0);
+
+    rerender();
+    expect(spy).toHaveBeenCalledTimes(1);
+
+    rerender();
+    expect(spy).toHaveBeenCalledTimes(2);
+
+    unmount();
+    expect(spy).toHaveBeenCalledTimes(2);
+  });
+
+  it('should accept dependencies as useEffect', () => {
+    const spy = jest.fn();
+
+    const { rerender, unmount } = renderHook(({ deps }) => useUpdateEffect(spy, deps), {
+      initialProps: { deps: [1, 2, 3] },
+    });
+
+    expect(spy).toHaveBeenCalledTimes(0);
+
+    rerender();
+    expect(spy).toHaveBeenCalledTimes(0);
+
+    rerender({ deps: [1, 2, 4] });
+    expect(spy).toHaveBeenCalledTimes(1);
+
+    rerender({ deps: [1, 2, 4] });
+    expect(spy).toHaveBeenCalledTimes(1);
+
+    unmount();
+    expect(spy).toHaveBeenCalledTimes(1);
+  });
+});

tests/ssr/useMountEffect.test.ts

@@ -5,9 +5,8 @@ describe('useMountEffect', () => {
   it('should call effector only on first render', () => {
     const spy = jest.fn();
 
-    const { result } = renderHook(() => useMountEffect(spy));
+    renderHook(() => useMountEffect(spy));
 
-    expect(result.current).toBe(undefined);
     expect(spy).toHaveBeenCalledTimes(0);
   });
 });

tests/ssr/useUnmountEffect.test.ts

@@ -5,9 +5,8 @@ describe('useUnmountEffect', () => {
   it('should call effector only when component unmounted', () => {
     const spy = jest.fn();
 
-    const { result } = renderHook(() => useUnmountEffect(spy));
+    renderHook(() => useUnmountEffect(spy));
 
-    expect(result.current).toBe(undefined);
     expect(spy).toHaveBeenCalledTimes(0);
   });
 });

tests/ssr/useUpdateEffect.test.ts

@@ -0,0 +1,12 @@
+import { renderHook } from '@testing-library/react-hooks/server';
+import { useUpdateEffect } from '../../src/useUpdateEffect';
+
+describe('useUpdateEffect', () => {
+  it('should not call effector on mount', () => {
+    const spy = jest.fn();
+
+    renderHook(() => useUpdateEffect(spy));
+
+    expect(spy).toHaveBeenCalledTimes(0);
+  });
+});