Add test for MessageService