mirror of
https://repo.dactyloidae.xyz/Dactyloidae/UXP.git
synced 2026-09-24 09:27:31 +09:00
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
# 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
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import os
|
|
|
|
from marionette_driver.addons import Addons, AddonInstallException
|
|
from marionette_harness import MarionetteTestCase, skip
|
|
|
|
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
|
|
class TestAddons(MarionetteTestCase):
|
|
|
|
def setUp(self):
|
|
MarionetteTestCase.setUp(self)
|
|
self.addons = Addons(self.marionette)
|
|
|
|
@property
|
|
def all_addon_ids(self):
|
|
with self.marionette.using_context('chrome'):
|
|
addons = self.marionette.execute_async_script("""
|
|
Components.utils.import("resource://gre/modules/AddonManager.jsm");
|
|
AddonManager.getAllAddons(function(addons){
|
|
let ids = addons.map(function(x) {
|
|
return x.id;
|
|
});
|
|
marionetteScriptFinished(ids);
|
|
});
|
|
""")
|
|
|
|
return addons
|
|
|
|
def test_install_and_remove_temporary_unsigned_addon(self):
|
|
addon_path = os.path.join(here, 'mn-restartless-unsigned.xpi')
|
|
|
|
addon_id = self.addons.install(addon_path, temp=True)
|
|
self.assertIn(addon_id, self.all_addon_ids)
|
|
|
|
self.addons.uninstall(addon_id)
|
|
self.assertNotIn(addon_id, self.all_addon_ids)
|
|
|
|
def test_install_unsigned_addon(self):
|
|
addon_path = os.path.join(here, 'mn-restartless-unsigned.xpi')
|
|
|
|
with self.assertRaises(AddonInstallException):
|
|
self.addons.install(addon_path)
|
|
|
|
@skip("Need to get the test extension signed")
|
|
def test_install_and_remove_signed_addon(self):
|
|
addon_path = os.path.join(here, 'mn-restartless-signed.xpi')
|
|
|
|
addon_id = self.addons.install(addon_path)
|
|
self.assertIn(addon_id, self.all_addon_ids)
|
|
|
|
self.addons.uninstall(addon_id)
|
|
self.assertNotIn(addon_id, self.all_addon_ids)
|