forked from mirrors/gecko-dev
00778dd54e
Differential Revision: https://phabricator.services.mozilla.com/D51214 --HG-- rename : toolkit/modules/WebChannel.jsm => toolkit/actors/WebChannelParent.jsm rename : browser/base/content/test/general/browser_web_channel.js => toolkit/modules/tests/browser/browser_web_channel.js rename : browser/base/content/test/general/browser_web_channel.html => toolkit/modules/tests/browser/file_web_channel.html rename : browser/base/content/test/general/browser_web_channel_iframe.html => toolkit/modules/tests/browser/file_web_channel_iframe.html extra : moz-landing-system : lando
78 lines
2.1 KiB
JavaScript
78 lines
2.1 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
"use strict";
|
|
|
|
const { WebChannelBroker } = ChromeUtils.import(
|
|
"resource://gre/modules/WebChannel.jsm"
|
|
);
|
|
|
|
const VALID_WEB_CHANNEL_ID = "id";
|
|
const URL_STRING = "http://example.com";
|
|
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, reject) => {
|
|
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,
|
|
},
|
|
});
|
|
});
|
|
});
|