Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
- This test gets skipped with pattern: os == 'android'
- Manifest: dom/push/test/xpcshell/xpcshell.toml
"use strict";
const { ForgetAboutSite } = ChromeUtils.importESModule(
);
var db;
var unregisterDefers = {};
var userAgentID = "4fe01c2d-72ac-4c13-93d2-bb072caf461d";
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 and expired subscriptions for a subdomain. The active subscription
// should be expired, then removed; the expired subscription should be
// removed immediately.
await putTestRecord(
db,
"active-sub-b",
4
);
await putTestRecord(
db,
"expired-sub",
0
);
// Active subscriptions for another subdomain. Should be unsubscribed and
// dropped.
await putTestRecord(
db,
"active-2",
16
);
await putTestRecord(
db,
"active-1-b",
8
);
await putTestRecord(
db,
"active-2-b",
16
);
// A privileged subscription with a real URL that should not be affected
// because its quota is set to `Infinity`.
await putTestRecord(
db,
"privileged",
Infinity
);
let handshakeDone;
let handshakePromise = new Promise(r => (handshakeDone = r));
PushService.init({
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");
resolve();
this.serverSendMsg(
JSON.stringify({
messageType: "unregister",
status: 200,
channelID: request.channelID,
})
);
},
});
},
});
// For cleared subscriptions, we only send unregister requests in the
// background and if we're connected.
await handshakePromise;
});
add_task(async function test_forgetAboutSubdomain() {
let modifiedScopes = [];
let promiseForgetSubs = Promise.all([
// Active subscriptions should be dropped.
promiseUnregister("active-sub"),
promiseObserverNotification(
PushServiceComponent.subscriptionModifiedTopic,
(subject, data) => {
modifiedScopes.push(data);
return modifiedScopes.length == 1;
}
),
]);
await ForgetAboutSite.removeDataFromDomain("sub.example.com");
await promiseForgetSubs;
deepEqual(
modifiedScopes.sort(compareAscending),
"Should fire modified notifications for active subscriptions"
);
let remainingIDs = await getAllKeyIDs(db);
deepEqual(
remainingIDs,
[
"active-1",
"active-1-b",
"active-2",
"active-2-b",
"active-sub-b",
"privileged",
],
"Should only forget subscriptions for subdomain"
);
});
add_task(async function test_forgetAboutRootDomain() {
let modifiedScopes = [];
let promiseForgetSubs = Promise.all([
promiseUnregister("active-1"),
promiseUnregister("active-2"),
promiseObserverNotification(
PushServiceComponent.subscriptionModifiedTopic,
(subject, data) => {
modifiedScopes.push(data);
return modifiedScopes.length == 2;
}
),
]);
await ForgetAboutSite.removeDataFromDomain("example.com");
await promiseForgetSubs;
deepEqual(
modifiedScopes.sort(compareAscending),
[
],
"Should fire modified notifications for entire domain"
);
let remainingIDs = await getAllKeyIDs(db);
deepEqual(
remainingIDs,
["active-1-b", "active-2-b", "active-sub-b", "privileged"],
"Should ignore privileged records with a real URL"
);
});
// Tests the legacy removeDataFromDomain method.
add_task(async function test_forgetAboutBaseDomain() {
let modifiedScopes = [];
let promiseForgetSubs = Promise.all([
promiseUnregister("active-sub-b"),
promiseUnregister("active-1-b"),
promiseUnregister("active-2-b"),
promiseObserverNotification(
PushServiceComponent.subscriptionModifiedTopic,
(subject, data) => {
modifiedScopes.push(data);
return modifiedScopes.length == 3;
}
),
]);
await ForgetAboutSite.removeDataFromDomain("example.net");
await promiseForgetSubs;
deepEqual(
modifiedScopes.sort(compareAscending),
[
],
"Should fire modified notifications for entire domain"
);
let remainingIDs = await getAllKeyIDs(db);
deepEqual(
remainingIDs,
["privileged"],
"Should ignore privileged records with a real URL"
);
});