зеркало из https://github.com/mozilla/gecko-dev.git
135 строки
3.5 KiB
JavaScript
135 строки
3.5 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 userAgentID = "bd744428-f125-436a-b6d0-dd0c9845837f";
|
|
|
|
let clearForPattern = async function(testRecords, pattern) {
|
|
let patternString = JSON.stringify(pattern);
|
|
await PushService._clearOriginData(patternString);
|
|
|
|
for (let length = testRecords.length; length--; ) {
|
|
let test = testRecords[length];
|
|
let originSuffix = ChromeUtils.originAttributesToSuffix(
|
|
test.originAttributes
|
|
);
|
|
|
|
let registration = await PushService.registration({
|
|
scope: test.scope,
|
|
originAttributes: originSuffix,
|
|
});
|
|
|
|
let url = test.scope + originSuffix;
|
|
|
|
if (ObjectUtils.deepEqual(test.clearIf, pattern)) {
|
|
ok(
|
|
!registration,
|
|
"Should clear registration " + url + " for pattern " + patternString
|
|
);
|
|
testRecords.splice(length, 1);
|
|
} else {
|
|
ok(
|
|
registration,
|
|
"Should not clear registration " + url + " for pattern " + patternString
|
|
);
|
|
}
|
|
}
|
|
};
|
|
|
|
function run_test() {
|
|
do_get_profile();
|
|
setPrefs({
|
|
userAgentID,
|
|
requestTimeout: 1000,
|
|
retryBaseInterval: 150,
|
|
});
|
|
run_next_test();
|
|
}
|
|
|
|
add_task(async function test_webapps_cleardata() {
|
|
let db = PushServiceWebSocket.newPushDB();
|
|
registerCleanupFunction(() => {
|
|
return db.drop().then(_ => db.close());
|
|
});
|
|
|
|
let testRecords = [
|
|
{
|
|
scope: "https://example.org/1",
|
|
originAttributes: {},
|
|
clearIf: { inIsolatedMozBrowser: false },
|
|
},
|
|
{
|
|
scope: "https://example.org/1",
|
|
originAttributes: { inIsolatedMozBrowser: true },
|
|
clearIf: {},
|
|
},
|
|
];
|
|
|
|
let unregisterDone;
|
|
let unregisterPromise = new Promise(
|
|
resolve => (unregisterDone = after(testRecords.length, resolve))
|
|
);
|
|
|
|
PushService.init({
|
|
serverURI: "wss://push.example.org",
|
|
db,
|
|
makeWebSocket(uri) {
|
|
return new MockWebSocket(uri, {
|
|
onHello(data) {
|
|
equal(data.messageType, "hello", "Handshake: wrong message type");
|
|
ok(
|
|
!data.uaid,
|
|
"Should not send UAID in handshake without local subscriptions"
|
|
);
|
|
this.serverSendMsg(
|
|
JSON.stringify({
|
|
messageType: "hello",
|
|
status: 200,
|
|
uaid: userAgentID,
|
|
})
|
|
);
|
|
},
|
|
onRegister(data) {
|
|
equal(data.messageType, "register", "Register: wrong message type");
|
|
this.serverSendMsg(
|
|
JSON.stringify({
|
|
messageType: "register",
|
|
status: 200,
|
|
channelID: data.channelID,
|
|
uaid: userAgentID,
|
|
pushEndpoint: "https://example.com/update/" + Math.random(),
|
|
})
|
|
);
|
|
},
|
|
onUnregister(data) {
|
|
equal(data.code, 200, "Expected manual unregister reason");
|
|
unregisterDone();
|
|
},
|
|
});
|
|
},
|
|
});
|
|
|
|
await Promise.all(
|
|
testRecords.map(test =>
|
|
PushService.register({
|
|
scope: test.scope,
|
|
originAttributes: ChromeUtils.originAttributesToSuffix(
|
|
test.originAttributes
|
|
),
|
|
})
|
|
)
|
|
);
|
|
|
|
// Removes all the records, Excluding where `inIsolatedMozBrowser` is true.
|
|
await clearForPattern(testRecords, { inIsolatedMozBrowser: false });
|
|
|
|
// Removes the all the remaining records where `inIsolatedMozBrowser` is true.
|
|
await clearForPattern(testRecords, {});
|
|
|
|
equal(testRecords.length, 0, "Should remove all test records");
|
|
await unregisterPromise;
|
|
});
|