Fix overload test

 * Wait for all messages to be processed by fake vnode in order to get
   accurate count
 * Abstract out some common code to update_msg_counter()