-
-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bug 1919755 - correctly calculate content analysis URL for local PDF …
…file a=dmeehan
- Loading branch information
Showing
8 changed files
with
278 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
241 changes: 241 additions & 0 deletions
241
toolkit/components/contentanalysis/tests/browser/browser_print_pdf_content_analysis_impl.js
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,241 @@ | ||
| /* Any copyright is dedicated to the Public Domain. | ||
| http://creativecommons.org/publicdomain/zero/1.0/ */ | ||
| Services.scriptloader.loadSubScript( | ||
| "chrome://mochitests/content/browser/toolkit/components/printing/tests/head.js", | ||
| this | ||
| ); | ||
| const PSSVC = Cc["@mozilla.org/gfx/printsettings-service;1"].getService( | ||
| Ci.nsIPrintSettingsService | ||
| ); | ||
| let mockCA = makeMockContentAnalysis(); | ||
| add_setup(async function test_setup() { | ||
| mockCA = mockContentAnalysisService(mockCA); | ||
| }); | ||
| let testPDFUrl; | ||
| // Callers should set testPDFUrl | ||
| function addUniqueSuffix(prefix) { | ||
| return `${prefix}-${Services.uuid | ||
| .generateUUID() | ||
| .toString() | ||
| .slice(1, -1)}.pdf`; | ||
| } | ||
| async function printToDestination(aBrowser, aDestination) { | ||
| let tmpDir = Services.dirsvc.get("TmpD", Ci.nsIFile); | ||
| let fileName = addUniqueSuffix(`printDestinationTest-${aDestination}`); | ||
| let filePath = PathUtils.join(tmpDir.path, fileName); | ||
| info(`Printing to ${filePath}`); | ||
| let settings = PSSVC.createNewPrintSettings(); | ||
| settings.outputFormat = Ci.nsIPrintSettings.kOutputFormatPDF; | ||
| settings.outputDestination = aDestination; | ||
| settings.headerStrCenter = ""; | ||
| settings.headerStrLeft = ""; | ||
| settings.headerStrRight = ""; | ||
| settings.footerStrCenter = ""; | ||
| settings.footerStrLeft = ""; | ||
| settings.footerStrRight = ""; | ||
| settings.unwriteableMarginTop = 1; /* Just to ensure settings are respected on both */ | ||
| let outStream = null; | ||
| if (aDestination == Ci.nsIPrintSettings.kOutputDestinationFile) { | ||
| settings.toFileName = PathUtils.join(tmpDir.path, fileName); | ||
| } else { | ||
| is(aDestination, Ci.nsIPrintSettings.kOutputDestinationStream); | ||
| outStream = Cc["@mozilla.org/network/file-output-stream;1"].createInstance( | ||
| Ci.nsIFileOutputStream | ||
| ); | ||
| let tmpFile = tmpDir.clone(); | ||
| tmpFile.append(fileName); | ||
| outStream.init(tmpFile, -1, 0o666, 0); | ||
| settings.outputStream = outStream; | ||
| } | ||
| await aBrowser.browsingContext.print(settings); | ||
| return filePath; | ||
| } | ||
| function assertContentAnalysisRequest(request, expectedUrl) { | ||
| is(request.url.spec, expectedUrl ?? testPDFUrl, "request has correct URL"); | ||
| is( | ||
| request.analysisType, | ||
| Ci.nsIContentAnalysisRequest.ePrint, | ||
| "request has print analysisType" | ||
| ); | ||
| is( | ||
| request.operationTypeForDisplay, | ||
| Ci.nsIContentAnalysisRequest.eOperationPrint, | ||
| "request has print operationTypeForDisplay" | ||
| ); | ||
| is(request.textContent, "", "request textContent should be empty"); | ||
| is(request.filePath, "", "request filePath should be empty"); | ||
| isnot(request.printDataHandle, 0, "request printDataHandle should not be 0"); | ||
| isnot(request.printDataSize, 0, "request printDataSize should not be 0"); | ||
| ok(!!request.requestToken.length, "request requestToken should not be empty"); | ||
| } | ||
| // Printing to a stream is different than going through the print preview dialog because it | ||
| // doesn't make a static clone of the document before the print, which causes the | ||
| // Content Analysis code to go through a different code path. This is similar to what | ||
| // happens when various preferences are set to skip the print preview dialog, for example | ||
| // print.prefer_system_dialog. | ||
| add_task( | ||
| async function testPrintToStreamWithContentAnalysisActiveAndAllowing() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| mockCA.setupForTest(true); | ||
| let filePath = await printToDestination( | ||
| helper.sourceBrowser, | ||
| Ci.nsIPrintSettings.kOutputDestinationFile | ||
| ); | ||
| is( | ||
| mockCA.calls.length, | ||
| 1, | ||
| "Correct number of calls to Content Analysis" | ||
| ); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| await waitForFileToAlmostMatchSize( | ||
| filePath, | ||
| mockCA.calls[0].printDataSize | ||
| ); | ||
| await IOUtils.remove(filePath); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| } | ||
| ); | ||
| add_task( | ||
| async function testPrintToStreamWithContentAnalysisActiveAndBlocking() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| mockCA.setupForTest(false); | ||
| try { | ||
| await printToDestination( | ||
| helper.sourceBrowser, | ||
| Ci.nsIPrintSettings.kOutputDestinationFile | ||
| ); | ||
| ok(false, "Content analysis should make this fail to print"); | ||
| } catch (e) { | ||
| ok( | ||
| /NS_ERROR_CONTENT_BLOCKED/.test(e.toString()), | ||
| "Got content blocked error" | ||
| ); | ||
| } | ||
| is( | ||
| mockCA.calls.length, | ||
| 1, | ||
| "Correct number of calls to Content Analysis" | ||
| ); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| } | ||
| ); | ||
| add_task(async function testPrintToStreamWithContentAnalysisReturningError() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| expectUncaughtException(); | ||
| mockCA.setupForTestWithError(Cr.NS_ERROR_NOT_AVAILABLE); | ||
| try { | ||
| await printToDestination( | ||
| helper.sourceBrowser, | ||
| Ci.nsIPrintSettings.kOutputDestinationFile | ||
| ); | ||
| ok(false, "Content analysis should make this fail to print"); | ||
| } catch (e) { | ||
| ok( | ||
| /NS_ERROR_NOT_AVAILABLE/.test(e.toString()), | ||
| "Error in mock CA was propagated out" | ||
| ); | ||
| } | ||
| is(mockCA.calls.length, 1, "Correct number of calls to Content Analysis"); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| }); | ||
| add_task(async function testPrintThroughDialogWithContentAnalysisActive() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| mockCA.setupForTest(true); | ||
| await helper.startPrint(); | ||
| let fileName = addUniqueSuffix(`printDialogTest`); | ||
| let file = helper.mockFilePicker(fileName); | ||
| info(`Printing to ${file.path}`); | ||
| await helper.assertPrintToFile(file, () => { | ||
| EventUtils.sendKey("return", helper.win); | ||
| }); | ||
| is(mockCA.calls.length, 1, "Correct number of calls to Content Analysis"); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| await waitForFileToAlmostMatchSize( | ||
| file.path, | ||
| mockCA.calls[0].printDataSize | ||
| ); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| }); | ||
| add_task( | ||
| async function testPrintThroughDialogWithContentAnalysisActiveAndBlocking() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| mockCA.setupForTest(false); | ||
| await helper.startPrint(); | ||
| let fileName = addUniqueSuffix(`printDialogTest`); | ||
| let file = helper.mockFilePicker(fileName); | ||
| info(`Printing to ${file.path}`); | ||
| try { | ||
| await helper.assertPrintToFile(file, () => { | ||
| EventUtils.sendKey("return", helper.win); | ||
| }); | ||
| } catch (e) { | ||
| ok( | ||
| /Wait for target file to get created/.test(e.toString()), | ||
| "Target file should not get created" | ||
| ); | ||
| } | ||
| ok(!file.exists(), "File should not exist"); | ||
| is( | ||
| mockCA.calls.length, | ||
| 1, | ||
| "Correct number of calls to Content Analysis" | ||
| ); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| } | ||
| ); | ||
| add_task( | ||
| async function testPrintThroughDialogWithContentAnalysisReturningError() { | ||
| await PrintHelper.withTestPage( | ||
| async helper => { | ||
| expectUncaughtException(); | ||
| mockCA.setupForTestWithError(Cr.NS_ERROR_NOT_AVAILABLE); | ||
| await helper.startPrint(); | ||
| let fileName = addUniqueSuffix(`printDialogTest`); | ||
| let file = helper.mockFilePicker(fileName); | ||
| info(`Printing to ${file.path}`); | ||
| try { | ||
| await helper.assertPrintToFile(file, () => { | ||
| EventUtils.sendKey("return", helper.win); | ||
| }); | ||
| } catch (e) { | ||
| ok( | ||
| /Wait for target file to get created/.test(e.toString()), | ||
| "Target file should not get created" | ||
| ); | ||
| } | ||
| ok(!file.exists(), "File should not exist"); | ||
| is( | ||
| mockCA.calls.length, | ||
| 1, | ||
| "Correct number of calls to Content Analysis" | ||
| ); | ||
| assertContentAnalysisRequest(mockCA.calls[0]); | ||
| }, | ||
| testPDFUrl, | ||
| true | ||
| ); | ||
| } | ||
| ); |
11 changes: 11 additions & 0 deletions
11
toolkit/components/contentanalysis/tests/browser/browser_print_pdf_local_content_analysis.js
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| /* Any copyright is dedicated to the Public Domain. | ||
| http://creativecommons.org/publicdomain/zero/1.0/ */ | ||
| "use strict"; | ||
| Services.scriptloader.loadSubScript( | ||
| "chrome://mochitests/content/browser/toolkit/components/contentanalysis/tests/browser/browser_print_pdf_content_analysis_impl.js", | ||
| this | ||
| ); | ||
| let dir = getChromeDir(getResolvedURI(gTestPath)); | ||
| dir.append("file_pdf.pdf"); | ||
| dir.normalize(); | ||
| testPDFUrl = Services.io.newFileURI(dir).spec; |
9 changes: 9 additions & 0 deletions
9
...kit/components/contentanalysis/tests/browser/browser_print_pdf_remote_content_analysis.js
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| /* Any copyright is dedicated to the Public Domain. | ||
| http://creativecommons.org/publicdomain/zero/1.0/ */ | ||
| "use strict"; | ||
| Services.scriptloader.loadSubScript( | ||
| "chrome://mochitests/content/browser/toolkit/components/contentanalysis/tests/browser/browser_print_pdf_content_analysis_impl.js", | ||
| this | ||
| ); | ||
| testPDFUrl = | ||
| "https://example.com/browser/toolkit/components/contentanalysis/tests/browser/file_pdf.pdf"; |
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters