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
|
#!/usr/bin/env 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
import mozunit
import pytest
from mozprofile import (
BaseProfile,
ChromeProfile,
ChromiumProfile,
FirefoxProfile,
Profile,
ThunderbirdProfile,
create_profile,
)
from mozprofile.prefs import Preferences
here = os.path.abspath(os.path.dirname(__file__))
def test_with_profile_should_cleanup():
with Profile() as profile:
assert os.path.exists(profile.profile)
# profile is cleaned
assert not os.path.exists(profile.profile)
def test_with_profile_should_cleanup_even_on_exception():
with pytest.raises(ZeroDivisionError):
# pylint --py3k W1619
with Profile() as profile:
assert os.path.exists(profile.profile)
1 / 0 # will raise ZeroDivisionError
# profile is cleaned
assert not os.path.exists(profile.profile)
@pytest.mark.parametrize(
"app,cls",
[
("chrome", ChromeProfile),
("chromium", ChromiumProfile),
("firefox", FirefoxProfile),
("thunderbird", ThunderbirdProfile),
("unknown", None),
],
)
def test_create_profile(tmpdir, app, cls):
path = tmpdir.strpath
if cls is None:
with pytest.raises(NotImplementedError):
create_profile(app)
return
profile = create_profile(app, profile=path)
assert isinstance(profile, BaseProfile)
assert profile.__class__ == cls
assert profile.profile == path
@pytest.mark.parametrize(
"cls",
[
Profile,
ChromeProfile,
ChromiumProfile,
],
)
def test_merge_profile(cls):
profile = cls(preferences={"foo": "bar"})
assert profile._addons == []
assert os.path.isfile(
os.path.join(profile.profile, profile.preference_file_names[0])
)
other_profile = os.path.join(here, "files", "dummy-profile")
profile.merge(other_profile)
# make sure to add a pref file for each preference_file_names in the dummy-profile
prefs = {}
for name in profile.preference_file_names:
path = os.path.join(profile.profile, name)
assert os.path.isfile(path)
try:
prefs.update(Preferences.read_json(path))
except ValueError:
prefs.update(Preferences.read_prefs(path))
assert "foo" in prefs
assert len(prefs) == len(profile.preference_file_names) + 1
assert all(name in prefs for name in profile.preference_file_names)
# for Google Chrome currently we ignore webext in profile prefs
if cls == Profile:
assert len(profile._addons) == 1
assert profile._addons[0].endswith("empty.xpi")
assert os.path.exists(profile._addons[0])
else:
assert len(profile._addons) == 0
if __name__ == "__main__":
mozunit.main()
|
