Add tests for JavaScript source fetching

Author: canova

src/test/unit/fetch-source.test.js

@@ -580,4 +580,108 @@ describe('fetchSource', function () {
       ],
     });
   });
+
+  it('fetches JS source from browser with sourceId', async function () {
+    expect(
+      await fetchSource(
+        '/path/to/script.js',
+        'https://symbolication.services.mozilla.com',
+        null,
+        new Map(),
+        {
+          fetchUrlResponse: async (_url: string, _postData?: MixedObject) => {
+            throw new Error('Should not fetch from URL');
+          },
+          queryBrowserSymbolicationApi: async (
+            _path: string,
+            _requestJson: string
+          ) => {
+            throw new Error('Should not query symbolication API');
+          },
+          fetchJSSourceFromBrowser: async ({
+            pid,
+            sourceId,
+          }: GlobalJSSourceId) => {
+            if (pid === '123' && sourceId === 42) {
+              return 'console.log("Hello from browser with sourceId 42");';
+            }
+            throw new Error(`Unexpected sourceId: ${sourceId}`);
+          },
+        },
+        { pid: '123', sourceId: 42 }
+      )
+    ).toEqual({
+      type: 'SUCCESS',
+      source: 'console.log("Hello from browser with sourceId 42");',
+    });
+  });
+
+  it('handles fetch JS source from browser with invalid sourceId', async function () {
+    expect(
+      await fetchSource(
+        '/path/to/script.js',
+        'https://symbolication.services.mozilla.com',
+        null,
+        new Map(),
+        {
+          fetchUrlResponse: async (_url: string, _postData?: MixedObject) => {
+            throw new Error('Should not fetch from URL');
+          },
+          queryBrowserSymbolicationApi: async (
+            _path: string,
+            _requestJson: string
+          ) => {
+            throw new Error('Should not query symbolication API');
+          },
+          fetchJSSourceFromBrowser: async ({ sourceId }: GlobalJSSourceId) => {
+            throw new Error(`Source not found for sourceId: ${sourceId}`);
+          },
+        },
+        { pid: '111', sourceId: 123 }
+      )
+    ).toEqual({
+      type: 'ERROR',
+      errors: [
+        {
+          type: 'BROWSER_API_ERROR',
+          apiErrorMessage: 'Source not found for sourceId: 123',
+        },
+      ],
+    });
+  });
+
+  it('falls back to other methods when fetchJSSourceFromBrowser fails', async function () {
+    expect(
+      await fetchSource(
+        'hg:hg.mozilla.org/mozilla-central:widget/cocoa/nsAppShell.mm:997f00815e6bc28806b75448c8829f0259d2cb28',
+        'https://symbolication.services.mozilla.com',
+        null,
+        new Map(),
+        {
+          fetchUrlResponse: async (url: string, _postData?: MixedObject) => {
+            const r = new Response(`Fallback response from ${url}`, {
+              status: 200,
+            });
+            return r;
+          },
+          queryBrowserSymbolicationApi: async (
+            _path: string,
+            _requestJson: string
+          ) => {
+            throw new Error('No browser connection');
+          },
+          fetchJSSourceFromBrowser: async (
+            _globalSourceId: GlobalJSSourceId
+          ) => {
+            throw new Error('Source not found in browser');
+          },
+        },
+        { pid: '123', sourceId: 42 } // Should still try browser first but fall back to URL fetch
+      )
+    ).toEqual({
+      type: 'SUCCESS',
+      source:
+        'Fallback response from https://hg.mozilla.org/mozilla-central/raw-file/997f00815e6bc28806b75448c8829f0259d2cb28/widget/cocoa/nsAppShell.mm',
+    });
+  });
 });