fune/browser/components/shell/test/browser_setDefaultPDFHandler.js

/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/ */

XPCOMUtils.defineLazyModuleGetters(this, {
  ExperimentAPI: "resource://nimbus/ExperimentAPI.jsm",
  ExperimentFakes: "resource://testing-common/NimbusTestUtils.jsm",
  NimbusFeatures: "resource://nimbus/ExperimentAPI.jsm",
  sinon: "resource://testing-common/Sinon.jsm",
});

XPCOMUtils.defineLazyServiceGetter(
  this,
  "XreDirProvider",
  "@mozilla.org/xre/directory-provider;1",
  "nsIXREDirProvider"
);

const _callExternalDefaultBrowserAgentStub = sinon
  .stub(ShellService, "_callExternalDefaultBrowserAgent")
  .callsFake(async () => ({
    async wait() {
      return { exitCode: 0 };
    },
  }));

const _userChoiceImpossibleTelemetryResultStub = sinon
  .stub(ShellService, "_userChoiceImpossibleTelemetryResult")
  .callsFake(() => null);

// Ensure we don't fall back to a real implementation.
const setDefaultStub = sinon.stub();
const shellStub = sinon
  .stub(ShellService, "shellService")
  .value({ setDefaultBrowser: setDefaultStub });

registerCleanupFunction(() => {
  _callExternalDefaultBrowserAgentStub.restore();
  _userChoiceImpossibleTelemetryResultStub.restore();
  shellStub.restore();

  ExperimentAPI._store._deleteForTests("shellService");
});

add_task(async function ready() {
  await ExperimentAPI.ready();
});

if (!AppConstants.isPlatformAndVersionAtLeast("win", "10")) {
  // Everything here is Windows 10+, but there's no way to filter out
  // Windows 7 test machines using the test manifest at this time...
  // and the test harness fails test files that make no assertions.
  // So we add a dummy assertion here.
  Assert.ok(true, "Skipping test on Windows version before Windows 10");
} else {
  // We're on a Windows 10 test machine.

  add_task(async function remoteEnableWithPDF() {
    let doCleanup = await ExperimentFakes.enrollWithRollout({
      featureId: NimbusFeatures.shellService.featureId,
      value: {
        setDefaultBrowserUserChoice: true,
        setDefaultPDFHandler: true,
        enabled: true,
      },
    });

    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultBrowserUserChoice"),
      true
    );
    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultPDFHandler"),
      true
    );

    _callExternalDefaultBrowserAgentStub.resetHistory();
    ShellService.setDefaultBrowser();

    const aumi = XreDirProvider.getInstallHash();
    Assert.ok(_callExternalDefaultBrowserAgentStub.called);
    Assert.deepEqual(_callExternalDefaultBrowserAgentStub.firstCall.args, [
      { arguments: ["set-default-browser-user-choice", aumi, ".pdf"] },
    ]);

    await doCleanup();
  });

  add_task(async function remoteEnableWithoutPDF() {
    let doCleanup = await ExperimentFakes.enrollWithRollout({
      featureId: NimbusFeatures.shellService.featureId,
      value: {
        setDefaultBrowserUserChoice: true,
        setDefaultPDFHandler: false,
        enabled: true,
      },
    });

    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultBrowserUserChoice"),
      true
    );
    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultPDFHandler"),
      false
    );

    _callExternalDefaultBrowserAgentStub.resetHistory();
    ShellService.setDefaultBrowser();

    const aumi = XreDirProvider.getInstallHash();
    Assert.ok(_callExternalDefaultBrowserAgentStub.called);
    Assert.deepEqual(_callExternalDefaultBrowserAgentStub.firstCall.args, [
      { arguments: ["set-default-browser-user-choice", aumi] },
    ]);

    await doCleanup();
  });

  add_task(async function remoteDisable() {
    let doCleanup = await ExperimentFakes.enrollWithRollout({
      featureId: NimbusFeatures.shellService.featureId,
      value: {
        setDefaultBrowserUserChoice: false,
        setDefaultPDFHandler: true,
        enabled: false,
      },
    });

    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultBrowserUserChoice"),
      false
    );
    Assert.equal(
      NimbusFeatures.shellService.getVariable("setDefaultPDFHandler"),
      true
    );

    _callExternalDefaultBrowserAgentStub.resetHistory();
    ShellService.setDefaultBrowser();

    Assert.ok(_callExternalDefaultBrowserAgentStub.notCalled);
    Assert.ok(setDefaultStub.called);

    await doCleanup();
  });
}