Address feedback

Author: trueadm

packages/react-dom/src/__tests__/ReactDOMServerPartialHydration-test.internal.js

@@ -2220,13 +2220,10 @@ describe('ReactDOMServerPartialHydration', () => {
 
     function Button() {
       const ref = React.useRef(null);
-      const effect = () => {
-        return setClick(ref.current, onEvent);
-      };
-      if (isServerRendering) {
-        React.useEffect(effect);
-      } else {
-        React.useLayoutEffect(effect);
+      if (!isServerRendering) {
+        React.useLayoutEffect(() => {
+          return setClick(ref.current, onEvent);
+        });
       }
       return <a ref={ref}>Click me</a>;
     }
@@ -2379,13 +2376,10 @@ describe('ReactDOMServerPartialHydration', () => {
     function Button() {
       const ref = React.useRef(null);
 
-      const effect = () => {
-        return setClick(ref.current, onEvent);
-      };
-      if (isServerRendering) {
-        React.useEffect(effect);
-      } else {
-        React.useLayoutEffect(effect);
+      if (!isServerRendering) {
+        React.useLayoutEffect(() => {
+          return setClick(ref.current, onEvent);
+        });
       }
 
       return <a ref={ref}>Click me</a>;

packages/react-dom/src/__tests__/ReactDOMServerSelectiveHydration-test.internal.js

@@ -354,15 +354,12 @@ describe('ReactDOMServerSelectiveHydration', () => {
     function Child({text}) {
       const ref = React.useRef(null);
       Scheduler.unstable_yieldValue(text);
-      const effect = () => {
-        return setClick(ref.current, () => {
-          Scheduler.unstable_yieldValue('Clicked ' + text);
+      if (!isServerRendering) {
+        React.useLayoutEffect(() => {
+          return setClick(ref.current, () => {
+            Scheduler.unstable_yieldValue('Clicked ' + text);
+          });
         });
-      };
-      if (isServerRendering) {
-        React.useEffect(effect);
-      } else {
-        React.useLayoutEffect(effect);
       }
 
       return <span ref={ref}>{text}</span>;
@@ -433,15 +430,12 @@ describe('ReactDOMServerSelectiveHydration', () => {
       }
       Scheduler.unstable_yieldValue(text);
 
-      const effect = () => {
-        return setClick(ref.current, () => {
-          Scheduler.unstable_yieldValue('Clicked ' + text);
+      if (!isServerRendering) {
+        React.useLayoutEffect(() => {
+          return setClick(ref.current, () => {
+            Scheduler.unstable_yieldValue('Clicked ' + text);
+          });
         });
-      };
-      if (isServerRendering) {
-        React.useEffect(effect);
-      } else {
-        React.useLayoutEffect(effect);
       }
 
       return <span ref={ref}>{text}</span>;
@@ -524,15 +518,12 @@ describe('ReactDOMServerSelectiveHydration', () => {
       }
       Scheduler.unstable_yieldValue(text);
 
-      const effect = () => {
-        return setClick(ref.current, () => {
-          Scheduler.unstable_yieldValue('Clicked ' + text);
+      if (!isServerRendering) {
+        React.useLayoutEffect(() => {
+          return setClick(ref.current, () => {
+            Scheduler.unstable_yieldValue('Clicked ' + text);
+          });
         });
-      };
-      if (isServerRendering) {
-        React.useEffect(effect);
-      } else {
-        React.useLayoutEffect(effect);
       }
       return <span ref={ref}>{text}</span>;
     }