Fix ReactFresh-test.js

facebook · Mar 26, 2024 · f9ac852 · f9ac852
1 parent 02ee4f1
commit f9ac852
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/react-refresh/src/__tests__/ReactFresh-test.js b/packages/react-refresh/src/__tests__/ReactFresh-test.js
@@ -79,6 +79,12 @@ describe('ReactFresh', () => {
     return Component;
   }
 
+  function patchSync(version) {
+    const Component = version();
+    ReactFreshRuntime.performReactRefresh();
+    return Component;
+  }
+
   function $RefreshReg$(type, id) {
     ReactFreshRuntime.register(type, id);
   }
@@ -2470,7 +2476,7 @@ describe('ReactFresh', () => {
     expect(el.firstChild).toBe(null); // Offscreen content not flushed yet.
 
     // Perform a hot update.
-    await patch(() => {
+    patchSync(() => {
       function Hello() {
         React.useLayoutEffect(() => {
           Scheduler.log('Hello#layout');
@@ -2509,7 +2515,7 @@ describe('ReactFresh', () => {
     expect(el.firstChild.style.color).toBe('red');
 
     // Hot reload while we're offscreen.
-    await patch(() => {
+    patchSync(() => {
       function Hello() {
         React.useLayoutEffect(() => {
           Scheduler.log('Hello#layout');