From 808b760c79e61ba1a24cab066348defb333d8976 Mon Sep 17 00:00:00 2001
From: Tim Yung <yungsters@meta.com>
Date: Mon, 12 Aug 2024 16:02:59 -0700
Subject: [PATCH] Unit Test for `findNodeHandle` Error Behavior

---
 .../__tests__/ReactFabric-test.internal.js    | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/packages/react-native-renderer/src/__tests__/ReactFabric-test.internal.js b/packages/react-native-renderer/src/__tests__/ReactFabric-test.internal.js
index d55be5efd24bf..1c638e8aa51f7 100644
--- a/packages/react-native-renderer/src/__tests__/ReactFabric-test.internal.js
+++ b/packages/react-native-renderer/src/__tests__/ReactFabric-test.internal.js
@@ -1306,6 +1306,50 @@ describe('ReactFabric', () => {
     );
   });
 
+  it('findNodeHandle errors when called from render', async () => {
+    class TestComponent extends React.Component {
+      render() {
+        ReactFabric.findNodeHandle(this);
+        return null;
+      }
+    }
+    await expect(async () => {
+      await act(() => {
+        ReactFabric.render(
+          <TestComponent />,
+          11,
+          null,
+          true,
+        );
+      });
+    }).toErrorDev([
+      'TestComponent is accessing findNodeHandle inside its render(). ' +
+      'render() should be a pure function of props and state. It should ' +
+      'never access something that requires stale data from the previous ' +
+      'render, such as refs. Move this logic to componentDidMount and ' +
+      'componentDidUpdate instead.',
+    ]);
+  });
+
+  it("findNodeHandle doesn't error when called outside render", async () => {
+    class TestComponent extends React.Component {
+      render() {
+        return null;
+      }
+      componentDidMount() {
+        ReactFabric.findNodeHandle(this);
+      }
+    }
+    await act(() => {
+      ReactFabric.render(
+        <TestComponent />,
+        11,
+        null,
+        true,
+      );
+    });
+  });
+
   it('should no-op if calling sendAccessibilityEvent on unmounted refs', async () => {
     const View = createReactNativeComponentClass('RCTView', () => ({
       validAttributes: {foo: true},