Class MockClients

packages/service-worker/worker/testing/clients.ts:51–114

}

export class MockClients implements Clients {
  private clients = new Map<string, MockClient>();

  add(clientId: string, url: string, type: ClientTypes = 'window'): void {
    if (this.clients.has(clientId)) {
      const existingClient = this.clients.get(clientId)!;
      if (existingClient.url === url) {
        return;
      }
      throw new Error(
        `Trying to add mock client with same ID (${existingClient.id}) and different URL ` +
          `(${existingClient.url} --> ${url})`,
      );
    }

    const client =
      type === 'window' ? new MockWindowClient(clientId, url) : new MockClient(clientId, url, type);
    this.clients.set(clientId, client);
  }

  remove(clientId: string): void {
    this.clients.delete(clientId);
  }

  async get(id: string): Promise<Client> {
    return this.clients.get(id)!;
  }

  getMock(id: string): MockClient | undefined {
    return this.clients.get(id);
  }

  async matchAll<T extends ClientQueryOptions>(
    options?: T,
  ): Promise<ReadonlyArray<T['type'] extends 'window' ? WindowClient : Client>> {
    const type = options?.type ?? 'window';
    const allClients = Array.from(this.clients.values());
    const matchedClients =
      type === 'all' ? allClients : allClients.filter((client) => client.type === type);

    // Order clients according to the [spec](https://w3c.github.io/ServiceWorker/#clients-matchall):
    // In most recently focused then most recently created order, with windows clients before other
    // clients.
    return (
      matchedClients
        // Sort in most recently created order.
        .reverse()
        // Sort in most recently focused order.
        .sort((a, b) => b.lastFocusedAt - a.lastFocusedAt)
        // Sort windows clients before other clients (otherwise leave existing order).
        .sort((a, b) => {
          const aScore = a.type === 'window' ? 1 : 0;
          const bScore = b.type === 'window' ? 1 : 0;
          return bScore - aScore;
        }) as any
    );
  }