1 files changed, 169 insertions, 0 deletions
diff --git a/toolkit/mozapps/extensions/test/xpcshell/test_webextension_icons.js b/toolkit/mozapps/extensions/test/xpcshell/test_webextension_icons.js
new file mode 100644
index 0000000000..25fb4115f2
--- /dev/null
+++ b/toolkit/mozapps/extensions/test/xpcshell/test_webextension_icons.js
@@ -0,0 +1,169 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+const ID = "webextension1@tests.mozilla.org";
+
+const profileDir = gProfD.clone();
+profileDir.append("extensions");
+profileDir.create(AM_Ci.nsIFile.DIRECTORY_TYPE, FileUtils.PERMS_DIRECTORY);
+
+createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "42");
+startupManager();
+
+const { Management } = Components.utils.import("resource://gre/modules/Extension.jsm", {});
+
+function promiseAddonStartup() {
+  return new Promise(resolve => {
+    let listener = (evt, extension) => {
+      Management.off("startup", listener);
+      resolve(extension);
+    };
+
+    Management.on("startup", listener);
+  });
+}
+
+// Test simple icon set parsing
+add_task(function*() {
+  yield promiseWriteWebManifestForExtension({
+    name: "Web Extension Name",
+    version: "1.0",
+    manifest_version: 2,
+    applications: {
+      gecko: {
+        id: ID
+      }
+    },
+    icons: {
+      16: "icon16.png",
+      32: "icon32.png",
+      48: "icon48.png",
+      64: "icon64.png"
+    }
+  }, profileDir);
+
+  yield promiseRestartManager();
+  yield promiseAddonStartup();
+
+  let uri = do_get_addon_root_uri(profileDir, ID);
+
+  let addon = yield promiseAddonByID(ID);
+  do_check_neq(addon, null);
+
+  function check_icons(addon_copy) {
+    deepEqual(addon_copy.icons, {
+        16: uri + "icon16.png",
+        32: uri + "icon32.png",
+        48: uri + "icon48.png",
+        64: uri + "icon64.png"
+    });
+
+    // iconURL should map to icons[48] and icons[64]
+    equal(addon.iconURL, uri + "icon48.png");
+    equal(addon.icon64URL, uri + "icon64.png");
+
+    // AddonManager gets the correct icon sizes from addon.icons
+    equal(AddonManager.getPreferredIconURL(addon, 1), uri + "icon16.png");
+    equal(AddonManager.getPreferredIconURL(addon, 16), uri + "icon16.png");
+    equal(AddonManager.getPreferredIconURL(addon, 30), uri + "icon32.png");
+    equal(AddonManager.getPreferredIconURL(addon, 48), uri + "icon48.png");
+    equal(AddonManager.getPreferredIconURL(addon, 64), uri + "icon64.png");
+    equal(AddonManager.getPreferredIconURL(addon, 128), uri + "icon64.png");
+  }
+
+  check_icons(addon);
+
+  // check if icons are persisted through a restart
+  yield promiseRestartManager();
+  yield promiseAddonStartup();
+
+  addon = yield promiseAddonByID(ID);
+  do_check_neq(addon, null);
+
+  check_icons(addon);
+
+  addon.uninstall();
+
+  yield promiseRestartManager();
+});
+
+// Test AddonManager.getPreferredIconURL for retina screen sizes
+add_task(function*() {
+  yield promiseWriteWebManifestForExtension({
+    name: "Web Extension Name",
+    version: "1.0",
+    manifest_version: 2,
+    applications: {
+      gecko: {
+        id: ID
+      }
+    },
+    icons: {
+      32: "icon32.png",
+      48: "icon48.png",
+      64: "icon64.png",
+      128: "icon128.png",
+      256: "icon256.png"
+    }
+  }, profileDir);
+
+  yield promiseRestartManager();
+  yield promiseAddonStartup();
+
+  let addon = yield promiseAddonByID(ID);
+  do_check_neq(addon, null);
+
+  let uri = do_get_addon_root_uri(profileDir, ID);
+
+  // AddonManager displays larger icons for higher pixel density
+  equal(AddonManager.getPreferredIconURL(addon, 32, {
+    devicePixelRatio: 2
+  }), uri + "icon64.png");
+
+  equal(AddonManager.getPreferredIconURL(addon, 48, {
+    devicePixelRatio: 2
+  }), uri + "icon128.png");
+
+  equal(AddonManager.getPreferredIconURL(addon, 64, {
+    devicePixelRatio: 2
+  }), uri + "icon128.png");
+
+  addon.uninstall();
+
+  yield promiseRestartManager();
+});
+
+// Handles no icons gracefully
+add_task(function*() {
+  yield promiseWriteWebManifestForExtension({
+    name: "Web Extension Name",
+    version: "1.0",
+    manifest_version: 2,
+    applications: {
+      gecko: {
+        id: ID
+      }
+    }
+  }, profileDir);
+
+  yield promiseRestartManager();
+  yield promiseAddonStartup();
+
+  let addon = yield promiseAddonByID(ID);
+  do_check_neq(addon, null);
+
+  let uri = do_get_addon_root_uri(profileDir, ID);
+
+  deepEqual(addon.icons, {});
+
+  equal(addon.iconURL, null);
+  equal(addon.icon64URL, null);
+
+  equal(AddonManager.getPreferredIconURL(addon, 128), null);
+
+  addon.uninstall();
+
+  yield promiseRestartManager();
+});
+