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
|
# 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 pathlib import Path
from marionette_driver import Wait
from marionette_harness import MarionetteTestCase
class PurgeHTTPCacheAtShutdownTestCase(MarionetteTestCase):
def setUp(self):
super().setUp()
self.marionette.enforce_gecko_prefs(
{
"privacy.sanitize.sanitizeOnShutdown": True,
"privacy.clearOnShutdown.cache": True,
"network.cache.shutdown_purge_in_background_task": True,
}
)
self.profile_path = Path(self.marionette.profile_path)
self.cache_path = self.profile_path.joinpath("cache2")
def tearDown(self):
self.marionette.cleanup()
super().tearDown()
def cacheDirExists(self):
return self.cache_path.exists()
def renamedDirExists(self):
return any(
child.name.endswith(".purge.bg_rm") for child in self.profile_path.iterdir()
)
def initLockDir(self):
self.lock_dir = None
with self.marionette.using_context("chrome"):
path = self.marionette.execute_script(
"""
return Services.dirsvc.get("UpdRootD", Ci.nsIFile).parent.parent.path;
"""
)
self.lock_dir = Path(path)
def assertNoLocks(self):
locks = [
x
for x in self.lock_dir.iterdir()
if x.is_file() and "-cachePurge" in x.name
]
self.assertEqual(locks, [], "All locks should have been removed")
# Tests are run in lexicographical order, so test_a* is run first to
# cleanup locks that may be there from previous runs.
def test_asetup(self):
self.initLockDir()
self.marionette.quit()
Wait(self.marionette, timeout=60).until(
lambda _: not self.cacheDirExists() and not self.renamedDirExists(),
message="Cache directory must be removed after orderly shutdown",
)
# delete locks from previous runs
locks = [
x
for x in self.lock_dir.iterdir()
if x.is_file() and "-cachePurge" in x.name
]
for lock in locks:
os.remove(lock)
# all locks should have been removed successfully.
self.assertNoLocks()
def test_ensure_cache_purge_after_in_app_quit(self):
self.assertTrue(self.cacheDirExists(), "Cache directory must exist")
self.initLockDir()
self.marionette.quit()
Wait(self.marionette, timeout=60).until(
lambda _: not self.cacheDirExists() and not self.renamedDirExists(),
message="Cache directory must be removed after orderly shutdown",
)
self.assertNoLocks()
def test_longstanding_cache_purge_after_in_app_quit(self):
self.assertTrue(self.cacheDirExists(), "Cache directory must exist")
self.initLockDir()
self.marionette.set_pref(
"toolkit.background_tasks.remove_directory.testing.sleep_ms", 5000
)
self.marionette.quit()
Wait(self.marionette, timeout=60).until(
lambda _: not self.cacheDirExists() and not self.renamedDirExists(),
message="Cache directory must be removed after orderly shutdown",
)
self.assertNoLocks()
def test_ensure_cache_purge_after_forced_restart(self):
"""
Doing forced restart here to prevent the shutdown phase purging and only allow startup
phase one, via `CacheFileIOManager::OnDelayedStartupFinished`.
"""
self.profile_path.joinpath("foo.purge.bg_rm").mkdir()
self.initLockDir()
self.marionette.restart(in_app=False)
Wait(self.marionette, timeout=60).until(
lambda _: not self.renamedDirExists(),
message="Directories with .purge.bg_rm postfix must be removed at startup after"
"disorderly shutdown",
)
self.assertNoLocks()
|