Added tests for test_message