1 files changed, 22 insertions, 10 deletions
diff --git a/remote/test/puppeteer/test/src/tracing.spec.ts b/remote/test/puppeteer/test/src/tracing.spec.ts
index 2c0a5aff19..7ee6d46192 100644
--- a/remote/test/puppeteer/test/src/tracing.spec.ts
+++ b/remote/test/puppeteer/test/src/tracing.spec.ts
@@ -8,6 +8,8 @@ import fs from 'fs';
 import path from 'path';
 
 import expect from 'expect';
+import * as utils from 'puppeteer-core/internal/common/util.js';
+import sinon from 'sinon';
 
 import {launch} from './mocha-utils.js';
 
@@ -113,16 +115,26 @@ describe('Tracing', function () {
     await page.tracing.start({screenshots: true});
     await page.goto(server.PREFIX + '/grid.html');
 
-    const oldBufferConcat = Buffer.concat;
-    try {
-      Buffer.concat = () => {
-        throw new Error('error');
-      };
-      const trace = await page.tracing.stop();
-      expect(trace).toEqual(undefined);
-    } finally {
-      Buffer.concat = oldBufferConcat;
-    }
+    const oldGetReadableAsBuffer = utils.getReadableAsBuffer;
+    sinon.stub(utils, 'getReadableAsBuffer').callsFake(() => {
+      return oldGetReadableAsBuffer({
+        getReader() {
+          return {
+            done: false,
+            read() {
+              if (!this.done) {
+                this.done = true;
+                return {done: false, value: 42};
+              }
+              return {done: true};
+            },
+          };
+        },
+      } as unknown as ReadableStream);
+    });
+
+    const trace = await page.tracing.stop();
+    expect(trace).toEqual(undefined);
   });
 
   it('should support a buffer without a path', async () => {