0f8e710c7c
Monkey patch `MessagePort.prototype.onmessage` and `MessagePort.prototype.onmessageerror` to make these properties's value(callback function) run in the zone when these value are set. PR Close #34610
50 lines
1.4 KiB
TypeScript
50 lines
1.4 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google Inc. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.io/license
|
|
*/
|
|
|
|
/**
|
|
* Test MessagePort monkey patch.
|
|
*/
|
|
describe('MessagePort onproperties', () => {
|
|
let iframe: any;
|
|
beforeEach(() => {
|
|
iframe = document.createElement('iframe');
|
|
const html = `<body>
|
|
<script>
|
|
window.addEventListener('message', onMessage);
|
|
function onMessage(e) {
|
|
// Use the transfered port to post a message back to the main frame
|
|
e.ports[0].postMessage('Message back from the IFrame');
|
|
}
|
|
</script>
|
|
</body>`;
|
|
iframe.src = 'data:text/html;charset=utf-8,' + encodeURI(html);
|
|
});
|
|
afterEach(() => {
|
|
if (iframe) {
|
|
document.body.removeChild(iframe);
|
|
}
|
|
});
|
|
|
|
it('onmessge should in the zone', (done) => {
|
|
const channel = new MessageChannel();
|
|
const zone = Zone.current.fork({name: 'zone'});
|
|
iframe.onload = function() {
|
|
zone.run(() => {
|
|
channel.port1.onmessage = function() {
|
|
expect(Zone.current.name).toBe(zone.name);
|
|
done();
|
|
};
|
|
Zone.current.fork({name: 'zone1'}).run(() => {
|
|
iframe.contentWindow.postMessage('Hello from the main page!', '*', [channel.port2]);
|
|
});
|
|
});
|
|
};
|
|
document.body.appendChild(iframe);
|
|
});
|
|
});
|