Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
Services.scriptloader.loadSubScript(
this
);
const RELATIVE_DIR = "toolkit/components/pdfjs/test/";
const pdfUrl = TESTROOT + "file_pdfjs_test.pdf";
const altTextPref = "pdfjs.enableAltText";
const guessAltTextPref = "pdfjs.enableGuessAltText";
const browserMLPref = "browser.ml.enable";
const annotationEditorModePref = "pdfjs.annotationEditorMode";
async function setupRemoteClient() {
const { removeMocks, remoteClients } = await createAndMockMLRemoteSettings({
autoDownloadFromRemoteSettings: false,
});
return {
remoteClients,
async cleanup() {
await removeMocks();
await waitForCondition(
() => EngineProcess.areAllEnginesTerminated(),
"Waiting for all of the engines to be terminated.",
100,
200
);
},
};
}
// Test that the model is loaded.
add_task(async function test() {
let mimeService = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
let handlerInfo = mimeService.getFromTypeAndExtension(
"application/pdf",
"pdf"
);
// Make sure pdf.js is the default handler.
is(
handlerInfo.alwaysAskBeforeHandling,
false,
"pdf handler defaults to always-ask is false"
);
is(
handlerInfo.preferredAction,
Ci.nsIHandlerInfo.handleInternally,
"pdf handler defaults to internal"
);
info("Pref action: " + handlerInfo.preferredAction);
await BrowserTestUtils.withNewTab(
{ gBrowser, url: "about:blank" },
async function test_alttext_load_engine(browser) {
const { EngineProcess } = ChromeUtils.importESModule(
"chrome://global/content/ml/EngineProcess.sys.mjs"
);
await SpecialPowers.pushPrefEnv({
set: [
[altTextPref, true],
[browserMLPref, true],
[guessAltTextPref, true],
[annotationEditorModePref, 0],
],
});
const setRC = await setupRemoteClient();
await waitForPdfJS(browser, pdfUrl);
await enableEditor(browser, "Stamp", 1);
const isEnabled = await SpecialPowers.spawn(browser, [], () => {
const viewer = content.wrappedJSObject.PDFViewerApplication;
return viewer.mlManager.isEnabledFor("altText");
});
ok(isEnabled, "AltText is enabled");
await setRC.remoteClients["ml-onnx-runtime"].resolvePendingDownloads(1);
await EngineProcess.destroyMLEngine();
await setRC.cleanup();
await waitForPdfJSClose(browser);
await SpecialPowers.popPrefEnv();
}
);
});