Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
const { WebChannelBroker } = ChromeUtils.importESModule(
"resource://gre/modules/WebChannel.sys.mjs"
);
const VALID_WEB_CHANNEL_ID = "id";
const VALID_WEB_CHANNEL_ORIGIN = Services.io.newURI(URL_STRING);
/**
* Test WebChannelBroker channel map
*/
add_test(function test_web_channel_broker_channel_map() {
let channel = {};
let channel2 = {};
Assert.equal(WebChannelBroker._channelMap.size, 0);
// make sure _channelMap works correctly
WebChannelBroker.registerChannel(channel);
Assert.equal(WebChannelBroker._channelMap.size, 1);
WebChannelBroker.registerChannel(channel2);
Assert.equal(WebChannelBroker._channelMap.size, 2);
WebChannelBroker.unregisterChannel(channel);
Assert.equal(WebChannelBroker._channelMap.size, 1);
// make sure the correct channel is unregistered
Assert.ok(!WebChannelBroker._channelMap.has(channel));
Assert.ok(WebChannelBroker._channelMap.has(channel2));
WebChannelBroker.unregisterChannel(channel2);
Assert.equal(WebChannelBroker._channelMap.size, 0);
run_next_test();
});
/**
* Test WebChannelBroker _listener test
*/
add_task(function test_web_channel_broker_listener() {
return new Promise(resolve => {
var channel = {
id: VALID_WEB_CHANNEL_ID,
_originCheckCallback: requestPrincipal => {
return VALID_WEB_CHANNEL_ORIGIN.prePath === requestPrincipal.origin;
},
deliver(data, sender) {
Assert.equal(data.id, VALID_WEB_CHANNEL_ID);
Assert.equal(data.message.command, "hello");
Assert.notEqual(sender, undefined);
WebChannelBroker.unregisterChannel(channel);
resolve();
},
};
WebChannelBroker.registerChannel(channel);
var mockEvent = {
id: VALID_WEB_CHANNEL_ID,
message: {
command: "hello",
},
};
WebChannelBroker.tryToDeliver(mockEvent, {
browsingContext: {},
principal: {
origin: URL_STRING,
},
});
});
});