1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
# 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
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(
"resource://gre/modules/AddonManager.sys.mjs"
);
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(
"resource://gre/modules/AddonManager.sys.mjs"
);
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, "webextension-signed.xpi")
addon_id = self.addons.install(addon_path)
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_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 += "/webextension-signed.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_uninstall_nonexistent_addon(self):
with self.assertRaises(UnknownException):
self.addons.uninstall("i-do-not-exist-as-an-id")
|
