Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import os
import sys
from unittest import skipIf
from marionette_driver.addons import Addons, AddonInstallException
from marionette_driver.errors import UnknownException
from marionette_harness import MarionetteTestCase
here = os.path.abspath(os.path.dirname(__file__))
class TestAddons(MarionetteTestCase):
def setUp(self):
super(TestAddons, self).setUp()
self.addons = Addons(self.marionette)
self.preinstalled_addons = self.all_addon_ids
def tearDown(self):
self.reset_addons()
super(TestAddons, self).tearDown()
@property
def all_addon_ids(self):
with self.marionette.using_context("chrome"):
addons = self.marionette.execute_async_script(
"""
const [resolve] = arguments;
const { AddonManager } = ChromeUtils.importESModule(
);
async function getAllAddons() {
const addons = await AddonManager.getAllAddons();
const ids = addons.map(x => x.id);
resolve(ids);
}
getAllAddons();
"""
)
return set(addons)
def reset_addons(self):
with self.marionette.using_context("chrome"):
for addon in self.all_addon_ids - self.preinstalled_addons:
addon_id = self.marionette.execute_async_script(
"""
const [addonId, resolve] = arguments;
const { AddonManager } = ChromeUtils.importESModule(
);
async function uninstall() {
const addon = await AddonManager.getAddonByID(addonId);
addon.uninstall();
resolve(addon.id);
}
uninstall();
""",
script_args=(addon,),
)
self.assertEqual(
addon_id, addon, msg="Failed to uninstall {}".format(addon)
)
def test_temporary_install_and_remove_unsigned_addon(self):
addon_path = os.path.join(here, "webextension-unsigned.xpi")
addon_id = self.addons.install(addon_path, temp=True)
self.assertIn(addon_id, self.all_addon_ids)
self.assertEqual(addon_id, "{d3e7c1f1-2e35-4a49-89fe-9f46eb8abf0a}")
self.addons.uninstall(addon_id)
self.assertNotIn(addon_id, self.all_addon_ids)
def test_temporary_install_invalid_addon(self):
addon_path = os.path.join(here, "webextension-invalid.xpi")
with self.assertRaises(AddonInstallException):
self.addons.install(addon_path, temp=True)
self.assertNotIn("{d3e7c1f1-2e35-4a49-89fe-9f46eb8abf0a}", self.all_addon_ids)
def test_install_and_remove_signed_addon(self):
addon_path = os.path.join(here, "amosigned.xpi")
addon_id = self.addons.install(addon_path)
self.assertIn(addon_id, self.all_addon_ids)
self.assertEqual(addon_id, "amosigned-xpi@tests.mozilla.org")
self.addons.uninstall(addon_id)
self.assertNotIn(addon_id, self.all_addon_ids)
def test_install_invalid_addon(self):
addon_path = os.path.join(here, "webextension-invalid.xpi")
with self.assertRaises(AddonInstallException):
self.addons.install(addon_path)
self.assertNotIn("{d3e7c1f1-2e35-4a49-89fe-9f46eb8abf0a}", self.all_addon_ids)
def test_install_unsigned_addon_fails(self):
addon_path = os.path.join(here, "webextension-unsigned.xpi")
with self.assertRaises(AddonInstallException):
self.addons.install(addon_path)
def test_install_nonexistent_addon(self):
addon_path = os.path.join(here, "does-not-exist.xpi")
with self.assertRaises(AddonInstallException):
self.addons.install(addon_path)
def test_install_with_relative_path(self):
with self.assertRaises(AddonInstallException):
self.addons.install("webextension.xpi")
@skipIf(sys.platform != "win32", "Only makes sense on Windows")
def test_install_mixed_separator_windows(self):
# Ensure the base path has only \
addon_path = here.replace("/", "\\")
addon_path += "/amosigned.xpi"
addon_id = self.addons.install(addon_path, temp=True)
self.assertIn(addon_id, self.all_addon_ids)
self.assertEqual(addon_id, "amosigned-xpi@tests.mozilla.org")
self.addons.uninstall(addon_id)
self.assertNotIn(addon_id, self.all_addon_ids)
def test_uninstall_nonexistent_addon(self):
with self.assertRaises(UnknownException):
self.addons.uninstall("i-do-not-exist-as-an-id")