/* 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/. */

const { logTest } = require("./utils/profiling");

module.exports = logTest(
  "speculative connect pageload",
  async function (context, commands) {
    context.log.info(
      "Starting a pageload for which we will first make a speculative connection to the host"
    );

    const url = "https://en.wikipedia.org/wiki/Barack_Obama";

    await commands.navigate("about:blank");
    await commands.wait.byTime(1000);

    context.log.debug("Make privileged call to speculativeConnect");
    const script = `
    var URI = Services.io.newURI("${url}");
    var principal = Services.scriptSecurityManager.createContentPrincipal(URI, {});
    var browsingContext = gBrowser.selectedBrowser.browsingContext;
    var callbacks = {
      QueryInterface: ChromeUtils.generateQI(["nsIInterfaceRequestor"]),
      getInterface(iid) {
        if (iid.equals(Ci.nsILoadContext)) {
          return browsingContext;
        }
        throw Components.Exception("", Cr.NS_ERROR_NO_INTERFACE);
      },
    };
    Services.io.speculativeConnect(URI, principal, callbacks, false);
  `;

    commands.js.runPrivileged(script);

    // More than enough time for the connection to be made
    await commands.wait.byTime(1000);

    await commands.measure.start();
    await commands.navigate(url);
    await commands.measure.stop();

    let connect_time = await commands.js.run(
      `return (window.performance.timing.connectEnd - window.performance.timing.navigationStart);`
    );
    context.log.info("connect_time: " + connect_time);

    await commands.measure.addObject({
      custom_data: { connect_time },
    });

    context.log.info("Speculative connect test finished.");
    return true;
  }
);