Prevent useBeforeMount to invoke twice after change in useIsMounted

src/hooks/useBeforeMount/useBeforeMount.test.tsx

@@ -1,6 +1,7 @@
 import { jest } from '@jest/globals';
 import { renderHook } from '@testing-library/react';
-import { useEffect } from 'react';
+import { useEffect, useState } from 'react';
+import { useMount } from '../useMount/useMount.js';
 import { useBeforeMount } from './useBeforeMount.js';
 
 describe('useBeforeMount', () => {
@@ -59,4 +60,23 @@ describe('useBeforeMount', () => {
     rerender();
     expect(spy).toBeCalledTimes(1);
   });
+
+  it('should only execute once when setState is called during useMount', async () => {
+    const spy = jest.fn();
+    const { rerender } = renderHook(() => {
+      // eslint-disable-next-line react/hook-use-state
+      const [, setState] = useState(false);
+
+      useBeforeMount(spy);
+
+      useMount(() => {
+        setState(true);
+      });
+    });
+    expect(spy).toBeCalledTimes(1);
+
+    await Promise.resolve();
+    rerender();
+    expect(spy).toBeCalledTimes(1);
+  });
 });

src/hooks/useBeforeMount/useBeforeMount.ts

@@ -1,4 +1,5 @@
-import { useIsMounted } from '../useIsMounted/useIsMounted.js';
+import { useRef } from 'react';
+import { useMount } from '../useMount/useMount.js';
 
 /**
  * Executes a callback during the initial render, before mounting, but not during subsequent
@@ -11,8 +12,13 @@ import { useIsMounted } from '../useIsMounted/useIsMounted.js';
  * during subsequent renders.
  */
 export function useBeforeMount(callback: () => void): void {
-  const isMounted = useIsMounted();
-  if (!isMounted.current) {
+  const isBeforeMount = useRef(true);
+
+  if (isBeforeMount.current) {
     callback();
   }
+
+  useMount(() => {
+    isBeforeMount.current = false;
+  });
 }