fix: Make ResponseWrapper an instance of Response

Author: FrederikBolding

packages/snaps-execution-environments/coverage.json

@@ -1,6 +1,6 @@
 {
   "branches": 80,
   "functions": 90.06,
-  "lines": 90.75,
-  "statements": 90.12
+  "lines": 90.77,
+  "statements": 90.15
 }

packages/snaps-execution-environments/src/common/BaseSnapExecutor.test.browser.ts

@@ -999,6 +999,57 @@ describe('BaseSnapExecutor', () => {
     });
   });
 
+  it('uses a response for fetch that works with instanceof', async () => {
+    const CODE = `
+      module.exports.onRpcRequest = () => fetch('https://metamask.io').then(res => res instanceof Response);
+    `;
+
+    const fetchSpy = spy(globalThis, 'fetch');
+
+    fetchSpy.mockImplementation(async () => {
+      return new Response('foo');
+    });
+
+    const executor = new TestSnapExecutor();
+    await executor.executeSnap(1, MOCK_SNAP_ID, CODE, ['fetch', 'Response']);
+
+    expect(await executor.readCommand()).toStrictEqual({
+      jsonrpc: '2.0',
+      id: 1,
+      result: 'OK',
+    });
+
+    await executor.writeCommand({
+      jsonrpc: '2.0',
+      id: 2,
+      method: 'snapRpc',
+      params: [
+        MOCK_SNAP_ID,
+        HandlerType.OnRpcRequest,
+        MOCK_ORIGIN,
+        { jsonrpc: '2.0', method: '', params: [] },
+      ],
+    });
+
+    expect(await executor.readCommand()).toStrictEqual({
+      jsonrpc: '2.0',
+      method: 'OutboundRequest',
+      params: { source: 'fetch' },
+    });
+
+    expect(await executor.readCommand()).toStrictEqual({
+      jsonrpc: '2.0',
+      method: 'OutboundResponse',
+      params: { source: 'fetch' },
+    });
+
+    expect(await executor.readCommand()).toStrictEqual({
+      id: 2,
+      jsonrpc: '2.0',
+      result: true,
+    });
+  });
+
   it('notifies execution service of out of band errors via unhandledrejection', async () => {
     const CODE = `
       module.exports.onRpcRequest = async () => 'foo';

packages/snaps-execution-environments/src/common/endowments/endowments.test.browser.ts

@@ -12,7 +12,7 @@ import CryptoEndowment from './crypto';
 import date from './date';
 import interval from './interval';
 import math from './math';
-import network from './network';
+import network, { ResponseWrapper } from './network';
 import timeout from './timeout';
 
 // @ts-expect-error - `globalThis.process` is not optional.
@@ -218,6 +218,21 @@ describe('endowments', () => {
         endowments: { fetchAttenuated },
         factory: () => undefined,
       },
+      ResponseWrapper: {
+        endowments: {
+          Response: ResponseHardened,
+          ResponseWrapper: harden(ResponseWrapper),
+        },
+        factory: () =>
+          new ResponseWrapper(
+            new Response(),
+            { lastTeardown: 0 },
+            // eslint-disable-next-line @typescript-eslint/no-empty-function
+            async () => {},
+            // eslint-disable-next-line @typescript-eslint/no-empty-function
+            async () => {},
+          ),
+      },
     };
 
     Object.entries(TEST_ENDOWMENTS).forEach(

packages/snaps-execution-environments/src/common/endowments/network.test.ts

@@ -1,6 +1,6 @@
 import fetchMock from 'jest-fetch-mock';
 
-import network from './network';
+import network, { ResponseWrapper } from './network';
 
 describe('Network endowments', () => {
   beforeAll(() => {
@@ -187,7 +187,7 @@ describe('Network endowments', () => {
       const clonedResult = result.clone();
       expect(clonedResult).toBeDefined();
       expect(await clonedResult.text()).toBe(RESULT);
-      expect(clonedResult).not.toBeInstanceOf(Response);
+      expect(clonedResult).toBeInstanceOf(ResponseWrapper);
     });
 
     it('should return when json is called', async () => {

packages/snaps-execution-environments/src/common/endowments/network.ts

@@ -7,7 +7,7 @@ import type { EndowmentFactoryOptions } from './commonEndowmentFactory';
  * This class wraps a Response object.
  * That way, a teardown process can stop any processes left.
  */
-class ResponseWrapper implements Response {
+export class ResponseWrapper extends Response {
   readonly #teardownRef: { lastTeardown: number };
 
   #ogResponse: Response;
@@ -22,6 +22,7 @@ class ResponseWrapper implements Response {
     onStart: () => Promise<void>,
     onFinish: () => Promise<void>,
   ) {
+    super();
     this.#ogResponse = ogResponse;
     this.#teardownRef = teardownRef;
     this.#onStart = onStart;