private async Task HeartbeatFinishedProcessing(Mocks mock, ConsumeRouter router) { // First we wait to be sure that a heartbeat has started being processed WaitOneSecondMaxForEvent("heatbeat", mock.HeartbeatCalled); // Then we wait to be sure that the current message is finished processing // (this message being the heartbeat or a following message) await router.StopProcessingTask(); }