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
141
142
|
# 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/.
from pathlib import Path
from marionette_driver import Wait
from marionette_harness import MarionetteTestCase
class MovedOriginDirectoryCleanupTestCase(MarionetteTestCase):
def setUp(self):
super().setUp()
self.marionette.enforce_gecko_prefs(
{
"privacy.sanitize.sanitizeOnShutdown": True,
"privacy.clearOnShutdown.offlineApps": True,
"dom.quotaManager.backgroundTask.enabled": False,
}
)
self.moved_origin_directory = (
Path(self.marionette.profile_path) / "storage" / "to-be-removed" / "foo"
)
self.moved_origin_directory.mkdir(parents=True, exist_ok=True)
self.to_be_removed_directory = (
Path(self.marionette.profile_path) / "storage" / "to-be-removed"
)
# Add a cookie to get a principal to be cleaned up
with self.marionette.using_context("chrome"):
self.marionette.execute_script(
"""
Services.cookies.add(
"example.local",
"path",
"name",
"value",
false,
false,
false,
Math.floor(Date.now() / 1000) + 24 * 60 * 60,
{},
Ci.nsICookie.SAMESITE_NONE,
Ci.nsICookie.SCHEME_UNSET
);
"""
)
def removeAllCookies(self):
with self.marionette.using_context("chrome"):
self.marionette.execute_script(
"""
Services.cookies.removeAll();
"""
)
def test_ensure_cleanup_by_quit(self):
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must exist",
)
# Cleanup happens via Sanitizer.sanitizeOnShutdown
self.marionette.quit()
self.assertFalse(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must disappear",
)
def test_ensure_cleanup_at_crashed_restart(self):
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must exist",
)
# Pending sanitization is added via Sanitizer.onStartup
# "offlineApps" corresponds to CLEAR_DOM_STORAGES
Wait(self.marionette).until(
lambda _: (
"offlineApps" in self.marionette.get_pref("privacy.sanitize.pending"),
),
message="privacy.sanitize.pending must include offlineApps",
)
# Cleanup happens via Sanitizer.onStartup after restart
self.marionette.restart(in_app=False)
# Wait longer for 30 sec for the restart to finish, given bug 1814281.
Wait(self.marionette, timeout=30).until(
lambda _: not self.moved_origin_directory.exists(),
message="to-be-removed subdirectory must disappear",
)
def test_ensure_cleanup_by_quit_with_background_task(self):
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must exist",
)
self.marionette.set_pref("dom.quotaManager.backgroundTask.enabled", True)
# Cleanup happens via Sanitizer.sanitizeOnShutdown
self.marionette.quit()
Wait(self.marionette).until(
lambda _: not self.moved_origin_directory.exists(),
message="to-be-removed subdirectory must disappear",
)
self.assertTrue(
self.to_be_removed_directory.exists(),
"to-be-removed parent directory should still be alive",
)
def test_ensure_no_cleanup_when_disabled(self):
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must exist",
)
self.marionette.set_pref("privacy.sanitize.sanitizeOnShutdown", False)
self.marionette.quit()
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must not disappear",
)
def test_ensure_no_cleanup_when_no_cookie(self):
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must exist",
)
self.removeAllCookies()
self.marionette.quit()
self.assertTrue(
self.moved_origin_directory.exists(),
"to-be-removed subdirectory must not disappear",
)
|
