From f9ac8528a8983a0c9a66cb35fb733dc917bd005b Mon Sep 17 00:00:00 2001
From: Ricky Hanlon <rickhanlonii@gmail.com>
Date: Tue, 26 Mar 2024 15:27:40 -0400
Subject: [PATCH] Fix ReactFresh-test.js

---
 .../react-refresh/src/__tests__/ReactFresh-test.js     | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/react-refresh/src/__tests__/ReactFresh-test.js b/packages/react-refresh/src/__tests__/ReactFresh-test.js
index 401e58c3badda..13ded584199e4 100644
--- 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');