Add testing for mail message class