tor-browser

browser_webchannel-get-js-sources.js (5642B)

---

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */

"use strict";

const FRONTEND_URL =
 "https://example.com/browser/devtools/client/performance-new/test/browser/webchannel-js-sources.html";

add_task(async function test_webchannel_get_js_sources() {
 info("Test the WebChannel GET_JS_SOURCES request functionality");

 // Register some mock JS sources for testing. We already test the logic that
 // retrieves the sources in other tests. This test focuses on the webchannel
 // itself only.
 const mockJSSources = {
   "uuid-12345-1": "function testFunction() { return 42; }",
   "uuid-12345-2": "console.log('Hello from source 2');",
   "uuid-67890-5": "function anotherFunction() { return 'test'; }",
   "uuid-67890-6": "alert('Hello world');",
 };

 const sourcesToRequest = ["uuid-12345-1", "uuid-12345-2", "uuid-67890-5"];

 const expectedResponse = [
   { sourceText: "function testFunction() { return 42; }" },
   { sourceText: "console.log('Hello from source 2');" },
   { sourceText: "function anotherFunction() { return 'test'; }" },
 ];

 await BrowserTestUtils.withNewTab(
   {
     gBrowser,
     url: "about:blank",
   },
   async browser => {
     // Register the mock sources for the current browser
     BackgroundJSM.registerProfileCaptureForBrowser(
       browser,
       Promise.resolve(new ArrayBuffer(0)), // Mock profile result
       null, // No symbolication service
       mockJSSources
     );

     // Now navigate to the test HTML page
     const url =
       FRONTEND_URL +
       "?testType=basic" +
       "&sourceUuids=" +
       encodeURIComponent(JSON.stringify(sourcesToRequest)) +
       "&expectedResponse=" +
       encodeURIComponent(JSON.stringify(expectedResponse));

     const loaded = BrowserTestUtils.browserLoaded(browser);
     BrowserTestUtils.startLoadingURIString(browser, url);
     await loaded;

     await waitForTabTitle("JS sources received");
     ok(true, "The JS sources are successfully retrieved by the WebChannel.");
   }
 );
});

add_task(async function test_webchannel_get_js_sources_nonexistent() {
 info("Test GET_JS_SOURCES WebChannel request with non-existent sources");

 // Register some mock JS sources
 const mockJSSources = {
   "uuid-12345-1": "function testFunction() { return 42; }",
 };

 // Request non-existent sources (should return null)
 const sourcesToRequest = [
   "uuid-12345-999", // Non-existent UUID
   "uuid-99999-1", // Non-existent UUID
 ];

 const expectedResponse = [
   { error: "Source not found in the browser" },
   { error: "Source not found in the browser" },
 ];

 await BrowserTestUtils.withNewTab(
   {
     gBrowser,
     url: "about:blank",
   },
   async browser => {
     // Register the mock sources for the current browser
     BackgroundJSM.registerProfileCaptureForBrowser(
       browser,
       Promise.resolve(new ArrayBuffer(0)),
       null,
       mockJSSources
     );

     // Now navigate to the test HTML page
     const url =
       FRONTEND_URL +
       "?testType=nonexistent" +
       "&sourceUuids=" +
       encodeURIComponent(JSON.stringify(sourcesToRequest)) +
       "&expectedResponse=" +
       encodeURIComponent(JSON.stringify(expectedResponse));

     const loaded = BrowserTestUtils.browserLoaded(browser);
     BrowserTestUtils.startLoadingURIString(browser, url);
     await loaded;

     await waitForTabTitle("JS sources received");
     ok(
       true,
       "Successfully verified GET_JS_SOURCES with non-existent sources"
     );
   }
 );
});

add_task(async function test_webchannel_get_js_sources_no_data() {
 info("Test GET_JS_SOURCES WebChannel request when no data is registered");

 const sourcesToRequest = ["uuid-12345-1"];

 // Open a new tab without registering JS sources
 await BrowserTestUtils.withNewTab(
   {
     gBrowser,
     url: "about:blank",
   },
   async browser => {
     // Don't register any JS sources - this should cause an error

     // Now navigate to the test HTML page
     const url =
       FRONTEND_URL +
       "?testType=no-data" +
       "&sourceUuids=" +
       encodeURIComponent(JSON.stringify(sourcesToRequest)) +
       "&expectError=true";

     const loaded = BrowserTestUtils.browserLoaded(browser);
     BrowserTestUtils.startLoadingURIString(browser, url);
     await loaded;

     await waitForTabTitle("JS sources received");
     ok(true, "Successfully verified GET_JS_SOURCES error handling");
   }
 );
});

add_task(async function test_webchannel_get_js_sources_invalid_request() {
 info("Test GET_JS_SOURCES WebChannel request with invalid sources parameter");

 await BrowserTestUtils.withNewTab(
   {
     gBrowser,
     url: "about:blank",
   },
   async browser => {
     // Don't need to register JS sources for this test

     // For this test, we need to pass invalid sourceUuids directly in the HTML
     // We'll use a special URL parameter that the HTML will use directly
     const url =
       FRONTEND_URL +
       "?testType=invalid-request" +
       "&sourceUuids=invalid_not_array" + // This is intentionally not JSON
       "&expectError=true";

     const loaded = BrowserTestUtils.browserLoaded(browser);
     BrowserTestUtils.startLoadingURIString(browser, url);
     await loaded;

     await waitForTabTitle("JS sources received");
     ok(true, "Successfully verified GET_JS_SOURCES invalid request handling");
   }
 );
});