gecko-dev/testing/xpcshell/head.js

1743 строки
49 KiB
JavaScript

/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et: */
/* 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/. */
/*
* This file contains common code that is loaded before each test file(s).
* See http://developer.mozilla.org/en/docs/Writing_xpcshell-based_unit_tests
* for more information.
*/
/* defined by the harness */
/* globals _HEAD_FILES, _HEAD_JS_PATH, _JSDEBUGGER_PORT, _JSCOV_DIR,
_MOZINFO_JS_PATH, _TEST_FILE, _TEST_NAME, _TESTING_MODULES_DIR:true,
_PREFS_FILE */
/* defined by XPCShellImpl.cpp */
/* globals load, sendCommand */
/* must be defined by tests using do_await_remote_message/do_send_remote_message */
/* globals Cc, Ci */
/* defined by this file but is defined as read-only for tests */
// eslint-disable-next-line no-redeclare
/* globals runningInParent: true */
/* may be defined in test files */
/* globals run_test */
var _quit = false;
var _passed = true;
var _tests_pending = 0;
var _cleanupFunctions = [];
var _pendingTimers = [];
var _profileInitialized = false;
var _fastShutdownDisabled = false;
// Assigned in do_load_child_test_harness.
var _XPCSHELL_PROCESS;
// Register the testing-common resource protocol early, to have access to its
// modules.
var _Services = ChromeUtils.import("resource://gre/modules/Services.jsm", null)
.Services;
_register_modules_protocol_handler();
var _AppConstants = ChromeUtils.import(
"resource://gre/modules/AppConstants.jsm",
null
).AppConstants;
var _PromiseTestUtils = ChromeUtils.import(
"resource://testing-common/PromiseTestUtils.jsm",
null
).PromiseTestUtils;
var _Task = ChromeUtils.import("resource://testing-common/Task.jsm", null).Task;
let _NetUtil = ChromeUtils.import("resource://gre/modules/NetUtil.jsm", null)
.NetUtil;
let _XPCOMUtils = ChromeUtils.import(
"resource://gre/modules/XPCOMUtils.jsm",
null
).XPCOMUtils;
// Support a common assertion library, Assert.jsm.
var AssertCls = ChromeUtils.import("resource://testing-common/Assert.jsm", null)
.Assert;
// Pass a custom report function for xpcshell-test style reporting.
var Assert = new AssertCls(function(err, message, stack) {
if (err) {
do_report_result(false, err.message, err.stack);
} else {
do_report_result(true, message, stack);
}
}, true);
// Bug 1506134 for followup. Some xpcshell tests use ContentTask.jsm, which
// expects browser-test.js to have set a testScope that includes record.
function record(condition, name, diag, stack) {
do_report_result(condition, name, stack);
}
var _add_params = function(params) {
if (typeof _XPCSHELL_PROCESS != "undefined") {
params.xpcshell_process = _XPCSHELL_PROCESS;
}
};
var _dumpLog = function(raw_msg) {
dump("\n" + JSON.stringify(raw_msg) + "\n");
};
var _LoggerClass = ChromeUtils.import(
"resource://testing-common/StructuredLog.jsm",
null
).StructuredLogger;
var _testLogger = new _LoggerClass("xpcshell/head.js", _dumpLog, [_add_params]);
// Disable automatic network detection, so tests work correctly when
// not connected to a network.
_Services.io.manageOfflineStatus = false;
_Services.io.offline = false;
// Determine if we're running on parent or child
var runningInParent = true;
try {
// Don't use Services.appinfo here as it disables replacing appinfo with stubs
// for test usage.
runningInParent =
// eslint-disable-next-line mozilla/use-services
Cc["@mozilla.org/xre/runtime;1"].getService(Ci.nsIXULRuntime).processType ==
Ci.nsIXULRuntime.PROCESS_TYPE_DEFAULT;
} catch (e) {}
// Only if building of places is enabled.
if (runningInParent && "mozIAsyncHistory" in Ci) {
// Ensure places history is enabled for xpcshell-tests as some non-FF
// apps disable it.
_Services.prefs.setBoolPref("places.history.enabled", true);
}
try {
if (runningInParent) {
// Disable IPv6 lookups for 'localhost' on windows.
if ("@mozilla.org/windows-registry-key;1" in Cc) {
_Services.prefs.setCharPref("network.dns.ipv4OnlyDomains", "localhost");
}
}
} catch (e) {}
// Configure crash reporting, if possible
// We rely on the Python harness to set MOZ_CRASHREPORTER,
// MOZ_CRASHREPORTER_NO_REPORT, and handle checking for minidumps.
// Note that if we're in a child process, we don't want to init the
// crashreporter component.
try {
if (runningInParent && "@mozilla.org/toolkit/crash-reporter;1" in Cc) {
let crashReporter = Cc["@mozilla.org/toolkit/crash-reporter;1"].getService(
Ci.nsICrashReporter
);
crashReporter.UpdateCrashEventsDir();
crashReporter.minidumpPath = do_get_minidumpdir();
}
} catch (e) {}
if (runningInParent) {
_Services.prefs.setBoolPref("dom.push.connection.enabled", false);
}
// Configure a console listener so messages sent to it are logged as part
// of the test.
try {
let levelNames = {};
for (let level of ["debug", "info", "warn", "error"]) {
levelNames[Ci.nsIConsoleMessage[level]] = level;
}
let listener = {
QueryInterface: ChromeUtils.generateQI(["nsIConsoleListener"]),
observe(msg) {
if (typeof info === "function") {
info(
"CONSOLE_MESSAGE: (" +
levelNames[msg.logLevel] +
") " +
msg.toString()
);
}
},
};
// Don't use _Services.console here as it causes one of the devtools tests
// to fail, probably due to initializing Services.console too early.
// eslint-disable-next-line mozilla/use-services
Cc["@mozilla.org/consoleservice;1"]
.getService(Ci.nsIConsoleService)
.registerListener(listener);
} catch (e) {}
/**
* Date.now() is not necessarily monotonically increasing (insert sob story
* about times not being the right tool to use for measuring intervals of time,
* robarnold can tell all), so be wary of error by erring by at least
* _timerFuzz ms.
*/
const _timerFuzz = 15;
function _Timer(func, delay) {
delay = Number(delay);
if (delay < 0) {
do_throw("do_timeout() delay must be nonnegative");
}
if (typeof func !== "function") {
do_throw("string callbacks no longer accepted; use a function!");
}
this._func = func;
this._start = Date.now();
this._delay = delay;
var timer = Cc["@mozilla.org/timer;1"].createInstance(Ci.nsITimer);
timer.initWithCallback(this, delay + _timerFuzz, timer.TYPE_ONE_SHOT);
// Keep timer alive until it fires
_pendingTimers.push(timer);
}
_Timer.prototype = {
QueryInterface: ChromeUtils.generateQI(["nsITimerCallback"]),
notify(timer) {
_pendingTimers.splice(_pendingTimers.indexOf(timer), 1);
// The current nsITimer implementation can undershoot, but even if it
// couldn't, paranoia is probably a virtue here given the potential for
// random orange on tinderboxen.
var end = Date.now();
var elapsed = end - this._start;
if (elapsed >= this._delay) {
try {
this._func.call(null);
} catch (e) {
do_throw("exception thrown from do_timeout callback: " + e);
}
return;
}
// Timer undershot, retry with a little overshoot to try to avoid more
// undershoots.
var newDelay = this._delay - elapsed;
do_timeout(newDelay, this._func);
},
};
function _isGenerator(val) {
return typeof val === "object" && val && typeof val.next === "function";
}
function _do_main() {
if (_quit) {
return;
}
_testLogger.info("running event loop");
var tm = Cc["@mozilla.org/thread-manager;1"].getService();
tm.spinEventLoopUntil(() => _quit);
tm.spinEventLoopUntilEmpty();
}
function _do_quit() {
_testLogger.info("exiting test");
_quit = true;
}
// This is useless, except to the extent that it has the side-effect of
// initializing the widget module, which some tests unfortunately
// accidentally rely on.
void Cc["@mozilla.org/widget/transferable;1"].createInstance();
/**
* Overrides idleService with a mock. Idle is commonly used for maintenance
* tasks, thus if a test uses a service that requires the idle service, it will
* start handling them.
* This behaviour would cause random failures and slowdown tests execution,
* for example by running database vacuum or cleanups for each test.
*
* @note Idle service is overridden by default. If a test requires it, it will
* have to call do_get_idle() function at least once before use.
*/
var _fakeIdleService = {
get registrar() {
delete this.registrar;
return (this.registrar = Components.manager.QueryInterface(
Ci.nsIComponentRegistrar
));
},
contractID: "@mozilla.org/widget/idleservice;1",
CID: Components.ID("{9163a4ae-70c2-446c-9ac1-bbe4ab93004e}"),
activate: function FIS_activate() {
if (!this.originalCID) {
this.originalCID = this.registrar.contractIDToCID(this.contractID);
// Replace with the mock.
this.registrar.registerFactory(
this.CID,
"Fake Idle Service",
this.contractID,
this.factory
);
}
},
deactivate: function FIS_deactivate() {
if (this.originalCID) {
// Unregister the mock.
this.registrar.unregisterFactory(this.CID, this.factory);
// Restore original factory.
this.registrar.registerFactory(
this.originalCID,
"Idle Service",
this.contractID,
null
);
delete this.originalCID;
}
},
factory: {
// nsIFactory
createInstance(aOuter, aIID) {
if (aOuter) {
throw Components.Exception("", Cr.NS_ERROR_NO_AGGREGATION);
}
return _fakeIdleService.QueryInterface(aIID);
},
lockFactory(aLock) {
throw Components.Exception("", Cr.NS_ERROR_NOT_IMPLEMENTED);
},
QueryInterface: ChromeUtils.generateQI(["nsIFactory"]),
},
// nsIIdleService
get idleTime() {
return 0;
},
addIdleObserver() {},
removeIdleObserver() {},
// eslint-disable-next-line mozilla/use-chromeutils-generateqi
QueryInterface(aIID) {
// Useful for testing purposes, see test_get_idle.js.
if (aIID.equals(Ci.nsIFactory)) {
return this.factory;
}
if (aIID.equals(Ci.nsIIdleService) || aIID.equals(Ci.nsISupports)) {
return this;
}
throw Components.Exception("", Cr.NS_ERROR_NO_INTERFACE);
},
};
/**
* Restores the idle service factory if needed and returns the service's handle.
* @return A handle to the idle service.
*/
function do_get_idle() {
_fakeIdleService.deactivate();
return Cc[_fakeIdleService.contractID].getService(Ci.nsIIdleService);
}
// Map resource://test/ to current working directory and
// resource://testing-common/ to the shared test modules directory.
function _register_protocol_handlers() {
let protocolHandler = _Services.io
.getProtocolHandler("resource")
.QueryInterface(Ci.nsIResProtocolHandler);
let curDirURI = _Services.io.newFileURI(do_get_cwd());
protocolHandler.setSubstitution("test", curDirURI);
_register_modules_protocol_handler();
}
function _register_modules_protocol_handler() {
if (!_TESTING_MODULES_DIR) {
throw new Error(
"Please define a path where the testing modules can be " +
"found in a variable called '_TESTING_MODULES_DIR' before " +
"head.js is included."
);
}
let protocolHandler = _Services.io
.getProtocolHandler("resource")
.QueryInterface(Ci.nsIResProtocolHandler);
let modulesFile = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
modulesFile.initWithPath(_TESTING_MODULES_DIR);
if (!modulesFile.exists()) {
throw new Error(
"Specified modules directory does not exist: " + _TESTING_MODULES_DIR
);
}
if (!modulesFile.isDirectory()) {
throw new Error(
"Specified modules directory is not a directory: " + _TESTING_MODULES_DIR
);
}
let modulesURI = _Services.io.newFileURI(modulesFile);
protocolHandler.setSubstitution("testing-common", modulesURI);
}
/* Debugging support */
// Used locally and by our self-tests.
function _setupDevToolsServer(breakpointFiles, callback) {
// Always allow remote debugging.
_Services.prefs.setBoolPref("devtools.debugger.remote-enabled", true);
// for debugging-the-debugging, let an env var cause log spew.
let env = Cc["@mozilla.org/process/environment;1"].getService(
Ci.nsIEnvironment
);
if (env.get("DEVTOOLS_DEBUGGER_LOG")) {
_Services.prefs.setBoolPref("devtools.debugger.log", true);
}
if (env.get("DEVTOOLS_DEBUGGER_LOG_VERBOSE")) {
_Services.prefs.setBoolPref("devtools.debugger.log.verbose", true);
}
let require;
try {
({ require } = ChromeUtils.import("resource://devtools/shared/Loader.jsm"));
} catch (e) {
throw new Error(
"resource://devtools appears to be inaccessible from the " +
"xpcshell environment.\n" +
"This can usually be resolved by adding:\n" +
" firefox-appdir = browser\n" +
"to the xpcshell.ini manifest.\n" +
"It is possible for this to alter test behevior by " +
"triggering additional browser code to run, so check " +
"test behavior after making this change.\n" +
"See also https://bugzil.la/1215378."
);
}
let { DevToolsServer } = require("devtools/server/devtools-server");
DevToolsServer.init();
DevToolsServer.registerAllActors();
let { createRootActor } = require("resource://testing-common/dbg-actors.js");
DevToolsServer.setRootActor(createRootActor);
DevToolsServer.allowChromeProcess = true;
// An observer notification that tells us when we can "resume" script
// execution.
const TOPICS = [
"devtools-thread-instantiated",
"devtools-thread-resumed",
"xpcshell-test-devtools-shutdown",
];
let observe = function(subject, topic, data) {
if (topic === "devtools-thread-instantiated") {
const threadActor = subject.wrappedJSObject;
threadActor.setBreakpointOnLoad(breakpointFiles);
return;
}
for (let topicToRemove of TOPICS) {
_Services.obs.removeObserver(observe, topicToRemove);
}
callback();
};
for (let topic of TOPICS) {
_Services.obs.addObserver(observe, topic);
}
const { SocketListener } = require("devtools/shared/security/socket");
return { DevToolsServer, SocketListener };
}
function _initDebugging(port) {
let initialized = false;
const { DevToolsServer, SocketListener } = _setupDevToolsServer(
_TEST_FILE,
() => {
initialized = true;
}
);
info("");
info("*******************************************************************");
info("Waiting for the debugger to connect on port " + port);
info("");
info("To connect the debugger, open a Firefox instance, select 'Connect'");
info("from the Developer menu and specify the port as " + port);
info("*******************************************************************");
info("");
const AuthenticatorType = DevToolsServer.Authenticators.get("PROMPT");
const authenticator = new AuthenticatorType.Server();
authenticator.allowConnection = () => {
return DevToolsServer.AuthenticationResult.ALLOW;
};
const socketOptions = {
authenticator,
portOrPath: port,
};
const listener = new SocketListener(DevToolsServer, socketOptions);
listener.open();
// spin an event loop until the debugger connects.
const tm = Cc["@mozilla.org/thread-manager;1"].getService();
tm.spinEventLoopUntil(() => {
if (initialized) {
return true;
}
info("Still waiting for debugger to connect...");
return false;
});
// NOTE: if you want to debug the harness itself, you can now add a 'debugger'
// statement anywhere and it will stop - but we've already added a breakpoint
// for the first line of the test scripts, so we just continue...
info("Debugger connected, starting test execution");
}
function _execute_test() {
// _JSDEBUGGER_PORT is dynamically defined by <runxpcshelltests.py>.
if (_JSDEBUGGER_PORT) {
try {
_initDebugging(_JSDEBUGGER_PORT);
} catch (ex) {
// Fail the test run immediately if debugging is requested but fails, so
// that the failure state is more obvious.
do_throw(`Failed to initialize debugging: ${ex}`, ex.stack);
}
}
_register_protocol_handlers();
// Override idle service by default.
// Call do_get_idle() to restore the factory and get the service.
_fakeIdleService.activate();
_PromiseTestUtils.init();
let coverageCollector = null;
if (typeof _JSCOV_DIR === "string") {
let _CoverageCollector = ChromeUtils.import(
"resource://testing-common/CoverageUtils.jsm",
{}
).CoverageCollector;
coverageCollector = new _CoverageCollector(_JSCOV_DIR);
}
let startTime = Cu.now();
// _HEAD_FILES is dynamically defined by <runxpcshelltests.py>.
_load_files(_HEAD_FILES);
// _TEST_FILE is dynamically defined by <runxpcshelltests.py>.
_load_files(_TEST_FILE);
// Tack Assert.jsm methods to the current scope.
this.Assert = Assert;
for (let func in Assert) {
this[func] = Assert[func].bind(Assert);
}
const { PerTestCoverageUtils } = ChromeUtils.import(
"resource://testing-common/PerTestCoverageUtils.jsm"
);
if (runningInParent) {
PerTestCoverageUtils.beforeTestSync();
}
try {
do_test_pending("MAIN run_test");
// Check if run_test() is defined. If defined, run it.
// Else, call run_next_test() directly to invoke tests
// added by add_test() and add_task().
if (typeof run_test === "function") {
run_test();
} else {
run_next_test();
}
do_test_finished("MAIN run_test");
_do_main();
_PromiseTestUtils.assertNoUncaughtRejections();
if (coverageCollector != null) {
coverageCollector.recordTestCoverage(_TEST_FILE[0]);
}
if (runningInParent) {
PerTestCoverageUtils.afterTestSync();
}
} catch (e) {
_passed = false;
// do_check failures are already logged and set _quit to true and throw
// NS_ERROR_ABORT. If both of those are true it is likely this exception
// has already been logged so there is no need to log it again. It's
// possible that this will mask an NS_ERROR_ABORT that happens after a
// do_check failure though.
if (!_quit || e.result != Cr.NS_ERROR_ABORT) {
let extra = {};
if (e.fileName) {
extra.source_file = e.fileName;
if (e.lineNumber) {
extra.line_number = e.lineNumber;
}
} else {
extra.source_file = "xpcshell/head.js";
}
let message = _exception_message(e);
if (e.stack) {
extra.stack = _format_stack(e.stack);
}
_testLogger.error(message, extra);
}
} finally {
if (coverageCollector != null) {
coverageCollector.finalize();
}
}
// Execute all of our cleanup functions.
let reportCleanupError = function(ex) {
let stack, filename;
if (ex && typeof ex == "object" && "stack" in ex) {
stack = ex.stack;
} else {
stack = Components.stack.caller;
}
if (stack instanceof Ci.nsIStackFrame) {
filename = stack.filename;
} else if (ex.fileName) {
filename = ex.fileName;
}
_testLogger.error(_exception_message(ex), {
stack: _format_stack(stack),
source_file: filename,
});
};
let complete = _cleanupFunctions.length == 0;
let cleanupStartTime = complete ? 0 : Cu.now();
(async () => {
for (let func of _cleanupFunctions.reverse()) {
try {
let result = await func();
if (_isGenerator(result)) {
Assert.ok(false, "Cleanup function returned a generator");
}
} catch (ex) {
reportCleanupError(ex);
}
}
_cleanupFunctions = [];
})()
.catch(reportCleanupError)
.then(() => (complete = true));
_Services.tm.spinEventLoopUntil(() => complete);
if (cleanupStartTime) {
ChromeUtils.addProfilerMarker(
"xpcshell-test",
cleanupStartTime,
"Cleanup functions"
);
}
ChromeUtils.addProfilerMarker("xpcshell-test", startTime, _TEST_NAME);
_Services.obs.notifyObservers(null, "test-complete");
// Restore idle service to avoid leaks.
_fakeIdleService.deactivate();
if (_profileInitialized) {
// Since we have a profile, we will notify profile shutdown topics at
// the end of the current test, to ensure correct cleanup on shutdown.
_Services.obs.notifyObservers(null, "profile-change-net-teardown");
_Services.obs.notifyObservers(null, "profile-change-teardown");
_Services.obs.notifyObservers(null, "profile-before-change");
_Services.obs.notifyObservers(null, "profile-before-change-qm");
_profileInitialized = false;
}
try {
_PromiseTestUtils.ensureDOMPromiseRejectionsProcessed();
_PromiseTestUtils.assertNoUncaughtRejections();
_PromiseTestUtils.assertNoMoreExpectedRejections();
} finally {
// It's important to terminate the module to avoid crashes on shutdown.
_PromiseTestUtils.uninit();
}
// Skip the normal shutdown path for optimized builds that don't do leak checking.
if (
runningInParent &&
!_AppConstants.RELEASE_OR_BETA &&
!_AppConstants.DEBUG &&
!_AppConstants.MOZ_CODE_COVERAGE &&
!_AppConstants.ASAN &&
!_AppConstants.TSAN
) {
if (_fastShutdownDisabled) {
_testLogger.info("fast shutdown disabled by the test.");
return;
}
// Setting this pref is required for Cu.isInAutomation to return true.
_Services.prefs.setBoolPref(
"security.turn_off_all_security_so_that_viruses_can_take_over_this_computer",
true
);
Cu.exitIfInAutomation();
}
}
/**
* Loads files.
*
* @param aFiles Array of files to load.
*/
function _load_files(aFiles) {
function load_file(element, index, array) {
try {
let startTime = Cu.now();
load(element);
ChromeUtils.addProfilerMarker(
"xpcshell-test",
startTime,
"load " + element.replace(/.*\/_?tests\/xpcshell\//, "")
);
} catch (e) {
let extra = {
source_file: element,
};
if (e.stack) {
extra.stack = _format_stack(e.stack);
}
_testLogger.error(_exception_message(e), extra);
}
}
aFiles.forEach(load_file);
}
function _wrap_with_quotes_if_necessary(val) {
return typeof val == "string" ? '"' + val + '"' : val;
}
/* ************* Functions to be used from the tests ************* */
/**
* Prints a message to the output log.
*/
function info(msg, data) {
ChromeUtils.addProfilerMarker("xpcshell-test", undefined, "INFO " + msg);
msg = _wrap_with_quotes_if_necessary(msg);
data = data ? data : null;
_testLogger.info(msg, data);
}
/**
* Calls the given function at least the specified number of milliseconds later.
* The callback will not undershoot the given time, but it might overshoot --
* don't expect precision!
*
* @param delay : uint
* the number of milliseconds to delay
* @param callback : function() : void
* the function to call
*/
function do_timeout(delay, func) {
new _Timer(func, Number(delay));
}
function executeSoon(callback, aName) {
let funcName = aName ? aName : callback.name;
do_test_pending(funcName);
_Services.tm.dispatchToMainThread({
run() {
try {
callback();
} catch (e) {
// do_check failures are already logged and set _quit to true and throw
// NS_ERROR_ABORT. If both of those are true it is likely this exception
// has already been logged so there is no need to log it again. It's
// possible that this will mask an NS_ERROR_ABORT that happens after a
// do_check failure though.
if (!_quit || e.result != Cr.NS_ERROR_ABORT) {
let stack = e.stack ? _format_stack(e.stack) : null;
_testLogger.testStatus(
_TEST_NAME,
funcName,
"FAIL",
"PASS",
_exception_message(e),
stack
);
_do_quit();
}
} finally {
do_test_finished(funcName);
}
},
});
}
/**
* Shows an error message and the current stack and aborts the test.
*
* @param error A message string or an Error object.
* @param stack null or nsIStackFrame object or a string containing
* \n separated stack lines (as in Error().stack).
*/
function do_throw(error, stack) {
let filename = "";
// If we didn't get passed a stack, maybe the error has one
// otherwise get it from our call context
stack = stack || error.stack || Components.stack.caller;
if (stack instanceof Ci.nsIStackFrame) {
filename = stack.filename;
} else if (error.fileName) {
filename = error.fileName;
}
_testLogger.error(_exception_message(error), {
source_file: filename,
stack: _format_stack(stack),
});
_abort_failed_test();
}
function _abort_failed_test() {
// Called to abort the test run after all failures are logged.
_passed = false;
_do_quit();
throw Components.Exception("", Cr.NS_ERROR_ABORT);
}
function _format_stack(stack) {
let normalized;
if (stack instanceof Ci.nsIStackFrame) {
let frames = [];
for (let frame = stack; frame; frame = frame.caller) {
frames.push(frame.filename + ":" + frame.name + ":" + frame.lineNumber);
}
normalized = frames.join("\n");
} else {
normalized = "" + stack;
}
return _Task.Debugging.generateReadableStack(normalized, " ");
}
// Make a nice display string from an object that behaves
// like Error
function _exception_message(ex) {
let message = "";
if (ex.name) {
message = ex.name + ": ";
}
if (ex.message) {
message += ex.message;
}
if (ex.fileName) {
message += " at " + ex.fileName;
if (ex.lineNumber) {
message += ":" + ex.lineNumber;
}
}
if (message !== "") {
return message;
}
// Force ex to be stringified
return "" + ex;
}
function do_report_unexpected_exception(ex, text) {
let filename = Components.stack.caller.filename;
text = text ? text + " - " : "";
_passed = false;
_testLogger.error(text + "Unexpected exception " + _exception_message(ex), {
source_file: filename,
stack: _format_stack(ex.stack),
});
_do_quit();
throw Components.Exception("", Cr.NS_ERROR_ABORT);
}
function do_note_exception(ex, text) {
let filename = Components.stack.caller.filename;
_testLogger.info(text + "Swallowed exception " + _exception_message(ex), {
source_file: filename,
stack: _format_stack(ex.stack),
});
}
function do_report_result(passed, text, stack, todo) {
// Match names like head.js, head_foo.js, and foo_head.js, but not
// test_headache.js
while (/(\/head(_.+)?|head)\.js$/.test(stack.filename) && stack.caller) {
stack = stack.caller;
}
let name = _gRunningTest ? _gRunningTest.name : stack.name;
let message;
if (name) {
message = "[" + name + " : " + stack.lineNumber + "] " + text;
} else {
message = text;
}
if (passed) {
if (todo) {
_testLogger.testStatus(
_TEST_NAME,
name,
"PASS",
"FAIL",
message,
_format_stack(stack)
);
_abort_failed_test();
} else {
_testLogger.testStatus(_TEST_NAME, name, "PASS", "PASS", message);
}
} else if (todo) {
_testLogger.testStatus(_TEST_NAME, name, "FAIL", "FAIL", message);
} else {
_testLogger.testStatus(
_TEST_NAME,
name,
"FAIL",
"PASS",
message,
_format_stack(stack)
);
_abort_failed_test();
}
}
function _do_check_eq(left, right, stack, todo) {
if (!stack) {
stack = Components.stack.caller;
}
var text =
_wrap_with_quotes_if_necessary(left) +
" == " +
_wrap_with_quotes_if_necessary(right);
do_report_result(left == right, text, stack, todo);
}
function todo_check_eq(left, right, stack) {
if (!stack) {
stack = Components.stack.caller;
}
_do_check_eq(left, right, stack, true);
}
function todo_check_true(condition, stack) {
if (!stack) {
stack = Components.stack.caller;
}
todo_check_eq(condition, true, stack);
}
function todo_check_false(condition, stack) {
if (!stack) {
stack = Components.stack.caller;
}
todo_check_eq(condition, false, stack);
}
function todo_check_null(condition, stack = Components.stack.caller) {
todo_check_eq(condition, null, stack);
}
// Check that |func| throws an nsIException that has
// |Components.results[resultName]| as the value of its 'result' property.
function do_check_throws_nsIException(
func,
resultName,
stack = Components.stack.caller,
todo = false
) {
let expected = Cr[resultName];
if (typeof expected !== "number") {
do_throw(
"do_check_throws_nsIException requires a Components.results" +
" property name, not " +
uneval(resultName),
stack
);
}
let msg =
"do_check_throws_nsIException: func should throw" +
" an nsIException whose 'result' is Components.results." +
resultName;
try {
func();
} catch (ex) {
if (!(ex instanceof Ci.nsIException) || ex.result !== expected) {
do_report_result(
false,
msg + ", threw " + legible_exception(ex) + " instead",
stack,
todo
);
}
do_report_result(true, msg, stack, todo);
return;
}
// Call this here, not in the 'try' clause, so do_report_result's own
// throw doesn't get caught by our 'catch' clause.
do_report_result(false, msg + ", but returned normally", stack, todo);
}
// Produce a human-readable form of |exception|. This looks up
// Components.results values, tries toString methods, and so on.
function legible_exception(exception) {
switch (typeof exception) {
case "object":
if (exception instanceof Ci.nsIException) {
return "nsIException instance: " + uneval(exception.toString());
}
return exception.toString();
case "number":
for (let name in Cr) {
if (exception === Cr[name]) {
return "Components.results." + name;
}
}
// Fall through.
default:
return uneval(exception);
}
}
function do_check_instanceof(
value,
constructor,
stack = Components.stack.caller,
todo = false
) {
do_report_result(
value instanceof constructor,
"value should be an instance of " + constructor.name,
stack,
todo
);
}
function todo_check_instanceof(
value,
constructor,
stack = Components.stack.caller
) {
do_check_instanceof(value, constructor, stack, true);
}
function do_test_pending(aName) {
++_tests_pending;
_testLogger.info(
"(xpcshell/head.js) | test" +
(aName ? " " + aName : "") +
" pending (" +
_tests_pending +
")"
);
}
function do_test_finished(aName) {
_testLogger.info(
"(xpcshell/head.js) | test" +
(aName ? " " + aName : "") +
" finished (" +
_tests_pending +
")"
);
if (--_tests_pending == 0) {
_do_quit();
}
}
function do_get_file(path, allowNonexistent) {
try {
let lf = _Services.dirsvc.get("CurWorkD", Ci.nsIFile);
let bits = path.split("/");
for (let i = 0; i < bits.length; i++) {
if (bits[i]) {
if (bits[i] == "..") {
lf = lf.parent;
} else {
lf.append(bits[i]);
}
}
}
if (!allowNonexistent && !lf.exists()) {
// Not using do_throw(): caller will continue.
_passed = false;
var stack = Components.stack.caller;
_testLogger.error(
"[" +
stack.name +
" : " +
stack.lineNumber +
"] " +
lf.path +
" does not exist"
);
}
return lf;
} catch (ex) {
do_throw(ex.toString(), Components.stack.caller);
}
return null;
}
// do_get_cwd() isn't exactly self-explanatory, so provide a helper
function do_get_cwd() {
return do_get_file("");
}
function do_load_manifest(path) {
var lf = do_get_file(path);
const nsIComponentRegistrar = Ci.nsIComponentRegistrar;
Assert.ok(Components.manager instanceof nsIComponentRegistrar);
// Previous do_check_true() is not a test check.
Components.manager.autoRegister(lf);
}
/**
* Parse a DOM document.
*
* @param aPath File path to the document.
* @param aType Content type to use in DOMParser.
*
* @return Document from the file.
*/
function do_parse_document(aPath, aType) {
switch (aType) {
case "application/xhtml+xml":
case "application/xml":
case "text/xml":
break;
default:
do_throw(
"type: expected application/xhtml+xml, application/xml or text/xml," +
" got '" +
aType +
"'",
Components.stack.caller
);
}
let file = do_get_file(aPath);
let url = _Services.io.newFileURI(file).spec;
file = null;
return new Promise((resolve, reject) => {
let xhr = new XMLHttpRequest();
xhr.open("GET", url);
xhr.responseType = "document";
xhr.onerror = reject;
xhr.onload = () => {
resolve(xhr.response);
};
xhr.send();
});
}
/**
* Registers a function that will run when the test harness is done running all
* tests.
*
* @param aFunction
* The function to be called when the test harness has finished running.
*/
function registerCleanupFunction(aFunction) {
_cleanupFunctions.push(aFunction);
}
/**
* Ensure the test finishes with a normal shutdown even when it could have
* otherwise used the fast Cu.exitIfInAutomation shutdown.
*/
function do_disable_fast_shutdown() {
_fastShutdownDisabled = true;
}
/**
* Returns the directory for a temp dir, which is created by the
* test harness. Every test gets its own temp dir.
*
* @return nsIFile of the temporary directory
*/
function do_get_tempdir() {
let env = Cc["@mozilla.org/process/environment;1"].getService(
Ci.nsIEnvironment
);
// the python harness sets this in the environment for us
let path = env.get("XPCSHELL_TEST_TEMP_DIR");
let file = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
file.initWithPath(path);
return file;
}
/**
* Returns the directory for crashreporter minidumps.
*
* @return nsIFile of the minidump directory
*/
function do_get_minidumpdir() {
let env = Cc["@mozilla.org/process/environment;1"].getService(
Ci.nsIEnvironment
);
// the python harness may set this in the environment for us
let path = env.get("XPCSHELL_MINIDUMP_DIR");
if (path) {
let file = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
file.initWithPath(path);
return file;
}
return do_get_tempdir();
}
/**
* Registers a directory with the profile service,
* and return the directory as an nsIFile.
*
* @param notifyProfileAfterChange Whether to notify for "profile-after-change".
* @return nsIFile of the profile directory.
*/
function do_get_profile(notifyProfileAfterChange = false) {
if (!runningInParent) {
_testLogger.info("Ignoring profile creation from child process.");
return null;
}
let env = Cc["@mozilla.org/process/environment;1"].getService(
Ci.nsIEnvironment
);
// the python harness sets this in the environment for us
let profd = env.get("XPCSHELL_TEST_PROFILE_DIR");
let file = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
file.initWithPath(profd);
let provider = {
getFile(prop, persistent) {
persistent.value = true;
if (
prop == "ProfD" ||
prop == "ProfLD" ||
prop == "ProfDS" ||
prop == "ProfLDS" ||
prop == "TmpD"
) {
return file.clone();
}
return null;
},
QueryInterface: ChromeUtils.generateQI(["nsIDirectoryServiceProvider"]),
};
_Services.dirsvc
.QueryInterface(Ci.nsIDirectoryService)
.registerProvider(provider);
try {
_Services.dirsvc.undefine("TmpD");
} catch (e) {
// This throws if the key is not already registered, but that
// doesn't matter.
if (e.result != Cr.NS_ERROR_FAILURE) {
throw e;
}
}
// We need to update the crash events directory when the profile changes.
if (runningInParent && "@mozilla.org/toolkit/crash-reporter;1" in Cc) {
let crashReporter = Cc["@mozilla.org/toolkit/crash-reporter;1"].getService(
Ci.nsICrashReporter
);
crashReporter.UpdateCrashEventsDir();
}
if (!_profileInitialized) {
_Services.obs.notifyObservers(
null,
"profile-do-change",
"xpcshell-do-get-profile"
);
_profileInitialized = true;
if (notifyProfileAfterChange) {
_Services.obs.notifyObservers(
null,
"profile-after-change",
"xpcshell-do-get-profile"
);
}
}
// The methods of 'provider' will retain this scope so null out everything
// to avoid spurious leak reports.
env = null;
profd = null;
provider = null;
return file.clone();
}
/**
* This function loads head.js (this file) in the child process, so that all
* functions defined in this file (do_throw, etc) are available to subsequent
* sendCommand calls. It also sets various constants used by these functions.
*
* (Note that you may use sendCommand without calling this function first; you
* simply won't have any of the functions in this file available.)
*/
function do_load_child_test_harness() {
// Make sure this isn't called from child process
if (!runningInParent) {
do_throw("run_test_in_child cannot be called from child!");
}
// Allow to be called multiple times, but only run once
if (typeof do_load_child_test_harness.alreadyRun != "undefined") {
return;
}
do_load_child_test_harness.alreadyRun = 1;
_XPCSHELL_PROCESS = "parent";
let command =
"const _HEAD_JS_PATH=" +
uneval(_HEAD_JS_PATH) +
"; " +
"const _HEAD_FILES=" +
uneval(_HEAD_FILES) +
"; " +
"const _MOZINFO_JS_PATH=" +
uneval(_MOZINFO_JS_PATH) +
"; " +
"const _TEST_NAME=" +
uneval(_TEST_NAME) +
"; " +
// We'll need more magic to get the debugger working in the child
"const _JSDEBUGGER_PORT=0; " +
"_XPCSHELL_PROCESS='child';";
if (typeof _JSCOV_DIR === "string") {
command += " const _JSCOV_DIR=" + uneval(_JSCOV_DIR) + ";";
}
if (_TESTING_MODULES_DIR) {
command +=
" const _TESTING_MODULES_DIR=" + uneval(_TESTING_MODULES_DIR) + ";";
}
command += " load(_HEAD_JS_PATH);";
sendCommand(command);
}
/**
* Runs an entire xpcshell unit test in a child process (rather than in chrome,
* which is the default).
*
* This function returns immediately, before the test has completed.
*
* @param testFile
* The name of the script to run. Path format same as load().
* @param optionalCallback.
* Optional function to be called (in parent) when test on child is
* complete. If provided, the function must call do_test_finished();
* @return Promise Resolved when the test in the child is complete.
*/
function run_test_in_child(testFile, optionalCallback) {
return new Promise(resolve => {
var callback = () => {
resolve();
if (typeof optionalCallback == "undefined") {
do_test_finished();
} else {
optionalCallback();
}
};
do_load_child_test_harness();
var testPath = do_get_file(testFile).path.replace(/\\/g, "/");
do_test_pending("run in child");
sendCommand(
"_testLogger.info('CHILD-TEST-STARTED'); " +
"const _TEST_FILE=['" +
testPath +
"']; " +
"_execute_test(); " +
"_testLogger.info('CHILD-TEST-COMPLETED');",
callback
);
});
}
/**
* Execute a given function as soon as a particular cross-process message is received.
* Must be paired with do_send_remote_message or equivalent ProcessMessageManager calls.
*
* @param optionalCallback
* Optional callback that is invoked when the message is received. If provided,
* the function must call do_test_finished().
* @return Promise Promise that is resolved when the message is received.
*/
function do_await_remote_message(name, optionalCallback) {
return new Promise(resolve => {
var listener = {
receiveMessage(message) {
if (message.name == name) {
mm.removeMessageListener(name, listener);
resolve(message.data);
if (optionalCallback) {
optionalCallback(message.data);
} else {
do_test_finished();
}
}
},
};
var mm;
if (runningInParent) {
mm = Cc["@mozilla.org/parentprocessmessagemanager;1"].getService();
} else {
mm = Cc["@mozilla.org/childprocessmessagemanager;1"].getService();
}
do_test_pending();
mm.addMessageListener(name, listener);
});
}
/**
* Asynchronously send a message to all remote processes. Pairs with do_await_remote_message
* or equivalent ProcessMessageManager listeners.
*/
function do_send_remote_message(name, data) {
var mm;
var sender;
if (runningInParent) {
mm = Cc["@mozilla.org/parentprocessmessagemanager;1"].getService();
sender = "broadcastAsyncMessage";
} else {
mm = Cc["@mozilla.org/childprocessmessagemanager;1"].getService();
sender = "sendAsyncMessage";
}
mm[sender](name, data);
}
/**
* Add a test function to the list of tests that are to be run asynchronously.
*
* @param funcOrProperties
* A function to be run or an object represents test properties.
* Supported properties:
* skip_if : An arrow function which has an expression to be
* evaluated whether the test is skipped or not.
* @param func
* A function to be run only if the funcOrProperies is not a function.
* @param isTask
* Optional flag that indicates whether `func` is a task. Defaults to `false`.
*
* Each test function must call run_next_test() when it's done. Test files
* should call run_next_test() in their run_test function to execute all
* async tests.
*
* @return the test function that was passed in.
*/
var _gTests = [];
var _gRunOnlyThisTest = null;
function add_test(properties, func = properties, isTask = false) {
if (typeof properties == "function") {
properties = { isTask };
_gTests.push([properties, func]);
} else if (typeof properties == "object") {
properties.isTask = isTask;
_gTests.push([properties, func]);
} else {
do_throw("add_test() should take a function or an object and a function");
}
func.skip = () => (properties.skip_if = () => true);
func.only = () => (_gRunOnlyThisTest = func);
return func;
}
/**
* Add a test function which is a Task function.
*
* @param funcOrProperties
* An async function to be run or an object represents test properties.
* Supported properties:
* skip_if : An arrow function which has an expression to be
* evaluated whether the test is skipped or not.
* @param func
* An async function to be run only if the funcOrProperies is not a function.
*
* Task functions are functions fed into Task.jsm's Task.spawn(). They are async
* functions that emit promises.
*
* If an exception is thrown, a do_check_* comparison fails, or if a rejected
* promise is yielded, the test function aborts immediately and the test is
* reported as a failure.
*
* Unlike add_test(), there is no need to call run_next_test(). The next test
* will run automatically as soon the task function is exhausted. To trigger
* premature (but successful) termination of the function, simply return or
* throw a Task.Result instance.
*
* Example usage:
*
* add_task(async function test() {
* let result = await Promise.resolve(true);
*
* do_check_true(result);
*
* let secondary = await someFunctionThatReturnsAPromise(result);
* do_check_eq(secondary, "expected value");
* });
*
* add_task(async function test_early_return() {
* let result = await somethingThatReturnsAPromise();
*
* if (!result) {
* // Test is ended immediately, with success.
* return;
* }
*
* do_check_eq(result, "foo");
* });
*
* add_task({
* skip_if: () => !("@mozilla.org/telephony/volume-service;1" in Components.classes),
* }, async function test_volume_service() {
* let volumeService = Cc["@mozilla.org/telephony/volume-service;1"]
* .getService(Ci.nsIVolumeService);
* ...
* });
*/
function add_task(properties, func = properties) {
return add_test(properties, func, true);
}
_Task.Debugging.maintainStack = true;
/**
* Runs the next test function from the list of async tests.
*/
var _gRunningTest = null;
var _gTestIndex = 0; // The index of the currently running test.
var _gTaskRunning = false;
function run_next_test() {
if (_gTaskRunning) {
throw new Error(
"run_next_test() called from an add_task() test function. " +
"run_next_test() should not be called from inside add_task() " +
"under any circumstances!"
);
}
function _run_next_test() {
if (_gTestIndex < _gTests.length) {
// Check for uncaught rejections as early and often as possible.
_PromiseTestUtils.assertNoUncaughtRejections();
let _properties;
[_properties, _gRunningTest] = _gTests[_gTestIndex++];
// Must set to pending before we check for skip, so that we keep the
// running counts correct.
_testLogger.info(_TEST_NAME + " | Starting " + _gRunningTest.name);
do_test_pending(_gRunningTest.name);
if (
(typeof _properties.skip_if == "function" && _properties.skip_if()) ||
(_gRunOnlyThisTest && _gRunningTest != _gRunOnlyThisTest)
) {
let _condition = _gRunOnlyThisTest
? "only one task may run."
: _properties.skip_if.toSource().replace(/\(\)\s*=>\s*/, "");
if (_condition == "true") {
_condition = "explicitly skipped.";
}
let _message =
_gRunningTest.name +
" skipped because the following conditions were" +
" met: (" +
_condition +
")";
_testLogger.testStatus(
_TEST_NAME,
_gRunningTest.name,
"SKIP",
"SKIP",
_message
);
executeSoon(run_next_test);
return;
}
if (_properties.isTask) {
_gTaskRunning = true;
let startTime = Cu.now();
(async () => _gRunningTest())().then(
result => {
_gTaskRunning = false;
ChromeUtils.addProfilerMarker(
"xpcshell-test",
startTime,
_gRunningTest.name || "task"
);
if (_isGenerator(result)) {
Assert.ok(false, "Task returned a generator");
}
run_next_test();
},
ex => {
_gTaskRunning = false;
ChromeUtils.addProfilerMarker(
"xpcshell-test",
startTime,
_gRunningTest.name || "task"
);
try {
do_report_unexpected_exception(ex);
} catch (error) {
// The above throws NS_ERROR_ABORT and we don't want this to show up
// as an unhandled rejection later.
}
}
);
} else {
// Exceptions do not kill asynchronous tests, so they'll time out.
let startTime = Cu.now();
try {
_gRunningTest();
} catch (e) {
do_throw(e);
} finally {
ChromeUtils.addProfilerMarker(
"xpcshell-test",
startTime,
_gRunningTest.name || undefined
);
}
}
}
}
// For sane stacks during failures, we execute this code soon, but not now.
// We do this now, before we call do_test_finished(), to ensure the pending
// counter (_tests_pending) never reaches 0 while we still have tests to run
// (executeSoon bumps that counter).
executeSoon(_run_next_test, "run_next_test " + _gTestIndex);
if (_gRunningTest !== null) {
// Close the previous test do_test_pending call.
do_test_finished(_gRunningTest.name);
}
}
try {
// Set global preferences
if (runningInParent) {
let prefsFile = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
prefsFile.initWithPath(_PREFS_FILE);
_Services.prefs.readUserPrefsFromFile(prefsFile);
}
} catch (e) {
do_throw(e);
}
/**
* Changing/Adding scalars or events to Telemetry is supported in build-faster/artifacts builds.
* These need to be loaded explicitly at start.
* It usually happens once all of Telemetry is initialized and set up.
* However in xpcshell tests Telemetry is not necessarily fully loaded,
* so we help out users by loading at least the dynamic-builtin probes.
*/
try {
// We only need to run this in the parent process.
// We only want to run this for local developer builds (which should have a "default" update channel).
if (runningInParent && _AppConstants.MOZ_UPDATE_CHANNEL == "default") {
let startTime = Cu.now();
let _TelemetryController = ChromeUtils.import(
"resource://gre/modules/TelemetryController.jsm",
null
).TelemetryController;
let complete = false;
_TelemetryController.testRegisterJsProbes().finally(() => {
ChromeUtils.addProfilerMarker(
"xpcshell-test",
startTime,
"TelemetryController.testRegisterJsProbes"
);
complete = true;
});
_Services.tm.spinEventLoopUntil(() => complete);
}
} catch (e) {
do_throw(e);
}
function _load_mozinfo() {
let mozinfoFile = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
mozinfoFile.initWithPath(_MOZINFO_JS_PATH);
let stream = Cc["@mozilla.org/network/file-input-stream;1"].createInstance(
Ci.nsIFileInputStream
);
stream.init(mozinfoFile, -1, 0, 0);
let bytes = _NetUtil.readInputStream(stream, stream.available());
let decoded = JSON.parse(new TextDecoder().decode(bytes));
stream.close();
return decoded;
}
Object.defineProperty(this, "mozinfo", {
configurable: true,
get() {
let _mozinfo = _load_mozinfo();
Object.defineProperty(this, "mozinfo", {
configurable: false,
value: _mozinfo,
});
return _mozinfo;
},
});