Try rewriting the test to see if it works in other Node envs

Author: sebmarkbage

packages/react-dom/src/__tests__/ReactDOMFizzServerNode-test.js

@@ -36,11 +36,8 @@ describe('ReactDOMFizzServer', () => {
     writable.on('error', error => {
       writable.error = error;
     });
-    return writable;
-  }
-
-  function waitUntilClose(writable) {
-    return new Promise(resolve => writable.on('close', resolve));
+    const closed = new Promise(resolve => writable.on('close', resolve));
+    return {writable, closed};
   }
 
   const theError = new Error('This is an error');
@@ -54,31 +51,32 @@ describe('ReactDOMFizzServer', () => {
 
   // @gate experimental
   it('should call pipeToNodeWritable', () => {
-    const writable = getTestWritable();
+    const {writable} = getTestWritable();
     ReactDOMFizzServer.pipeToNodeWritable(<div>hello world</div>, writable);
     jest.runAllTimers();
     expect(writable.result).toBe('<div>hello world</div>');
   });
 
   // @gate experimental
   it('should error the stream when an error is thrown at the root', async () => {
-    const writable = getTestWritable();
+    const {writable, closed} = getTestWritable();
     ReactDOMFizzServer.pipeToNodeWritable(
       <div>
         <Throw />
       </div>,
       writable,
     );
+    jest.runAllTimers();
 
-    await waitUntilClose(writable);
+    await closed;
 
     expect(writable.error).toBe(theError);
     expect(writable.result).toBe('');
   });
 
   // @gate experimental
   it('should error the stream when an error is thrown inside a fallback', async () => {
-    const writable = getTestWritable();
+    const {writable, closed} = getTestWritable();
     ReactDOMFizzServer.pipeToNodeWritable(
       <div>
         <Suspense fallback={<Throw />}>
@@ -88,15 +86,17 @@ describe('ReactDOMFizzServer', () => {
       writable,
     );
 
-    await waitUntilClose(writable);
+    jest.runAllTimers();
+
+    await closed;
 
     expect(writable.error).toBe(theError);
     expect(writable.result).toBe('');
   });
 
   // @gate experimental
   it('should not error the stream when an error is thrown inside suspense boundary', async () => {
-    const writable = getTestWritable();
+    const {writable, closed} = getTestWritable();
     ReactDOMFizzServer.pipeToNodeWritable(
       <div>
         <Suspense fallback={<div>Loading</div>}>
@@ -106,7 +106,9 @@ describe('ReactDOMFizzServer', () => {
       writable,
     );
 
-    await waitUntilClose(writable);
+    jest.runAllTimers();
+
+    await closed;
 
     expect(writable.error).toBe(undefined);
     expect(writable.result).toContain('Loading');