Fix gates

sebmarkbage · sebmarkbage · commit 0fddb4e8e215 · 2024-04-15T23:07:59.000-04:00
diff --git a/packages/react-client/src/__tests__/ReactFlight-test.js b/packages/react-client/src/__tests__/ReactFlight-test.js
@@ -2347,8 +2347,10 @@ describe('ReactFlight', () => {
       },
     );
 
-    // Wait for the iterator to finish
-    await iteratorPromise;
+    if (gate(flag => flag.enableFlightReadableStream)) {
+      // Wait for the iterator to finish
+      await iteratorPromise;
+    }
     await 0; // One more tick for the return value / closing.
 
     const transport = ReactNoopFlightServer.render(
diff --git a/packages/react-server-dom-webpack/src/__tests__/ReactFlightDOMEdge-test.js b/packages/react-server-dom-webpack/src/__tests__/ReactFlightDOMEdge-test.js
@@ -454,6 +454,7 @@ describe('ReactFlightDOMEdge', () => {
     expect(result.get('value')).toBe('hello');
   });
 
+  // @gate enableFlightReadableStream
   it('can pass an async import to a ReadableStream while enqueuing in order', async () => {
     let resolve;
     const promise = new Promise(r => (resolve = r));
@@ -496,6 +497,7 @@ describe('ReactFlightDOMEdge', () => {
     expect(await reader.read()).toEqual({value: undefined, done: true});
   });
 
+  // @gate enableFlightReadableStream
   it('can pass an async import a AsyncIterable while allowing peaking at future values', async () => {
     let resolve;
     const promise = new Promise(r => (resolve = r));
diff --git a/packages/react-server-dom-webpack/src/__tests__/ReactFlightDOMReplyEdge-test.js b/packages/react-server-dom-webpack/src/__tests__/ReactFlightDOMReplyEdge-test.js
@@ -109,33 +109,35 @@ describe('ReactFlightDOMReplyEdge', () => {
     expect(await result.arrayBuffer()).toEqual(await blob.arrayBuffer());
   });
 
-  it('can transport FormData (blobs)', async () => {
-    const bytes = new Uint8Array([
-      123, 4, 10, 5, 100, 255, 244, 45, 56, 67, 43, 124, 67, 89, 100, 20,
-    ]);
-    const blob = new Blob([bytes, bytes], {
-      type: 'application/x-test',
+  if (typeof FormData !== 'undefined' && typeof File !== 'undefined') {
+    it('can transport FormData (blobs)', async () => {
+      const bytes = new Uint8Array([
+        123, 4, 10, 5, 100, 255, 244, 45, 56, 67, 43, 124, 67, 89, 100, 20,
+      ]);
+      const blob = new Blob([bytes, bytes], {
+        type: 'application/x-test',
+      });
+
+      const formData = new FormData();
+      formData.append('hi', 'world');
+      formData.append('file', blob, 'filename.test');
+
+      expect(formData.get('file') instanceof File).toBe(true);
+      expect(formData.get('file').name).toBe('filename.test');
+
+      const body = await ReactServerDOMClient.encodeReply(formData);
+      const result = await ReactServerDOMServer.decodeReply(
+        body,
+        webpackServerMap,
+      );
+
+      expect(result instanceof FormData).toBe(true);
+      expect(result.get('hi')).toBe('world');
+      const resultBlob = result.get('file');
+      expect(resultBlob instanceof Blob).toBe(true);
+      expect(resultBlob.name).toBe('filename.test'); // In this direction we allow file name to pass through but not other direction.
+      expect(resultBlob.size).toBe(bytes.length * 2);
+      expect(await resultBlob.arrayBuffer()).toEqual(await blob.arrayBuffer());
     });
-
-    const formData = new FormData();
-    formData.append('hi', 'world');
-    formData.append('file', blob, 'filename.test');
-
-    expect(formData.get('file') instanceof File).toBe(true);
-    expect(formData.get('file').name).toBe('filename.test');
-
-    const body = await ReactServerDOMClient.encodeReply(formData);
-    const result = await ReactServerDOMServer.decodeReply(
-      body,
-      webpackServerMap,
-    );
-
-    expect(result instanceof FormData).toBe(true);
-    expect(result.get('hi')).toBe('world');
-    const resultBlob = result.get('file');
-    expect(resultBlob instanceof Blob).toBe(true);
-    expect(resultBlob.name).toBe('filename.test'); // In this direction we allow file name to pass through but not other direction.
-    expect(resultBlob.size).toBe(bytes.length * 2);
-    expect(await resultBlob.arrayBuffer()).toEqual(await blob.arrayBuffer());
-  });
+  }
 });
