Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
const { PromptTestUtils } = ChromeUtils.importESModule(
);
let mockCA = makeMockContentAnalysis();
add_setup(async function test_setup() {
mockCA = await mockContentAnalysisService(mockCA);
});
// Using an external page so the test can checks that the URL matches in the nsIContentAnalysisRequest
const PAGE_URL =
const CLIPBOARD_TEXT_STRING = "Just some text";
const TEST_MODES = Object.freeze({
ALLOW: { caAllow: true, shouldPaste: true, shouldRunCA: true },
BLOCK: { caAllow: false, shouldPaste: false, shouldRunCA: true },
PREFOFF: { caAllow: false, shouldPaste: true, shouldRunCA: false },
});
async function testClipboardPaste(testMode) {
mockCA.setupForTest(testMode.caAllow);
let tab = await BrowserTestUtils.openNewForegroundTab(gBrowser, PAGE_URL);
let browser = tab.linkedBrowser;
let promptPromise = SpecialPowers.spawn(browser, [], async () => {
return content.prompt();
});
let prompt = await PromptTestUtils.waitForPrompt(browser, {
modalType: Services.prompt.MODAL_TYPE_CONTENT,
});
// Resolves when the contents are updated. We cannot reliably detect that
// the input element rejected the paste due to CA (there is no adjacent
// event) but we can detect when it has been permitted, and the test is currently
// written to give the element time to update before checking that content
// was blocked.
let selectionChangePromise = new Promise(resolve => {
prompt.ui.loginTextbox.addEventListener(
"selectionchange",
() => {
ok(true, "got selectionchange");
// Since mockCA uses setTimeout before invoking the callback,
// do it here too
setTimeout(() => {
resolve();
}, 0);
},
{ once: true }
);
});
// Paste text into prompt()
setClipboardData(CLIPBOARD_TEXT_STRING);
prompt.ui.loginTextbox.focus();
await EventUtils.synthesizeKey("v", { accelKey: true });
// If the contents should update, wait for that. If not, just wait for a
// cycle.
if (testMode.shouldPaste) {
await selectionChangePromise;
} else {
await new Promise(res => {
setTimeout(res, 0);
});
}
// Close the prompt
await PromptTestUtils.handlePrompt(prompt);
let result = await promptPromise;
is(
result,
testMode.shouldPaste ? CLIPBOARD_TEXT_STRING : "",
"prompt has correct value"
);
is(
mockCA.calls.length,
testMode.shouldRunCA ? 1 : 0,
"Correct number of calls to Content Analysis"
);
if (testMode.shouldRunCA) {
assertContentAnalysisRequest(
mockCA.calls[0],
CLIPBOARD_TEXT_STRING,
mockCA.calls[0].userActionId,
1
);
}
is(
mockCA.browsingContextsForURIs.length,
1,
"Correct number of calls to getURIForBrowsingContext()"
);
BrowserTestUtils.removeTab(tab);
}
function assertContentAnalysisRequest(
request,
expectedText,
expectedUserActionId,
expectedRequestsCount
) {
is(request.url.spec, PAGE_URL, "request has correct URL");
is(
request.analysisType,
Ci.nsIContentAnalysisRequest.eBulkDataEntry,
"request has correct analysisType"
);
is(
request.reason,
Ci.nsIContentAnalysisRequest.eClipboardPaste,
"request has correct reason"
);
is(
request.operationTypeForDisplay,
Ci.nsIContentAnalysisRequest.eClipboard,
"request has correct operationTypeForDisplay"
);
is(request.filePath, null, "request filePath should match");
is(request.textContent, expectedText, "request textContent should match");
is(
request.userActionRequestsCount,
expectedRequestsCount,
"request userActionRequestsCount should match"
);
is(
request.userActionId,
expectedUserActionId,
"request userActionId should match"
);
ok(request.userActionId.length, "request userActionId should not be empty");
is(request.printDataHandle, 0, "request printDataHandle should not be 0");
is(request.printDataSize, 0, "request printDataSize should not be 0");
ok(!!request.requestToken.length, "request requestToken should not be empty");
}
function setClipboardData(clipboardString) {
const trans = Cc["@mozilla.org/widget/transferable;1"].createInstance(
Ci.nsITransferable
);
trans.init(null);
trans.addDataFlavor("text/plain");
const str = Cc["@mozilla.org/supports-string;1"].createInstance(
Ci.nsISupportsString
);
str.data = clipboardString;
trans.setTransferData("text/plain", str);
// Write to clipboard.
Services.clipboard.setData(trans, null, Ci.nsIClipboard.kGlobalClipboard);
}
add_task(async function testClipboardPasteWithContentAnalysisAllow() {
await testClipboardPaste(TEST_MODES.ALLOW);
});
add_task(async function testClipboardPasteWithContentAnalysisBlock() {
await testClipboardPaste(TEST_MODES.BLOCK);
});
add_task(async function testClipboardPasteWithContentAnalysisBlockButPrefOff() {
await SpecialPowers.pushPrefEnv({
set: [
["browser.contentanalysis.interception_point.clipboard.enabled", false],
],
});
await testClipboardPaste(TEST_MODES.PREFOFF);
await SpecialPowers.popPrefEnv();
});