зеркало из https://github.com/mozilla/gecko-dev.git
281 строка
8.4 KiB
JavaScript
281 строка
8.4 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
'use strict';
|
|
|
|
const {PushDB, PushService, PushServiceWebSocket} = serviceExports;
|
|
const {BroadcastService} = ChromeUtils.import("resource://gre/modules/PushBroadcastService.jsm", null);
|
|
const {JSONFile} = ChromeUtils.import("resource://gre/modules/JSONFile.jsm");
|
|
|
|
const {FileTestUtils} = ChromeUtils.import("resource://testing-common/FileTestUtils.jsm");
|
|
const {broadcastHandler} = ChromeUtils.import("resource://test/broadcast_handler.jsm");
|
|
|
|
const broadcastService = pushBroadcastService;
|
|
const assert = Assert;
|
|
const userAgentID = 'bd744428-f125-436a-b6d0-dd0c9845837f';
|
|
const channelID = '0ef2ad4a-6c49-41ad-af6e-95d2425276bf';
|
|
|
|
function run_test() {
|
|
do_get_profile();
|
|
setPrefs({
|
|
userAgentID,
|
|
alwaysConnect: true,
|
|
requestTimeout: 1000,
|
|
retryBaseInterval: 150
|
|
});
|
|
run_next_test();
|
|
}
|
|
|
|
function getPushServiceMock() {
|
|
return {
|
|
subscribed: [],
|
|
subscribeBroadcast: function(broadcastId, version) {
|
|
this.subscribed.push([broadcastId, version]);
|
|
},
|
|
};
|
|
}
|
|
|
|
add_task(async function test_register_success() {
|
|
await broadcastService._resetListeners();
|
|
let db = PushServiceWebSocket.newPushDB();
|
|
broadcastHandler.reset();
|
|
let notifications = broadcastHandler.notifications;
|
|
let socket;
|
|
registerCleanupFunction(() => {return db.drop().then(_ => db.close());});
|
|
|
|
await broadcastService.addListener("broadcast-test", "2018-02-01", {
|
|
moduleURI: "resource://test/broadcast_handler.jsm",
|
|
symbolName: "broadcastHandler",
|
|
});
|
|
|
|
PushServiceWebSocket._generateID = () => channelID;
|
|
|
|
var broadcastSubscriptions = [];
|
|
|
|
await PushService.init({
|
|
serverURI: "wss://push.example.org/",
|
|
db,
|
|
makeWebSocket(uri) {
|
|
return new MockWebSocket(uri, {
|
|
onHello(data) {
|
|
socket = this;
|
|
deepEqual(data.broadcasts, {"broadcast-test": "2018-02-01"}, "Handshake: doesn't consult listeners");
|
|
equal(data.messageType, 'hello', 'Handshake: wrong message type');
|
|
equal(data.uaid, userAgentID, 'Handshake: wrong device ID');
|
|
this.serverSendMsg(JSON.stringify({
|
|
messageType: 'hello',
|
|
status: 200,
|
|
uaid: userAgentID
|
|
}));
|
|
},
|
|
|
|
onBroadcastSubscribe(data) {
|
|
broadcastSubscriptions.push(data);
|
|
},
|
|
});
|
|
}
|
|
})
|
|
|
|
socket.serverSendMsg(JSON.stringify({
|
|
messageType: "broadcast",
|
|
broadcasts: {
|
|
"broadcast-test": "2018-03-02"
|
|
}
|
|
}));
|
|
|
|
await broadcastHandler.wasNotified;
|
|
|
|
deepEqual(notifications, [["2018-03-02", "broadcast-test"]], "Broadcast notification didn't get delivered");
|
|
|
|
deepEqual(await broadcastService.getListeners(), {
|
|
"broadcast-test": "2018-03-02"
|
|
}, "Broadcast version wasn't updated");
|
|
|
|
await broadcastService.addListener("example-listener", "2018-03-01", {
|
|
moduleURI: "resource://gre/modules/not-real-example.jsm",
|
|
symbolName: "doesntExist"
|
|
});
|
|
|
|
deepEqual(broadcastSubscriptions, [{
|
|
messageType: "broadcast_subscribe",
|
|
broadcasts: {"example-listener": "2018-03-01"}
|
|
}]);
|
|
});
|
|
|
|
add_task(async function test_handle_hello_broadcasts() {
|
|
PushService.uninit();
|
|
await broadcastService._resetListeners();
|
|
let db = PushServiceWebSocket.newPushDB();
|
|
broadcastHandler.reset();
|
|
let notifications = broadcastHandler.notifications;
|
|
registerCleanupFunction(() => {return db.drop().then(_ => db.close());});
|
|
|
|
await broadcastService.addListener("broadcast-test", "2018-02-01", {
|
|
moduleURI: "resource://test/broadcast_handler.jsm",
|
|
symbolName: "broadcastHandler",
|
|
});
|
|
|
|
PushServiceWebSocket._generateID = () => channelID;
|
|
|
|
await PushService.init({
|
|
serverURI: "wss://push.example.org/",
|
|
db,
|
|
makeWebSocket(uri) {
|
|
return new MockWebSocket(uri, {
|
|
onHello(data) {
|
|
deepEqual(data.broadcasts, {"broadcast-test": "2018-02-01"}, "Handshake: doesn't consult listeners");
|
|
equal(data.messageType, 'hello', 'Handshake: wrong message type');
|
|
equal(data.uaid, userAgentID, 'Handshake: wrong device ID');
|
|
this.serverSendMsg(JSON.stringify({
|
|
messageType: 'hello',
|
|
status: 200,
|
|
uaid: userAgentID,
|
|
broadcasts: {
|
|
"broadcast-test": "2018-02-02"
|
|
}
|
|
}));
|
|
},
|
|
|
|
onBroadcastSubscribe(data) {
|
|
broadcastSubscriptions.push(data);
|
|
},
|
|
});
|
|
}
|
|
})
|
|
|
|
await broadcastHandler.wasNotified;
|
|
|
|
deepEqual(notifications, [["2018-02-02", "broadcast-test"]], "Broadcast notification on hello was delivered");
|
|
|
|
deepEqual(await broadcastService.getListeners(), {
|
|
"broadcast-test": "2018-02-02"
|
|
}, "Broadcast version wasn't updated");
|
|
});
|
|
|
|
add_task(async function test_broadcast_unit() {
|
|
const fakeListenersData = {
|
|
"abc": {
|
|
version: "2018-03-04",
|
|
sourceInfo: {
|
|
moduleURI: "resource://gre/modules/abc.jsm",
|
|
symbolName: "getAbc"
|
|
}
|
|
},
|
|
"def": {
|
|
version: "2018-04-05",
|
|
sourceInfo: {
|
|
moduleURI: "resource://gre/modules/def.jsm",
|
|
symbolName: "getDef"
|
|
}
|
|
}
|
|
};
|
|
const path = FileTestUtils.getTempFile("broadcast-listeners.json").path;
|
|
|
|
const jsonFile = new JSONFile({path});
|
|
jsonFile.data = {
|
|
listeners: fakeListenersData,
|
|
};
|
|
await jsonFile._save();
|
|
|
|
const pushServiceMock = getPushServiceMock();
|
|
|
|
const broadcastService = new BroadcastService(pushServiceMock, path);
|
|
const listeners = await broadcastService.getListeners();
|
|
deepEqual(listeners, {
|
|
"abc": "2018-03-04",
|
|
"def": "2018-04-05"
|
|
});
|
|
|
|
await broadcastService.addListener("ghi", "2018-05-06", {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
});
|
|
|
|
deepEqual(pushServiceMock.subscribed, [
|
|
["ghi", "2018-05-06"]
|
|
]);
|
|
|
|
await broadcastService._saveImmediately();
|
|
|
|
const newJSONFile = new JSONFile({path});
|
|
await newJSONFile.load();
|
|
|
|
deepEqual(newJSONFile.data, {
|
|
listeners: {
|
|
...fakeListenersData,
|
|
ghi: {
|
|
version: "2018-05-06",
|
|
sourceInfo: {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
}
|
|
}
|
|
},
|
|
version: 1,
|
|
});
|
|
|
|
deepEqual(await broadcastService.getListeners(), {
|
|
"abc": "2018-03-04",
|
|
"def": "2018-04-05",
|
|
"ghi": "2018-05-06"
|
|
});
|
|
});
|
|
|
|
add_task(async function test_broadcast_initialize_sane() {
|
|
const path = FileTestUtils.getTempFile("broadcast-listeners.json").path;
|
|
const broadcastService = new BroadcastService(getPushServiceMock(), path);
|
|
deepEqual(await broadcastService.getListeners(), {}, "listeners should start out sane");
|
|
await broadcastService._saveImmediately();
|
|
let onDiskJSONFile = new JSONFile({path});
|
|
await onDiskJSONFile.load();
|
|
deepEqual(onDiskJSONFile.data, {listeners: {}, version: 1},
|
|
"written JSON file has listeners and version fields");
|
|
|
|
await broadcastService.addListener("ghi", "2018-05-06", {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
});
|
|
|
|
await broadcastService._saveImmediately();
|
|
|
|
onDiskJSONFile = new JSONFile({path});
|
|
await onDiskJSONFile.load();
|
|
|
|
deepEqual(onDiskJSONFile.data, {
|
|
listeners: {
|
|
ghi: {
|
|
version: "2018-05-06",
|
|
sourceInfo: {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
}
|
|
}
|
|
},
|
|
version: 1,
|
|
}, "adding listeners to initial state is written OK");
|
|
});
|
|
|
|
add_task(async function test_broadcast_reject_invalid_sourceinfo() {
|
|
const path = FileTestUtils.getTempFile("broadcast-listeners.json").path;
|
|
const broadcastService = new BroadcastService(getPushServiceMock(), path);
|
|
|
|
await assert.rejects(broadcastService.addListener("ghi", "2018-05-06", {
|
|
moduleName: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
}), /moduleURI must be a string/, "rejects sourceInfo that doesn't have moduleURI");
|
|
});
|
|
|
|
add_task(async function test_broadcast_reject_version_not_string() {
|
|
await assert.rejects(broadcastService.addListener("ghi", {}, {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
}), /version should be a string/, "rejects version that isn't a string");
|
|
});
|
|
|
|
add_task(async function test_broadcast_reject_version_empty_string() {
|
|
await assert.rejects(broadcastService.addListener("ghi", "", {
|
|
moduleURI: "resource://gre/modules/ghi.jsm",
|
|
symbolName: "getGhi"
|
|
}), /version should not be an empty string/, "rejects version that is an empty string");
|
|
});
|