Extend removeFromMessages unit tests