Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
let mockCA = makeMockContentAnalysis();
const DOWNLOAD_URL =
const TEST_URL =
async function triggerSaveAs({ selector }) {
const contextMenu = window.document.getElementById("contentAreaContextMenu");
const popupshown = BrowserTestUtils.waitForEvent(contextMenu, "popupshown");
await BrowserTestUtils.synthesizeMouseAtCenter(
selector,
{ type: "contextmenu", button: 2 },
gBrowser.selectedBrowser
);
await popupshown;
let saveLinkCommand = window.document.getElementById("context-savelink");
contextMenu.activateItem(saveLinkCommand);
}
let destFilePath;
add_setup(async () => {
mockCA = await mockContentAnalysisService(mockCA);
});
function setupMockFilePicker() {
const tempDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
tempDir.append("test-download-dir");
if (!tempDir.exists()) {
tempDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0o755);
}
let MockFilePicker = SpecialPowers.MockFilePicker;
MockFilePicker.init(window.browsingContext);
registerCleanupFunction(function () {
MockFilePicker.cleanup();
if (tempDir.exists()) {
tempDir.remove(true);
}
});
MockFilePicker.displayDirectory = tempDir;
MockFilePicker.showCallback = function (fp) {
info("MockFilePicker: shown");
const filename = fp.defaultString;
info("MockFilePicker: save as " + filename);
let destFile = tempDir.clone();
destFile.append(filename);
MockFilePicker.setFiles([destFile]);
destFilePath = destFile.path;
info("MockFilePicker: showCallback done");
};
}
function assertContentAnalysisDownloadRequest(request, expectedFilePath) {
is(request.url.spec, DOWNLOAD_URL, "request has correct URL");
is(
request.analysisType,
Ci.nsIContentAnalysisRequest.eFileDownloaded,
"request has correct analysisType"
);
is(
request.reason,
Ci.nsIContentAnalysisRequest.eSaveAsDownload,
"request has correct reason"
);
is(
request.operationTypeForDisplay,
Ci.nsIContentAnalysisRequest.eDownload,
"request has correct operationTypeForDisplay"
);
is(request.filePath, expectedFilePath, "request filePath should match");
ok(!request.textContent?.length, "request textContent should be empty");
is(
request.userActionRequestsCount,
1,
"request userActionRequestsCount should match"
);
ok(request.userActionId.length, "request userActionId should not be empty");
is(request.printDataHandle, 0, "request printDataHandle should be 0");
is(request.printDataSize, 0, "request printDataSize should be 0");
ok(!!request.requestToken.length, "request requestToken should not be empty");
}
add_task(async function test_download_content_analysis_download_save_as() {
await SpecialPowers.pushPrefEnv({
set: [
["browser.contentanalysis.interception_point.download.enabled", true],
],
});
mockCA.setupForTest(true);
await BrowserTestUtils.withNewTab({ gBrowser, url: TEST_URL }, async () => {
setupMockFilePicker();
await triggerSaveAs({ selector: "a.test-link" });
});
await TestUtils.waitForCondition(async () => {
return (
destFilePath &&
(await IOUtils.exists(destFilePath)) &&
(await IOUtils.stat(destFilePath)).size > 0
);
}, "Wait for the file to be downloaded");
is(mockCA.calls.length, 1, "Content analysis should be called once");
assertContentAnalysisDownloadRequest(mockCA.calls[0], destFilePath);
await IOUtils.remove(destFilePath);
await SpecialPowers.popPrefEnv();
});