angular-docs-cn/packages/service-worker/testing/mock.ts

63 lines
2.0 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
*/
import {Subject} from 'rxjs/Subject';
export class MockServiceWorkerContainer {
private onControllerChange: Function[] = [];
private onMessage: Function[] = [];
private registration: MockServiceWorkerRegistration|null = null;
controller: MockServiceWorker|null = null;
messages = new Subject();
addEventListener(event: 'controllerchange'|'message', handler: Function) {
if (event === 'controllerchange') {
this.onControllerChange.push(handler);
} else if (event === 'message') {
this.onMessage.push(handler);
}
}
removeEventListener(event: 'controllerchange', handler: Function) {
if (event === 'controllerchange') {
this.onControllerChange = this.onControllerChange.filter(h => h !== handler);
} else if (event === 'message') {
this.onMessage = this.onMessage.filter(h => h !== handler);
}
}
async register(url: string): Promise<void> { return; }
async getRegistration(): Promise<ServiceWorkerRegistration> { return this.registration as any; }
setupSw(url: string = '/ngsw-worker.js'): void {
this.registration = new MockServiceWorkerRegistration();
this.controller = new MockServiceWorker(this, url);
this.onControllerChange.forEach(onChange => onChange(this.controller));
}
get mockRegistration(): Promise<MockServiceWorkerRegistration> {
return Promise.resolve(this.registration !);
}
sendMessage(value: Object): void {
this.onMessage.forEach(onMessage => onMessage({
data: value,
}));
}
}
export class MockServiceWorker {
constructor(private mock: MockServiceWorkerContainer, readonly scriptURL: string) {}
postMessage(value: Object) { this.mock.messages.next(value); }
}
export class MockServiceWorkerRegistration {}