Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
const { PdfJsGuessAltTextFeature } = ChromeUtils.importESModule(
"resource://pdf.js/PdfJsAIFeature.sys.mjs"
);
const { sinon } = ChromeUtils.importESModule(
);
const { MLUninstallService } = ChromeUtils.importESModule(
"chrome://global/content/ml/Utils.sys.mjs"
);
const PREF_ENABLED = "pdfjs.enableGuessAltText";
const PREF_ALT_TEXT_MODEL_DOWNLOAD = "pdfjs.enableAltTextModelDownload";
const PREF_ALT_TEXT_ENABLED = "pdfjs.enableAltText";
function clearGuessAltTextPrefs() {
for (const pref of [
PREF_ENABLED,
PREF_ALT_TEXT_ENABLED,
PREF_ALT_TEXT_MODEL_DOWNLOAD,
]) {
if (Services.prefs.prefHasUserValue(pref)) {
Services.prefs.clearUserPref(pref);
}
}
}
registerCleanupFunction(() => {
clearGuessAltTextPrefs();
sinon.restore();
});
add_task(function test_id_is_feature_id() {
Assert.equal(
PdfJsGuessAltTextFeature.id,
"pdfjs-alt-text",
"PdfJsGuessAltTextFeature.id should be the feature id"
);
});
add_task(function test_isEnabled_with_pref() {
clearGuessAltTextPrefs();
Assert.ok(
!PdfJsGuessAltTextFeature.isEnabled,
"With no user prefs set, isEnabled should be false"
);
Services.prefs.setBoolPref(PREF_ENABLED, true);
Services.prefs.setBoolPref(PREF_ALT_TEXT_MODEL_DOWNLOAD, true);
Services.prefs.setBoolPref(PREF_ALT_TEXT_ENABLED, true);
Services.prefs.getBoolPref("browser.ml.enable", true);
Assert.ok(PdfJsGuessAltTextFeature.isEnabled, "isEnabled should be true");
});
add_task(async function test_enable_sets_all_prefs_true() {
clearGuessAltTextPrefs();
await PdfJsGuessAltTextFeature.enable();
Assert.equal(
Services.prefs.getBoolPref(PREF_ENABLED, false),
true,
"enable() should set pdfjs.enableGuessAltText=true"
);
Assert.equal(
Services.prefs.getBoolPref(PREF_ALT_TEXT_ENABLED, false),
true,
"enable() should set pdfjs.enableAltText=true"
);
Assert.equal(
Services.prefs.getBoolPref(PREF_ALT_TEXT_MODEL_DOWNLOAD, false),
true,
"enable() should set pdfjs.enableAltTextModelDownload=true"
);
Assert.ok(
PdfJsGuessAltTextFeature.isEnabled,
"After enable(), isEnabled should be true"
);
});
add_task(async function test_disable_sets_prefs_false_and_uninstalls_models() {
clearGuessAltTextPrefs();
await PdfJsGuessAltTextFeature.enable();
Assert.ok(
PdfJsGuessAltTextFeature.isEnabled,
"Sanity check: enabled before disable()"
);
const uninstallStub = sinon.stub(MLUninstallService, "uninstall").resolves();
await PdfJsGuessAltTextFeature.disable();
Assert.equal(
Services.prefs.getBoolPref(PREF_ENABLED, true),
false,
"disable() should set pdfjs.enableGuessAltText=false"
);
Assert.equal(
Services.prefs.getBoolPref(PREF_ALT_TEXT_MODEL_DOWNLOAD, true),
false,
"disable() should set pdfjs.enableAltTextModelDownload=false"
);
Assert.equal(
Services.prefs.getBoolPref(PREF_ALT_TEXT_ENABLED, true),
false,
"disable() changes pdfjs.enableAltText"
);
Assert.ok(
!PdfJsGuessAltTextFeature.isEnabled,
"After disable(), isEnabled should be false"
);
Assert.ok(
uninstallStub.calledOnce,
"disable() should uninstall ML engine files via MLUninstallService.uninstall()"
);
const expectedEngineIds = [PdfJsGuessAltTextFeature.engineId];
const uninstallArgs = uninstallStub.getCall(0).args[0];
Assert.deepEqual(
(uninstallArgs.engineIds || []).slice(),
expectedEngineIds,
"disable() should uninstall files for the expected engine IDs"
);
Assert.equal(
uninstallArgs.actor,
PdfJsGuessAltTextFeature.engineId,
"disable() should pass the expected actor attribution"
);
uninstallStub.restore();
});