зеркало из https://github.com/mozilla/gecko-dev.git
78 строки
1.9 KiB
JavaScript
78 строки
1.9 KiB
JavaScript
"use strict";
|
|
|
|
Components.utils.import("resource://gre/modules/ExtensionUtils.jsm");
|
|
var {
|
|
EventManager,
|
|
} = ExtensionUtils;
|
|
|
|
// WeakMap[Extension -> Set(callback)]
|
|
var messageHandlers = new WeakMap();
|
|
|
|
/* eslint-disable mozilla/balanced-listeners */
|
|
extensions.on("startup", (type, extension) => {
|
|
messageHandlers.set(extension, new Set());
|
|
});
|
|
|
|
extensions.on("shutdown", (type, extension) => {
|
|
messageHandlers.delete(extension);
|
|
});
|
|
|
|
extensions.on("test-message", (type, extension, ...args) => {
|
|
let handlers = messageHandlers.get(extension);
|
|
for (let handler of handlers) {
|
|
handler(...args);
|
|
}
|
|
});
|
|
/* eslint-enable mozilla/balanced-listeners */
|
|
|
|
extensions.registerSchemaAPI("test", (extension, context) => {
|
|
return {
|
|
test: {
|
|
sendMessage: function(...args) {
|
|
extension.emit("test-message", ...args);
|
|
},
|
|
|
|
notifyPass: function(msg) {
|
|
extension.emit("test-done", true, msg);
|
|
},
|
|
|
|
notifyFail: function(msg) {
|
|
extension.emit("test-done", false, msg);
|
|
},
|
|
|
|
log: function(msg) {
|
|
extension.emit("test-log", true, msg);
|
|
},
|
|
|
|
fail: function(msg) {
|
|
extension.emit("test-result", false, msg);
|
|
},
|
|
|
|
succeed: function(msg) {
|
|
extension.emit("test-result", true, msg);
|
|
},
|
|
|
|
assertTrue: function(value, msg) {
|
|
extension.emit("test-result", Boolean(value), msg);
|
|
},
|
|
|
|
assertFalse: function(value, msg) {
|
|
extension.emit("test-result", !value, msg);
|
|
},
|
|
|
|
assertEq: function(expected, actual, msg) {
|
|
extension.emit("test-eq", expected === actual, msg, String(expected), String(actual));
|
|
},
|
|
|
|
onMessage: new EventManager(context, "test.onMessage", fire => {
|
|
let handlers = messageHandlers.get(extension);
|
|
handlers.add(fire);
|
|
|
|
return () => {
|
|
handlers.delete(fire);
|
|
};
|
|
}).api(),
|
|
},
|
|
};
|
|
});
|