Тестирование eventEmitter, который имеет прослушиватель асинхронной функции с использованием jest

Question

Я определил функцию async, которая содержит операторы await и использует ее в качестве прослушивателя сообщения подписки pubsub. Хотел бы проверить это с помощью шутки. Но столкнулся с проблемой.

код:

const pubsubPull = () => {
  const pubsub = new PubSub();
  const subscription = pubsub.subscription("test-sub");
  const handler = async message => {
    try {
      const payload = Buffer.from(message.data, "base64").toString();
      const data = JSON.parse(payload).data;
      await repository.insert(data);
      message.ack();
    } catch (err) {
      console.log("error")
    }
  };
  subscription.on("message", handler);
};

тест:

jest.mock("@google-cloud/pubsub");
jest.mock("./repository");

describe("listener", () => {
  const data = {
    id: "3ce91594-f648-41bf-9a37-4fa09c4afb3b",
    name: "ABCD",
    value: 1234
  };
  const eventBuffer = Buffer.from(JSON.stringify(data));
  const message = {
    id: "1561624788277",
    data: eventBuffer,
    ack: jest.fn()
  };

  const mockSubscripton = new EventEmitter();
  const pubsubClient = {
    subscription: jest.fn().mockReturnValue(mockSubscripton)
  };

  beforeAll(() => {
    PubSub.mockImplementation(() => pubsubClient);
    repository.insert.mockResolvedValue(true);
  });

  it("should ack the message", (done) => {
    pubsubPull();
    mockSubscripton.emit("message", message);

    expect(message.ack).toHaveBeenCalled();
    done();
  });
});

Тест не пройден с Expected mock function to have been called. But it was not called. Это означает, что утверждение выполняется еще до завершения выполнения функции слушателя. Как я могу ждать, пока это будет сделано, и запустить утверждение. Любая помощь приветствуется. узел v10.11.0, шутка "24.8.0"