Class MockMessage

packages/message/mock_message.ts:53–88

}

export class MockMessage implements Message {
  constructor(protected EE: EventEmitter<string, any>) {}

  connect(data: TMessage): Promise<MessageConnect> {
    return new Promise((resolve) => {
      const EE = new EventEmitter<string, any>();
      const con = new MockMessageConnect(EE);
      resolve(con);
      sleep(1).then(() => {
        this.EE.emit("connect", data, con);
      });
    });
  }

  sendMessage<T = any>(data: TMessage): Promise<T> {
    return new Promise((resolve) => {
      this.EE.emit("message", data, (resp: T) => {
        resolve(resp);
      });
    });
  }

  onConnect(callback: (data: TMessage, con: MessageConnect) => void): void {
    this.EE.on("connect", (data: any, con: MessageConnect) => {
      callback(data, con);
    });
  }

  onMessage(
    callback: (data: TMessage, sendResponse: (data: any) => void, _sender: RuntimeMessageSender) => void
  ): void {
    this.EE.on("message", (data: TMessage, sendResponse: (data: any) => void, sender: RuntimeMessageSender) => {
      callback(data, sendResponse, sender);
    });
  }
}