toolkit/components/pdfjs/test/browser_pdfjs_hcm.js

   1 /* Any copyright is dedicated to the Public Domain.
   2  * http://creativecommons.org/publicdomain/zero/1.0/ */
   3
   4 const RELATIVE_DIR = "toolkit/components/pdfjs/test/";
   5 const TESTROOT = "http://example.com/browser/" + RELATIVE_DIR;
   6
   7 /**
   8  * Get the first and last pixels on the drawn canvas.
   9  * @param {Object} browser
  10  * @returns {Object}
  11  */
  12 async function getFirstLastPixels(browser) {
  13   return SpecialPowers.spawn(browser, [], async function () {
  14     const { document } = content;
  15     const canvas = document.querySelector("canvas");
  16
  17     Assert.ok(!!canvas, "We must have a canvas");
  18
  19     const data = Array.from(
  20       canvas.getContext("2d").getImageData(0, 0, canvas.width, canvas.height)
  21         .data
  22     );
  23
  24     return {
  25       first: data.slice(0, 3),
  26       last: data.slice(-4, -1),
  27     };
  28   });
  29 }
  30
  31 /**
  32  * Test that the pdf has the correct color depending on the high contrast mode.
  33  */
  34 add_task(async function test() {
  35   let mimeService = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
  36   let handlerInfo = mimeService.getFromTypeAndExtension(
  37     "application/pdf",
  38     "pdf"
  39   );
  40
  41   // Make sure pdf.js is the default handler.
  42   is(
  43     handlerInfo.alwaysAskBeforeHandling,
  44     false,
  45     "pdf handler defaults to always-ask is false"
  46   );
  47   is(
  48     handlerInfo.preferredAction,
  49     Ci.nsIHandlerInfo.handleInternally,
  50     "pdf handler defaults to internal"
  51   );
  52
  53   info("Pref action: " + handlerInfo.preferredAction);
  54
  55   await SpecialPowers.pushPrefEnv({
  56     set: [
  57       ["browser.display.document_color_use", 0],
  58       ["browser.display.background_color", "#ff0000"],
  59       ["browser.display.foreground_color", "#00ff00"],
  60       ["browser.display.use_system_colors", false],
  61     ],
  62   });
  63
  64   await BrowserTestUtils.withNewTab(
  65     { gBrowser, url: "about:blank" },
  66     async function (browser) {
  67       // check that PDF is opened with internal viewer
  68       await waitForPdfJSCanvas(
  69         browser,
  70         `${TESTROOT}file_pdfjs_hcm.pdf#zoom=800`
  71       );
  72
  73       const { first, last } = await getFirstLastPixels(browser);
  74       // The first pixel must be black and not green.
  75       Assert.deepEqual(first, [0, 0, 0]);
  76
  77       // The last pixel must be white and not red.
  78       Assert.deepEqual(last, [255, 255, 255]);
  79
  80       await waitForPdfJSClose(browser);
  81     }
  82   );
  83
  84   // Enable HCM.
  85   await SpecialPowers.pushPrefEnv({
  86     set: [["ui.useAccessibilityTheme", 1]],
  87   });
  88
  89   await BrowserTestUtils.withNewTab(
  90     { gBrowser, url: "about:blank" },
  91     async function (browser) {
  92       // check that PDF is opened with internal viewer
  93       await waitForPdfJSCanvas(
  94         browser,
  95         `${TESTROOT}file_pdfjs_hcm.pdf#zoom=800`
  96       );
  97
  98       const { first, last } = await getFirstLastPixels(browser);
  99       // The first pixel must be green.
 100       Assert.deepEqual(first, [0, 255, 0]);
 101
 102       // The last pixel must be red.
 103       Assert.deepEqual(last, [255, 0, 0]);
 104
 105       await waitForPdfJSClose(browser);
 106     }
 107   );
 108
 109   await SpecialPowers.popPrefEnv();
 110 });