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
|
"use strict";
var { MigrationUtils } = ChromeUtils.importESModule(
"resource:///modules/MigrationUtils.sys.mjs"
);
var { LoginHelper } = ChromeUtils.import(
"resource://gre/modules/LoginHelper.jsm"
);
var { NetUtil } = ChromeUtils.import("resource://gre/modules/NetUtil.jsm");
var { PlacesUtils } = ChromeUtils.importESModule(
"resource://gre/modules/PlacesUtils.sys.mjs"
);
var { Preferences } = ChromeUtils.importESModule(
"resource://gre/modules/Preferences.sys.mjs"
);
var { PromiseUtils } = ChromeUtils.importESModule(
"resource://gre/modules/PromiseUtils.sys.mjs"
);
var { XPCOMUtils } = ChromeUtils.importESModule(
"resource://gre/modules/XPCOMUtils.sys.mjs"
);
var { TestUtils } = ChromeUtils.importESModule(
"resource://testing-common/TestUtils.sys.mjs"
);
var { PlacesTestUtils } = ChromeUtils.importESModule(
"resource://testing-common/PlacesTestUtils.sys.mjs"
);
ChromeUtils.defineESModuleGetters(this, {
FileUtils: "resource://gre/modules/FileUtils.sys.mjs",
Sqlite: "resource://gre/modules/Sqlite.sys.mjs",
});
// Initialize profile.
var gProfD = do_get_profile();
var { getAppInfo, newAppInfo, updateAppInfo } = ChromeUtils.importESModule(
"resource://testing-common/AppInfo.sys.mjs"
);
updateAppInfo();
/**
* Migrates the requested resource and waits for the migration to be complete.
*
* @param {MigratorBase} migrator
* The migrator being used to migrate the data.
* @param {number} resourceType
* This is a bitfield with bits from nsIBrowserProfileMigrator flipped to indicate what
* resources should be migrated.
* @param {object|string|null} [aProfile=null]
* The profile to be migrated. If set to null, the default profile will be
* migrated.
* @param {boolean} succeeds
* True if this migration is expected to succeed.
* @returns {Promise<Array<string[]>>}
* An array of the results from each nsIObserver topics being observed to
* verify if the migration succeeded or failed. Those results are 2-element
* arrays of [subject, data].
*/
async function promiseMigration(
migrator,
resourceType,
aProfile = null,
succeeds = null
) {
// Ensure resource migration is available.
let availableSources = await migrator.getMigrateData(aProfile);
Assert.ok(
(availableSources & resourceType) > 0,
"Resource supported by migrator"
);
let promises = [TestUtils.topicObserved("Migration:Ended")];
if (succeeds !== null) {
// Check that the specific resource type succeeded
promises.push(
TestUtils.topicObserved(
succeeds ? "Migration:ItemAfterMigrate" : "Migration:ItemError",
(_, data) => data == resourceType
)
);
}
// Start the migration.
migrator.migrate(resourceType, null, aProfile);
return Promise.all(promises);
}
/**
* Replaces a directory service entry with a given nsIFile.
*
* @param {string} key
* The nsIDirectoryService directory key to register a fake path for.
* For example: "AppData", "ULibDir".
* @param {nsIFile} file
* The nsIFile to map the key to. Note that this nsIFile should represent
* a directory and not an individual file.
* @see nsDirectoryServiceDefs.h for the list of directories that can be
* overridden.
*/
function registerFakePath(key, file) {
let dirsvc = Services.dirsvc.QueryInterface(Ci.nsIProperties);
let originalFile;
try {
// If a file is already provided save it and undefine, otherwise set will
// throw for persistent entries (ones that are cached).
originalFile = dirsvc.get(key, Ci.nsIFile);
dirsvc.undefine(key);
} catch (e) {
// dirsvc.get will throw if nothing provides for the key and dirsvc.undefine
// will throw if it's not a persistent entry, in either case we don't want
// to set the original file in cleanup.
originalFile = undefined;
}
dirsvc.set(key, file);
registerCleanupFunction(() => {
dirsvc.undefine(key);
if (originalFile) {
dirsvc.set(key, originalFile);
}
});
}
|
