Update test cases to handle new logic in fetchUtils

Author: Lei-k

packages/replay-internal/test/unit/coreHandlers/handleNetworkBreadcrumbs.test.ts

@@ -40,6 +40,8 @@ function getMockResponse(contentLength?: string, body?: string, headers?: Record
     ...headers,
   };
 
+  const encoder = new TextEncoder();
+
   const response = {
     headers: {
       has: (prop: string) => {
@@ -49,6 +51,24 @@ function getMockResponse(contentLength?: string, body?: string, headers?: Record
         return internalHeaders[prop.toLowerCase() ?? ''];
       },
     },
+    body: {
+      getReader: () => {
+        return {
+          read: () => {
+            return Promise.resolve({
+              done: true,
+              value: encoder.encode(body),
+            });
+          },
+          cancel: async () => {
+            // noop
+          },
+          releaseLock: async () => {
+            // noop
+          },
+        };
+      },
+    },
     clone: () => response,
     text: () => Promise.resolve(body),
   } as unknown as Response;
@@ -741,6 +761,7 @@ other-header: test`;
       options.networkCaptureBodies = true;
 
       const largeBody = JSON.stringify({ a: LARGE_BODY });
+      const encoder = new TextEncoder();
 
       const breadcrumb: Breadcrumb = {
         category: 'fetch',
@@ -756,6 +777,24 @@ other-header: test`;
           get: () => '',
         },
         clone: () => mockResponse,
+        body: {
+          getReader: () => {
+            return {
+              read: () => {
+                return Promise.resolve({
+                  done: true,
+                  value: encoder.encode(largeBody),
+                });
+              },
+              cancel: async () => {
+                // noop
+              },
+              releaseLock: async () => {
+                // noop
+              },
+            };
+          },
+        },
         text: () => Promise.resolve(largeBody),
       } as unknown as Response;
 

packages/replay-internal/test/unit/coreHandlers/util/fetchUtils.test.ts

@@ -43,6 +43,19 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
     });
 
     it('works with text body', async () => {
+      const encoder = new TextEncoder();
+
+      const mockRead = vi
+        .fn()
+        .mockResolvedValueOnce({
+          value: encoder.encode('text body'),
+          done: false,
+        })
+        .mockResolvedValueOnce({
+          value: null,
+          done: true,
+        });
+
       const response = {
         headers: {
           has: () => {
@@ -52,6 +65,19 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
             return undefined;
           },
         },
+        body: {
+          getReader: () => {
+            return {
+              read: mockRead,
+              cancel: async (reason?: any) => {
+                mockRead.mockRejectedValue(new Error(reason));
+              },
+              releaseLock: async () => {
+                // noop
+              },
+            };
+          },
+        },
         clone: () => response,
         text: () => Promise.resolve('text body'),
       } as unknown as Response;
@@ -74,6 +100,8 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
     });
 
     it('works with body that fails', async () => {
+      const mockRead = vi.fn().mockRejectedValueOnce(new Error('cannot read'));
+
       const response = {
         headers: {
           has: () => {
@@ -83,6 +111,19 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
             return undefined;
           },
         },
+        body: {
+          getReader: () => {
+            return {
+              read: mockRead,
+              cancel: async (_?: any) => {
+                // noop
+              },
+              releaseLock: async () => {
+                // noop
+              },
+            };
+          },
+        },
         clone: () => response,
         text: () => Promise.reject('cannot read'),
       } as unknown as Response;
@@ -105,6 +146,12 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
     });
 
     it('works with body that times out', async () => {
+      const encoder = new TextEncoder();
+      const mockRead = vi.fn();
+
+      let cancelled = false;
+      let cancellReason = '';
+
       const response = {
         headers: {
           has: () => {
@@ -114,6 +161,38 @@ describe('Unit | coreHandlers | util | fetchUtils', () => {
             return undefined;
           },
         },
+        body: {
+          getReader: () => {
+            return {
+              read: async () => {
+                if (cancelled) {
+                  mockRead.mockRejectedValue(new Error(cancellReason));
+                } else {
+                  mockRead.mockResolvedValueOnce({
+                    value: encoder.encode('chunk'),
+                    done: false,
+                  });
+                }
+
+                await new Promise(res => {
+                  setTimeout(() => {
+                    res(1);
+                  }, 200);
+                });
+
+                // eslint-disable-next-line no-return-await
+                return await mockRead();
+              },
+              cancel: async (reason?: any) => {
+                cancelled = true;
+                cancellReason = reason;
+              },
+              releaseLock: async () => {
+                // noop
+              },
+            };
+          },
+        },
         clone: () => response,
         text: () => new Promise(resolve => setTimeout(() => resolve('text body'), 1000)),
       } as unknown as Response;