Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* Any copyright is dedicated to the Public Domain.
"use strict";
var db;
var unregisterDefers = {};
var userAgentID = "4ce480ef-55b2-4f83-924c-dcd35ab978b4";
function promiseUnregister(keyID) {
return new Promise(r => (unregisterDefers[keyID] = r));
}
function run_test() {
do_get_profile();
setPrefs({
userAgentID,
});
run_next_test();
}
add_task(async function setup() {
db = PushServiceWebSocket.newPushDB();
registerCleanupFunction(() => db.drop().then(() => db.close()));
// Active subscriptions; should be expired then dropped.
await putTestRecord(db, "active-1", "https://example.info/some-page", 8);
await putTestRecord(db, "active-2", "https://example.com/another-page", 16);
// Expired subscription; should be dropped.
await putTestRecord(db, "expired", "https://example.net/yet-another-page", 0);
// A privileged subscription that should not be affected by sanitizing data
// because its quota is set to `Infinity`.
await putTestRecord(db, "privileged", "app://chrome/only", Infinity);
let handshakeDone;
let handshakePromise = new Promise(r => (handshakeDone = r));
PushService.init({
serverURI: "wss://push.example.org/",
db,
makeWebSocket(uri) {
return new MockWebSocket(uri, {
onHello() {
this.serverSendMsg(
JSON.stringify({
messageType: "hello",
uaid: userAgentID,
status: 200,
use_webpush: true,
})
);
handshakeDone();
},
onUnregister(request) {
let resolve = unregisterDefers[request.channelID];
equal(
typeof resolve,
"function",
"Dropped unexpected channel ID " + request.channelID
);
delete unregisterDefers[request.channelID];
equal(request.code, 200, "Expected manual unregister reason");
this.serverSendMsg(
JSON.stringify({
messageType: "unregister",
channelID: request.channelID,
status: 200,
})
);
resolve();
},
});
},
});
await handshakePromise;
});
add_task(async function test_sanitize() {
let modifiedScopes = [];
let changeScopes = [];
let promiseCleared = Promise.all([
// Active subscriptions should be unregistered.
promiseUnregister("active-1"),
promiseUnregister("active-2"),
promiseObserverNotification(
PushServiceComponent.subscriptionModifiedTopic,
(subject, data) => {
modifiedScopes.push(data);
return modifiedScopes.length == 3;
}
),
// Privileged should be recreated.
promiseUnregister("privileged"),
promiseObserverNotification(
PushServiceComponent.subscriptionChangeTopic,
(subject, data) => {
changeScopes.push(data);
return changeScopes.length == 1;
}
),
]);
await PushService.clear({
domain: "*",
});
await promiseCleared;
deepEqual(
modifiedScopes.sort(compareAscending),
[
],
"Should modify active subscription scopes"
);
deepEqual(
changeScopes,
"Should fire change notification for privileged scope"
);
let remainingIDs = await getAllKeyIDs(db);
deepEqual(remainingIDs, [], "Should drop all subscriptions");
});