Update test to use new utils

facebook · Apr 16, 2023 · 443ceca · 443ceca
1 parent 72a8ad7
commit 443ceca
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/packages/react-dom/src/__tests__/ReactDOMFizzServer-test.js b/packages/react-dom/src/__tests__/ReactDOMFizzServer-test.js
@@ -2348,7 +2348,7 @@ describe('ReactDOMFizzServer', () => {
     },
   );
 
-  it('can hydrate uSES in StrictMode with different client and server snapshot', async () => {
+  it.only('can hydrate uSES in StrictMode with different client and server snapshot', async () => {
     function subscribe() {
       return () => {};
     }
@@ -2365,7 +2365,7 @@ describe('ReactDOMFizzServer', () => {
         getClientSnapshot,
         getServerSnapshot,
       );
-      Scheduler.unstable_yieldValue(value);
+      Scheduler.log(value);
 
       return value;
     }
@@ -2380,12 +2380,16 @@ describe('ReactDOMFizzServer', () => {
       const {pipe} = renderToPipeableStream(element);
       pipe(writable);
     });
-    expect(Scheduler).toHaveYielded(['Nay!']);
 
-    ReactDOMClient.hydrateRoot(container, element);
-    expect(() => {
-      expect(Scheduler).toFlushAndYield(['Nay!', 'Yay!']);
-    }).toErrorDev([]);
+    assertLog(['Nay!']);
+    expect(getVisibleChildren(container)).toEqual('Nay!');
+
+    await clientAct(() => {
+      ReactDOMClient.hydrateRoot(container, element);
+    });
+
+    expect(getVisibleChildren(container)).toEqual('Yay!');
+    assertLog(['Nay!', 'Yay!']);
   });
 
   it(