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
|
/* Any copyright is dedicated to the Public Domain.
http://creativecommons.org/publicdomain/zero/1.0/ */
// Import common head.
{
/* import-globals-from ../head_common.js */
let commonFile = do_get_file("../head_common.js", false);
let uri = Services.io.newFileURI(commonFile);
Services.scriptloader.loadSubScript(uri.spec, this);
}
// Put any other stuff relative to this test folder below.
ChromeUtils.defineESModuleGetters(this, {
PlacesDBUtils: "resource://gre/modules/PlacesDBUtils.sys.mjs",
});
async function createCorruptDb(filename) {
let path = PathUtils.join(PathUtils.profileDir, filename);
await IOUtils.remove(path, { ignoreAbsent: true });
// Create a corrupt database.
let dir = do_get_cwd().path;
let src = PathUtils.join(dir, "corruptDB.sqlite");
await IOUtils.copy(src, path);
}
/**
* Used in _replaceOnStartup_ tests as common test code. It checks whether we
* are properly cloning or replacing a corrupt database.
*
* @param {string[]} src
* Array of strings which form a path to a test database, relative to
* the parent of this test folder.
* @param {string} filename
* Database file name
* @param {boolean} shouldClone
* Whether we expect the database to be cloned
* @param {boolean} dbStatus
* The expected final database status
*/
async function test_database_replacement(src, filename, shouldClone, dbStatus) {
registerCleanupFunction(() => {
Services.prefs.clearUserPref("places.database.cloneOnCorruption");
});
Services.prefs.setBoolPref("places.database.cloneOnCorruption", shouldClone);
// Only the main database file (places.sqlite) will be cloned, because
// attached databased would break due to OS file lockings.
let willClone = shouldClone && filename == DB_FILENAME;
// Ensure that our databases don't exist yet.
let dest = PathUtils.join(PathUtils.profileDir, filename);
Assert.ok(
!(await IOUtils.exists(dest)),
`"${filename} should not exist initially`
);
let corrupt = PathUtils.join(PathUtils.profileDir, `${filename}.corrupt`);
Assert.ok(
!(await IOUtils.exists(corrupt)),
`${filename}.corrupt should not exist initially`
);
let dir = PathUtils.parent(do_get_cwd().path);
src = PathUtils.join(dir, ...src);
await IOUtils.copy(src, dest);
// Create some unique stuff to check later.
let db = await Sqlite.openConnection({ path: dest });
await db.execute(`CREATE TABLE moz_cloned (id INTEGER PRIMARY KEY)`);
await db.execute(`CREATE TABLE not_cloned (id INTEGER PRIMARY KEY)`);
await db.execute(`DELETE FROM moz_cloned`); // Shouldn't throw.
await db.close();
// Open the database with Places.
Services.prefs.setCharPref(
"places.database.replaceDatabaseOnStartup",
filename
);
Assert.equal(PlacesUtils.history.databaseStatus, dbStatus);
Assert.ok(await IOUtils.exists(dest), "The database should exist");
// Check the new database still contains our special data.
db = await Sqlite.openConnection({ path: dest });
if (willClone) {
await db.execute(`DELETE FROM moz_cloned`); // Shouldn't throw.
}
// Check the new database is really a new one.
await Assert.rejects(
db.execute(`DELETE FROM not_cloned`),
/no such table/,
"The database should have been replaced"
);
await db.close();
if (willClone) {
Assert.ok(
!(await IOUtils.exists(corrupt)),
"The corrupt db should not exist"
);
Assert.ok(
!(await IOUtils.exists(corrupt + "-wal")),
"The corrupt db wal should not exist"
);
Assert.ok(
!(await IOUtils.exists(corrupt + "-shm")),
"The corrupt db shm should not exist"
);
} else {
Assert.ok(await IOUtils.exists(corrupt), "The corrupt db should exist");
}
Assert.equal(
Services.prefs.getCharPref("places.database.replaceDatabaseOnStartup", ""),
"",
"The replaceDatabaseOnStartup pref should have been unset"
);
}
|
