Assert messages are received instead of flushing.

Since at the moment we call flush there is no guarantee
we received all desired messages.