1 files changed, 66 insertions, 0 deletions
diff --git a/dom/manifest/test/browser_ManifestIcons_browserFetchIcon.js b/dom/manifest/test/browser_ManifestIcons_browserFetchIcon.js
new file mode 100644
index 0000000000..e56e1e25c2
--- /dev/null
+++ b/dom/manifest/test/browser_ManifestIcons_browserFetchIcon.js
@@ -0,0 +1,66 @@
+"use strict";
+
+Services.prefs.setBoolPref("dom.manifest.enabled", true);
+
+const { ManifestIcons } = ChromeUtils.importESModule(
+  "resource://gre/modules/ManifestIcons.sys.mjs"
+);
+const { ManifestObtainer } = ChromeUtils.importESModule(
+  "resource://gre/modules/ManifestObtainer.sys.mjs"
+);
+
+const defaultURL = new URL(
+  "https://example.org/browser/dom/manifest/test/resource.sjs"
+);
+defaultURL.searchParams.set("Content-Type", "application/manifest+json");
+
+const manifestMock = JSON.stringify({
+  icons: [
+    {
+      sizes: "50x50",
+      src: "red-50.png?Content-type=image/png",
+    },
+    {
+      sizes: "150x150",
+      src: "blue-150.png?Content-type=image/png",
+    },
+  ],
+});
+
+function makeTestURL() {
+  const url = new URL(defaultURL);
+  const body = `<link rel="manifest" href='${defaultURL}&body=${manifestMock}'>`;
+  url.searchParams.set("Content-Type", "text/html; charset=utf-8");
+  url.searchParams.set("body", encodeURIComponent(body));
+  return url.href;
+}
+
+function getIconColor(icon) {
+  return new Promise((resolve, reject) => {
+    const canvas = content.document.createElement("canvas");
+    const ctx = canvas.getContext("2d");
+    const image = new content.Image();
+    image.onload = function () {
+      ctx.drawImage(image, 0, 0);
+      resolve(ctx.getImageData(1, 1, 1, 1).data);
+    };
+    image.onerror = function () {
+      reject(new Error("could not create image"));
+    };
+    image.src = icon;
+  });
+}
+
+add_task(async function () {
+  const tabOptions = { gBrowser, url: makeTestURL() };
+  await BrowserTestUtils.withNewTab(tabOptions, async function (browser) {
+    const manifest = await ManifestObtainer.browserObtainManifest(browser);
+    let icon = await ManifestIcons.browserFetchIcon(browser, manifest, 25);
+    let color = await SpecialPowers.spawn(browser, [icon], getIconColor);
+    is(color[0], 255, "Fetched red icon");
+
+    icon = await ManifestIcons.browserFetchIcon(browser, manifest, 500);
+    color = await SpecialPowers.spawn(browser, [icon], getIconColor);
+    is(color[2], 255, "Fetched blue icon");
+  });
+});