Function createMockEnvironment

test/unit/persisters/partykit.test.ts:55–105

(
  config: TinyBasePartyKitServerConfig = {},
)

};

const createMockEnvironment = (
  config: TinyBasePartyKitServerConfig = {},
): MockEnvironment => {
  const storage = createMockStorage();
  const sockets = new Map<string, MockSocket>();
  const party = {
    storage,
    broadcast: async (message: string, without: string[] = []): Promise<void> =>
      sockets.forEach((socket, socketId) =>
        without.includes(socketId) ? 0 : socket.receive(message),
      ),
  };
  const server = new TinyBasePartyKitServer(party as any);
  Object.assign(server.config, config);

  const createSocket = (): MockSocket => {
    const id = 'c' + sockets.size;
    const listeners = new Set<MessageListener>();
    const socket = {
      name: 'tinybase',
      partySocketOptions: {host: 'localhost:1999', room: 'room1'},
      send: (message: string): void => {
        void server.onMessage(message, {id} as any);
      },
      addEventListener: (_type: string, listener: MessageListener): void => {
        listeners.add(listener);
      },
      removeEventListener: (_type: string, listener: MessageListener): void => {
        listeners.delete(listener);
      },
      receive: (message: string): void => {
        listeners.forEach((listener) =>
          listener({data: message} as MessageEvent),
        );
      },
    } as unknown as MockSocket;
    sockets.set(id, socket);
    return socket;
  };

  return {
    storage,
    server,
    fetch: async (input: RequestInfo | URL, init?: RequestInit) => {
      const request =
        input instanceof Request ? input : new Request(String(input), init);
      return await server.onRequest(request as any);
    },
    createSocket,
  };
};

const createClient = (
  environment: MockEnvironment,