made the ref extensible for the forward ref scenario

facebook · Jan 22, 2025 · 8acf0a8 · 8acf0a8
1 parent 94387b6
commit 8acf0a8
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/packages/react-refresh/src/__tests__/ReactFresh-test.js b/packages/react-refresh/src/__tests__/ReactFresh-test.js
@@ -902,8 +902,22 @@ describe('ReactFresh', () => {
         await patch(() => {
           const Test = React.forwardRef((props, ref) => {
             const [count, setCount] = React.useState(0);
+            const handleClick = () => setCount(c => c + 1);
+
+            // Ensure ref is extensible
+            const divRef = React.useRef(null);
+            React.useEffect(() => {
+              if (ref) {
+                if (typeof ref === 'function') {
+                  ref(divRef.current);
+                } else if (Object.isExtensible(ref)) {
+                  ref.current = divRef.current;
+                }
+              }
+            }, [ref]);
+
             return (
-              <div ref={ref} onClick={() => setCount(c => c + 1)}>
+              <div ref={divRef} onClick={handleClick}>
                 count: {count}
               </div>
             );