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
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
/*
* Infrastructure for the mochitest-chrome tests located in this folder.
*
* See "loader_common.js" in the parent folder for a general overview.
*
* Unless you are adding new features to the framework, you shouldn't have to
* modify this file. Use "head_common.js" or "head.js" for shared code.
*/
"use strict";
var { classes: Cc, interfaces: Ci, utils: Cu, results: Cr } = Components;
Cu.import("resource://gre/modules/XPCOMUtils.jsm", this);
Cu.import("resource://gre/modules/Services.jsm", this);
Services.scriptloader.loadSubScript(
"chrome://mochikit/content/tests/SimpleTest/SimpleTest.js", this);
var sharedUrl = SimpleTest.getTestFileURL("loader_common.js");
Services.scriptloader.loadSubScript(sharedUrl, this);
var parentScript = SpecialPowers.loadChromeScript(
SimpleTest.getTestFileURL("loader_parent.js"));
// Replace the extension of the loaded HTML file with ".js"
var testUrl = location.href.replace(/\.\w+$/, ".js");
// Start loading the test script in the parent process.
var promiseParentInitFinished = new Promise(function (resolve) {
parentScript.addMessageListener("finish_load_in_parent", resolve);
});
parentScript.sendAsyncMessage("start_load_in_parent", { testUrl: testUrl });
// Define output functions so they look the same across all frameworks.
var Output = {
print: info,
};
// Define assertion functions so they look the same across all frameworks.
var Assert = {
ok: _mochitestAssert.ok,
equal: _mochitestAssert.equal,
};
var executeSoon = SimpleTest.executeSoon;
var gTestTasks = [];
// Define task registration functions, see description in "loader_common.js".
function add_task(taskFn) {
gTestTasks.push([taskFn, "content", taskFn.name]);
}
function add_task_in_parent_process(taskFn, taskIdOverride) {
let taskId = taskIdOverride || getTaskId(Components.stack.caller);
gTestTasks.push([taskFn, "parent", taskId]);
}
function add_task_in_both_processes(taskFn) {
// We need to define a task ID based on our direct caller.
add_task_in_parent_process(taskFn, getTaskId(Components.stack.caller));
add_task(taskFn);
}
var add_task_in_child_process = add_task;
window.addEventListener("load", function onLoad() {
window.removeEventListener("load", onLoad);
Task.spawn(function* () {
try {
for (let [taskFn, taskType, taskId] of gTestTasks) {
if (taskType == "content") {
// This is a normal task executed in the current process.
info("Running " + taskFn.name);
yield Task.spawn(taskFn);
} else {
// This is a task executed in the parent process.
info("Running task in parent process: " + taskFn.name);
let promiseFinished = new Promise(function (resolve) {
parentScript.addMessageListener("finish_task_" + taskId, resolve);
});
parentScript.sendAsyncMessage("start_task_" + taskId);
yield promiseFinished;
info("Finished task in parent process: " + taskFn.name);
}
}
} catch (ex) {
ok(false, ex);
}
SimpleTest.finish();
});
});
// Wait for the test script to be loaded in the parent process. This means that
// test tasks are registered and ready, but have not been executed yet.
add_task(function* wait_loading_in_parent_process() {
yield promiseParentInitFinished;
});
var headUrl = SimpleTest.getTestFileURL("head_common.js");
Services.scriptloader.loadSubScript(headUrl, this);
Output.print("Loading test file: " + testUrl);
Services.scriptloader.loadSubScript(testUrl, this);
// Register the execution of termination tasks after all other tasks.
add_task(terminationTaskFn);
add_task_in_parent_process(terminationTaskFn, terminationTaskFn.name);
SimpleTest.waitForExplicitFinish();
// Reminder: unless you are adding new features to the framework, you shouldn't
// have to modify this file. Use "head_common.js" or "head.js" for shared code.
|
