browser_save_page_complete_content_analysis.js

Enable keyboard shortcuts

/* Any copyright is dedicated to the Public Domain.

   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

let MockFilePicker = SpecialPowers.MockFilePicker;

let mockCA;

const TEST_URL =

  "https://example.com/browser/toolkit/components/contentanalysis/tests/browser/save_page_complete.html";

let tempDir;

add_setup(async function test_setup() {

  mockCA = await mockContentAnalysisService(makeMockContentAnalysis());

  tempDir = Services.dirsvc.get("TmpD", Ci.nsIFile);

  tempDir.append("test-save-page-complete-dir");

  tempDir.createUnique(Ci.nsIFile.DIRECTORY_TYPE, 0o755);

  MockFilePicker.init();

  MockFilePicker.returnValue = MockFilePicker.returnOK;

  MockFilePicker.displayDirectory = tempDir;

  registerCleanupFunction(async function () {

    MockFilePicker.cleanup();

    let list = await Downloads.getList(Downloads.ALL);

    await list.removeFinished();

    tempDir.remove(true);

});

});

function createPromiseForFilePicker() {

  return new Promise(resolve => {

    MockFilePicker.showCallback = fp => {

      let destFile = tempDir.clone();

      destFile.append(fp.defaultString);

      let filesFolder = tempDir.clone();

      filesFolder.append(fp.defaultString.replace(/\.[^.]*$/, "") + "_files");

      MockFilePicker.setFiles([destFile]);

      MockFilePicker.filterIndex = 0; // kSaveAsType_Complete

      resolve({ destFile, filesFolder });

};

});

function promiseDownloadFinished(list) {

  return new Promise(resolve => {

    list.addView({

      onDownloadChanged(download) {

        download.launchWhenSucceeded = false;

        if (download.succeeded || download.error) {

          list.removeView(this);

          resolve(download);

},

});

});

async function testSavePageComplete(allow) {

  mockCA.setupForTest(allow);

  await SpecialPowers.pushPrefEnv({

    set: [

      ["browser.contentanalysis.interception_point.download.enabled", true],

],

});

  await BrowserTestUtils.withNewTab({ gBrowser, url: TEST_URL }, async () => {

    let downloadList = await Downloads.getList(Downloads.ALL);

    let filePickerShown = createPromiseForFilePicker();

    let downloadFinishedPromise = promiseDownloadFinished(downloadList);

    saveBrowser(gBrowser.selectedBrowser);

    info("Waiting for a filename to be picked from the file picker");

    let { destFile, filesFolder } = await filePickerShown;

    let download = await downloadFinishedPromise;

    is(mockCA.calls.length, 1, "Content Analysis called once");

    is(download.target.path, destFile.path, "Download saved to picked file");

is(

      download.target.filesFolderPath,

      filesFolder.path,

      "Download knows about its files folder"

);

is(

      await IOUtils.exists(download.target.path),

      allow,

      "Target file existence"

);

    is(await IOUtils.exists(filesFolder.path), allow, "Files folder existence");

    if (allow) {

      ok(!download.error, "Download should not have an error");

      await IOUtils.remove(download.target.path);

      await IOUtils.remove(filesFolder.path, { recursive: true });

    } else {

ok(

        download.error.becauseBlockedByContentAnalysis,

        "Download blocked by content analysis"

);

    await downloadList.removeFinished();

});

  await SpecialPowers.popPrefEnv();

add_task(async function testSavePageCompleteAllow() {

  await testSavePageComplete(true);

});

add_task(async function testSavePageCompleteBlock() {

  await testSavePageComplete(false);

});