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
|
# 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 shutil
from marionette_harness import MarionetteTestCase
class TestAutoConfig(MarionetteTestCase):
def tearDown(self):
self.marionette.quit(in_app=False, clean=True)
if hasattr(self, "pref_file"):
os.remove(self.pref_file)
if hasattr(self, "autoconfig_file"):
os.remove(self.autoconfig_file)
if hasattr(self, "pref_file_dir_created"):
os.rmdir(self.pref_file_dir)
super(TestAutoConfig, self).tearDown()
def pref_has_user_value(self, pref):
with self.marionette.using_context("chrome"):
return self.marionette.execute_script(
"""
return Services.prefs.prefHasUserValue(arguments[0]);
""",
script_args=(pref,),
)
def pref_is_locked(self, pref):
with self.marionette.using_context("chrome"):
return self.marionette.execute_script(
"""
return Services.prefs.prefIsLocked(arguments[0]);
""",
script_args=(pref,),
)
def test_autoconfig(self):
with self.marionette.using_context("chrome"):
self.exe_dir = self.marionette.execute_script(
"""
return Services.dirsvc.get("GreD", Ci.nsIFile).path;
"""
)
self.marionette.quit()
test_dir = os.path.dirname(__file__)
self.pref_file_dir = os.path.join(self.exe_dir, "defaults", "pref")
if not os.path.exists(self.pref_file_dir):
os.makedirs(self.pref_file_dir, exist_ok=True)
self.pref_file_dir_created = True
self.pref_file = os.path.join(self.pref_file_dir, "autoconfig.js")
shutil.copyfile(os.path.join(test_dir, "autoconfig.js"), self.pref_file)
self.autoconfig_file = os.path.join(self.exe_dir, "autoconfig.cfg")
shutil.copyfile(os.path.join(test_dir, "autoconfig.cfg"), self.autoconfig_file)
self.marionette.start_session()
with self.marionette.using_context("chrome"):
self.assertTrue(
self.pref_has_user_value("_autoconfig_.test.userpref"),
"Pref should have user value",
)
self.assertEqual(
self.marionette.get_pref("_autoconfig_.test.userpref"),
"userpref",
"User pref should be set",
)
self.assertEqual(
self.marionette.get_pref("_autoconfig_.test.defaultpref", True),
"defaultpref",
"Default pref should be set",
)
self.assertTrue(
self.pref_is_locked("_autoconfig_.test.lockpref"),
"Pref should be locked",
)
self.assertEqual(
self.marionette.get_pref("_autoconfig_.test.lockpref"),
"lockpref",
"Locked pref should be set",
)
self.assertFalse(
self.pref_is_locked("_autoconfig_.test.unlockpref"),
"Pref should be unlocked",
)
self.assertEqual(
self.marionette.get_pref("_autoconfig_.test.unlockpref"),
"unlockpref",
"Unlocked pref should be set",
)
self.assertFalse(
self.pref_has_user_value("_autoconfig_.test.clearpref"),
"Pref should be cleared",
)
|
